Update app.py

app.py

@@ -6,6 +6,7 @@ from PIL import Image
 import torch
 import torchxrayvision as xrv
 from torchvision import transforms
+import re
 
 # -----------------------------
 # Imaging Agent (Chest X-ray, proxy for lung cancer risk)
@@ -55,7 +56,6 @@ CANCER_MARKERS = {
     "afp": {"unit": "ng/mL", "high": 10},
 }
 
-import re
 def lab_agent(text: str):
     if not text.strip():
         return "No lab text provided."
@@ -88,34 +88,15 @@ def coordinator(imaging_txt, lab_txt):
 # -----------------------------
 # Gradio UI
 # -----------------------------
-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)")
-            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 (e.g., 'PSA: 8 ng/mL')")
-            lab_out = gr.Textbox(label="Lab Agent Output")
-
-    run_btn = gr.Button("Run Agents")
-    coord_out = gr.Textbox(label="Coordinator Summary", lines=10)
-
-    def run_all(image, labs):
-        txt, raw = imaging_agent(image) if image else ("No image.", None)
-        lab = lab_agent(labs)
-        coord = coordinator(txt, lab)
-        return txt, raw, lab, coord
-
-    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"
 
+def run_all(image, labs):
+    txt, raw = imaging_agent(image) if image else ("No image.", None)
+    lab = lab_agent(labs)
+    coord = coordinator(txt, lab)
+    return txt, raw, lab, coord
+
 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.")
@@ -132,8 +113,10 @@ with gr.Blocks(theme="soft") as demo:
     run_btn = gr.Button("Run Agents")
     coord_out = gr.Textbox(label="Coordinator Summary", lines=10)
 
+    run_btn.click(run_all, inputs=[img_in, lab_in], outputs=[imaging_out, imaging_raw, lab_out, coord_out])
 
 demo.launch()
 
 
 
+