#
#  Import modules
#


from qiskit import QuantumCircuit
from qiskit.primitives import StatevectorSampler
from qiskit.visualization import plot_histogram
import matplotlib.pyplot as plt

#
#  Define quantum circuit
#

qc = QuantumCircuit(2)              # starts as                                                |00>
qc.h(0)                             # Hadamard on bit 0:                             1/sqrt(2) (|00>+|01>)
qc.cx(0, 1)                         # Flip q1 if q0: |00> -> |00>  |01> -> |11>      1/sqrt(2) (|00>+|11>)
qc.x(0)                             # Not q0:                                        1/sqrt(2) (|10>+|01>)
qc.measure_all()                    # measurement is 50%  10 and 50% 01


#
# Visualize circuit and resultss
#

qc.draw("mpl")

sampler = StatevectorSampler()
result = sampler.run([qc], shots=1024).result()
print(result[0].data.meas.get_counts())

counts = result[0].data.meas.get_counts()
plot_histogram(counts)
plt.show()