Changed UI

app.py

@@ -3,7 +3,7 @@ import time
 from src.pipeline import generate_report
 
 # ------------------------------------------------------------------
-# 1. Pre-load models (unchanged)
+# 1. Pre-load models
 # ------------------------------------------------------------------
 from src.tools_loader import get_tools
 _ = get_tools()
@@ -17,13 +17,13 @@ def process_upload(image_path: str):
     Gradio shows spinner automatically during execution.
     """
     if image_path is None:
-        yield " **Please upload a chest X-ray image to begin analysis.**"
+        yield "**Please upload a chest X-ray image to begin analysis.**"
         return
 
     start = time.time()
     
     # Show loading state immediately
-    yield " **Analyzing X-ray image...**\n\nThis might take a couple of seconds..."
+    yield "**Analyzing X-ray image...**\n\nThis will take a few seconds..."
     
     # Generate the actual report
     report = generate_report(image_path)
@@ -36,7 +36,7 @@ def process_upload(image_path: str):
 {report}
 
 ---
-*Generated in {elapsed:.1f} seconds*"""
+*Report generated in {elapsed:.1f} seconds*"""
 
 # ------------------------------------------------------------------
 # 3. Gradio UI - Vertical Layout
@@ -46,15 +46,17 @@ with gr.Blocks(
     title="Multi-Agent Radiology Assistant",
     css="""
     .image-container { max-width: 600px; margin: 0 auto; }
-    .report-container { margin-top: 20px; }
-    .generate-btn { margin: 20px auto; display: block; }
+    .report-container { margin-top: 40px; padding-top: 20px; }
+    .generate-btn { margin: 30px auto; display: block; }
+    .progress-bar { z-index: 1000 !important; position: relative; }
+    .gradio-container .wrap { z-index: auto; }
     """
 ) as demo:
     
     # Header
     gr.Markdown(
         "# Multi-Agent Radiology Assistant\n"
-        "Upload a chest X-ray image to receive a professional radiology report"
+        "Upload a chest X-ray image to receive an AI-powered radiology report"
     )
     
     # Image upload section (centered, top)
@@ -66,13 +68,16 @@ with gr.Blocks(
             elem_classes=["image-container"]
         )
         
-        # Generate button (centered)
+        # Generate button (centered with more spacing)
         generate_btn = gr.Button(
             "Generate Report",
             variant="primary",
             size="lg",
             elem_classes=["generate-btn"]
         )
+        
+        # Add some spacing to prevent overlap
+        gr.HTML("<div style='height: 20px;'></div>")
     
     # Report output section (bottom)
     with gr.Column(elem_classes=["report-container"]):
@@ -81,18 +86,19 @@ with gr.Blocks(
             label="Analysis Results"
         )
     
-    # Event handler
+    # Event handler with progress settings
     generate_btn.click(
         fn=process_upload,
         inputs=input_image,
         outputs=output_report,
-        show_progress="full"  # Shows Gradio's built-in progress bar
+        show_progress="full",  # Shows Gradio's built-in progress bar
+        concurrency_limit=1   # Prevent multiple simultaneous requests
     )
     
     # Footer with example hint
     gr.Markdown(
         "### Need an example?\n"
-        "Download and use any frontal chest X-ray PNG/JPG file and click **Generate Report**."
+        "Use any frontal chest X-ray PNG/JPG file and click **Generate Report**."
     )
 
 if __name__ == "__main__":