Update app.py

app.py

@@ -114,21 +114,39 @@ print(counts)
 from qiskit import QuantumCircuit, transpile
 from qiskit_aer import AerSimulator
 from qiskit.circuit.library import GroverOperator
-from qiskit.algorithms import AmplificationProblem
 
-# Define a 2-qubit oracle that marks the state |01> as a solution
-def oracle(circuit):
-    circuit.cz(0, 1)  # Apply a controlled-Z gate for the |01> state
-
-# Initialize a 2-qubit quantum circuit
-qc = QuantumCircuit(2)
-
-# Apply the oracle
-oracle(qc)
-
-# Use GroverOperator to amplify the marked solution
-problem = AmplificationProblem(oracle_circuit=qc)
-grover_circuit = GroverOperator(problem)
+# Define the oracle circuit to mark the solution state |01>
+def create_oracle():
+    oracle_circuit = QuantumCircuit(2)
+    oracle_circuit.cz(0, 1)  # Mark |01> as the solution
+    return oracle_circuit
+
+# Define the full Grover search circuit
+def create_grover_circuit():
+    oracle = create_oracle()
+    grover_circuit = QuantumCircuit(2, 2)
+
+    # Apply Hadamard to all qubits
+    grover_circuit.h([0, 1])
+
+    # Apply the oracle
+    grover_circuit.append(oracle.to_gate(), [0, 1])
+
+    # Apply Grover diffusion operator
+    grover_circuit.h([0, 1])
+    grover_circuit.x([0, 1])
+    grover_circuit.h(1)
+    grover_circuit.cz(0, 1)
+    grover_circuit.h(1)
+    grover_circuit.x([0, 1])
+    grover_circuit.h([0, 1])
+
+    # Measure the qubits
+    grover_circuit.measure([0, 1], [0, 1])
+    return grover_circuit
+
+# Create the Grover circuit
+grover_circuit = create_grover_circuit()
 
 # Use AerSimulator as the backend
 simulator = AerSimulator()