Simplified UI

app.py

@@ -4,37 +4,54 @@ from src.pipeline import generate_report
 
 # Pre-load models
 from src.tools_loader import get_tools
-_ = get_tools()
+_ = get_tools()  # Load necessary tools/models at startup for faster inference
 
 def process_upload(image_path: str):
+    # Handle case where no image is uploaded
     if image_path is None:
         yield "Please upload a chest X-ray image."
         return
 
-    start = time.time()
-    yield "Analyzing image..."
+    start = time.time()  # Start timer for report generation
+    yield "Analyzing image..."  # Inform user that analysis is in progress
     
-    report = generate_report(image_path)
-    elapsed = time.time() - start
+    report = generate_report(image_path)  # Generate radiology report using pipeline
+    elapsed = time.time() - start  # Calculate elapsed time
     
+    # Display the generated report along with time taken
     yield f"### Radiology Report\n{report}\n\n*Generated in {elapsed:.1f}s*"
 
-with gr.Blocks(theme=gr.themes.Soft(), title="Radiology Assistant") as demo:
-    gr.Markdown("#Multi-Agent Radiology Assistant")
+# Create enhanced Gradio interface with custom layout
+with gr.Blocks(title="Multi-Agent Radiology Assistant") as demo:
+    gr.Markdown("# Multi-Agent Radiology Assistant")  # App title
+    gr.Markdown("Upload a chest X-ray image to receive an AI-generated radiology report")  # App description
     
-    image_input = gr.Image(type="filepath", label="Upload Chest X-ray")
-    generate_btn = gr.Button("Generate Report", variant="primary")
-    report_output = gr.Markdown("Upload an image and click Generate Report.")
+    with gr.Row():  # Layout row for input
+        with gr.Column(scale=1):  # Column for image upload and button
+            input_image = gr.Image(
+                type="filepath", 
+                label="Upload Chest X-ray",
+                height=400  # Set image display height
+            )
+            process_btn = gr.Button("Generate Report", variant="primary")  # Button to trigger report generation
     
-    generate_btn.click(process_upload, image_input, report_output)
-
-
-gr.Markdown(
+    with gr.Row():  # Layout row for output
+        output_report = gr.Markdown(
+            label="Radiology Report",
+            height=300  # Set output display height
+        )
+    
+    # Connect the button to processing function
+    process_btn.click(
+        fn=process_upload,  # Function to call on button click
+        inputs=[input_image],  # Input to the function
+        outputs=[output_report]  # Output from the function
+    )
+    
+    gr.Markdown(
         "Download and use any frontal chest X-ray PNG/JPG file from the internet and click **Generate Report**.\n"
         "### NOTE: This is just a demo. It is not intended to diagnose or suggest treatment of any disease or condition, and should not be used for medical advice."
-    )
+    )  # Disclaimer and instructions
 
 if __name__ == "__main__":
-    demo.launch()
-
-
+    demo.launch()  # Launch the Gradio app if run as main script