import os
import time
from softqcos_sdk import QCOSClient

# Initialize QCOS Client
client = QCOSClient()

# Define an 8-qubit entangled GHZ state circuit
ghz_circuit_qasm = """
OPENQASM 2.0;
include "qelib1.inc";
qreg q[8];
creg c[8];
h q[0];
cx q[0], q[1];
cx q[1], q[2];
cx q[2], q[3];
cx q[3], q[4];
cx q[4], q[5];
cx q[5], q[6];
cx q[6], q[7];
measure q -> c;
"""

# 1. Submit the job asynchronously
print("Submitting GHZ state job to QCOS simulator...")
job_info = client.jobs.submit(
    qasm=ghz_circuit_qasm,
    backend_id="simulator_aer",
    shots=1000
)
job_id = job_info["job_id"]
print(f"Submitted successfully. Job ID: {job_id}")

# 2. Poll the job status using client.jobs.get
while True:
    status_report = client.jobs.get(job_id)
    status = status_report.get("status")
    print(f"Job Status: {status}")
    
    if status in ("completed", "failed", "cancelled"):
        break
    time.sleep(2)

# 3. Retrieve final counts
if status == "completed":
    result_data = client.jobs.result(job_id)
    print("Execution results:")
    print("Counts:", result_data.get("counts", result_data))
else:
    print("Job failed or was cancelled.")

# Synchronous one-liner submission
result = client.jobs.execute(
    qasm=ghz_circuit_qasm,
    backend_id="simulator_aer",
    shots=1024
)

print("Measurement counts (blocking):", result["result"]["counts"])

raw_qasm = """
OPENQASM 2.0;
include "qelib1.inc";
qreg q[2];
h q[0];
h q[0];  # Redundant identity gates
cx q[0], q[1];
"""

# Optimize gates and depth using the circuits module
optimized = client.circuits.optimize(raw_qasm, optimization_level=3)
print("Optimized QASM:")
print(optimized["qasm"])
print(f"Gate count reduced from {optimized['original_gates']} to {optimized['optimized_gates']}")

# Translate to another framework representation (e.g. Qiskit, Cirq, Braket)
converted = client.circuits.convert(raw_qasm, target_format="qiskit")
print("Framework format details:", converted["framework_details"])

# Validate QASM syntax
validation = client.circuits.validate(raw_qasm)
if validation["valid"]:
    print("✅ Circuit syntax is correct!")
else:
    print(f"❌ Syntax Error: {validation['error_details']}")

# Retrieve circuit properties (depth, qubit count, gate distribution)
# Pre-requisite: Submit circuit first
circuit_meta = client.circuits.submit(raw_qasm)
stats = client.circuits.stats(circuit_meta["circuit_id"])
print(f"Qubit count: {stats['num_qubits']}")
print(f"Circuit depth: {stats['depth']}")
print(f"Gate distribution: {stats['gates_distribution']}")