Update app.py

app.py

@@ -112,6 +112,27 @@ with gr.Blocks(theme="soft") as demo:
 
     run_btn.click(run_all, inputs=[img_in, lab_in], outputs=[imaging_out, imaging_raw, lab_out, coord_out])
 
+    # Preload with demo samples
+SAMPLE_XRAY = "samples/sample_xray.jpg"
+SAMPLE_LABS = "PSA: 8 ng/mL\nCA125: 20 U/mL\nAFP: 15 ng/mL"
+
+with gr.Blocks(theme="soft") as demo:
+    gr.Markdown("# 🏥 AI Diagnostics Agent: Early Cancer Discovery (Demo)")
+    gr.Markdown("Upload a chest X-ray or paste tumor marker labs.\n\n⚠️ Research demo only. Not for clinical use.")
+
+    with gr.Row():
+        with gr.Column():
+            img_in = gr.Image(type="filepath", label="Chest X-ray (PNG/JPG)", value=SAMPLE_XRAY)
+            imaging_out = gr.Textbox(label="Imaging Agent Output")
+            imaging_raw = gr.Code(label="Probabilities JSON", language="json")
+        with gr.Column():
+            lab_in = gr.Textbox(lines=6, label="Lab Results", value=SAMPLE_LABS)
+            lab_out = gr.Textbox(label="Lab Agent Output")
+
+    run_btn = gr.Button("Run Agents")
+    coord_out = gr.Textbox(label="Coordinator Summary", lines=10)
+
+
 demo.launch()