Update app.py

app.py

@@ -1,4 +1,3 @@
-import json
 import numpy as np
 import gradio as gr
 from PIL import Image
@@ -9,6 +8,7 @@ from skimage.transform import resize as sk_resize
 import matplotlib.pyplot as plt
 import io
 import re
+import json
 
 # -----------------------------
 # Imaging Agent (Chest X-ray)
@@ -66,11 +66,11 @@ def imaging_agent(image_path: str):
 
         # Format outputs
         lines = [f"{name}: {p*100:.1f}%" for name, p in sorted(focus, key=lambda x: x[1], reverse=True)]
-        table = {name: round(p, 4) for name, p in focus}
+        prob_text = "\n".join(lines)
 
         return (
             "🖼️ Imaging Agent (Chest X-ray for cancer risk)\n" + "\n".join(lines),
-            json.dumps(table, indent=2),
+            prob_text,
             heatmap_img
         )
 
@@ -142,10 +142,10 @@ SAMPLE_TEXTS = {
 # Runner
 # -----------------------------
 def run_all(image, labs):
-    txt, raw, heatmap = imaging_agent(image) if image else ("No image.", None, None)
+    txt, prob_text, heatmap = imaging_agent(image) if image else ("No image.", None, None)
     lab = lab_agent(labs)
     coord = coordinator(txt, lab)
-    return txt, raw, heatmap, lab, coord
+    return txt, prob_text, heatmap, lab, coord
 
 
 # -----------------------------
@@ -164,7 +164,7 @@ with gr.Blocks(theme="soft") as demo:
             )
             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")
+            imaging_raw = gr.Textbox(label="Probabilities (%)", lines=6)
             imaging_heatmap = gr.Image(label="Heatmap Overlay")
         with gr.Column():
             text_dropdown = gr.Dropdown(
@@ -209,3 +209,4 @@ demo.launch()
 
 
 
+