Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import os
 from utils import load_dicom
+from model import generate_report
 
 def process_scan(dicom_file):
     if dicom_file is None:
@@ -9,6 +10,15 @@ def process_scan(dicom_file):
     image = load_dicom(dicom_file.name)
     return image, gr.Button(interactive=True)
 
+def run_generate(dicom_file):
+    if dicom_file is None:
+        return "", gr.DownloadButton(visible=False)
+    
+    image = load_dicom(dicom_file.name)
+    report = generate_report(image)
+    
+    return report, gr.DownloadButton(visible=True)
+
 # Build the UI
 with gr.Blocks(title="MRI Brain Report Generator") as app:
     
@@ -52,4 +62,10 @@ with gr.Blocks(title="MRI Brain Report Generator") as app:
         outputs=[scan_display, generate_btn]
     )
 
+    generate_btn.click(
+        fn=run_generate,
+        inputs=upload_btn,
+        outputs=[report_box, download_btn]
+    )
+
 app.launch()