Upload app.py

app.py

@@ -19,6 +19,7 @@ from quread.heatmap import (
     plotly_available,
     HeatmapConfig,
 )
+from quread.noise_model import sample_noisy_counts
 from quread.metrics import (
     compute_metrics_from_csv,
     to_metrics_csv,
@@ -125,9 +126,27 @@ def apply_cnot(qc, last_counts, control, target):
     return qc, last_counts, f"✅ Applied CNOT (q{control} -> q{target})."
 
 
-def sample_shots(qc, shots):
+def sample_shots(
+    qc,
+    shots,
+    noise_preview_enabled,
+    readout_scale,
+    depolarizing_prob,
+    calibration_text,
+):
+    if bool(noise_preview_enabled):
+        last_counts = sample_noisy_counts(
+            state=qc.state,
+            n_qubits=qc.n_qubits,
+            shots=int(shots),
+            calibration_json=str(calibration_text or ""),
+            readout_scale=float(readout_scale),
+            depolarizing_prob=float(depolarizing_prob),
+        )
+        return last_counts, "✅ Sampled noisy hardware-preview shots."
+
     last_counts = qc.sample(shots=int(shots))
-    return last_counts, "✅ Sampled shots."
+    return last_counts, "✅ Sampled ideal shots."
 
 
 def measure_collapse(qc, shots):
@@ -375,6 +394,10 @@ with gr.Blocks(theme=theme, css=CSS, title="Quread.ai — State Vector Studio")
             gr.Markdown("---")
             gr.Markdown("### Quick Actions")
             reset_btn = gr.Button("Reset Simulator", variant="secondary")
+            with gr.Accordion("Hardware Noise Preview", open=False):
+                noise_preview_enabled = gr.Checkbox(label="Enable noise-aware sampling", value=False)
+                noise_readout_scale = gr.Slider(0.0, 3.0, value=1.0, step=0.05, label="Readout error scale")
+                noise_depolarizing = gr.Slider(0.0, 0.5, value=0.0, step=0.01, label="Depolarizing probability")
             gr.Markdown("<div class='small-note'>Analytics controls are in the <b>Hardware Analytics</b> tab.</div>")
 
         # Main
@@ -633,7 +656,14 @@ with gr.Blocks(theme=theme, css=CSS, title="Quread.ai — State Vector Studio")
 
     sample_btn.click(
         fn=sample_shots,
-        inputs=[qc_state, shots],
+        inputs=[
+            qc_state,
+            shots,
+            noise_preview_enabled,
+            noise_readout_scale,
+            noise_depolarizing,
+            calibration_json,
+        ],
         outputs=[last_counts_state, status],
     ).then(
         fn=update_views,