Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import os
 from utils import load_dicom
 from model import generate_report
+from utils import load_dicom, generate_pdf
 
 def process_scan(dicom_file):
     if dicom_file is None:
@@ -19,6 +20,12 @@ def run_generate(dicom_file):
     
     return report, gr.DownloadButton(visible=True)
 
+def download_report(report_text):
+    if not report_text or not report_text.strip():
+        gr.Warning("No report to download yet.")
+        return None
+    return generate_pdf(report_text)
+
 # Build the UI
 with gr.Blocks(title="MRI Brain Report Generator") as app:
     
@@ -68,4 +75,10 @@ with gr.Blocks(title="MRI Brain Report Generator") as app:
         outputs=[report_box, download_btn]
     )
 
+    download_btn.click(
+        fn=download_report,
+        inputs=report_out,
+        outputs=download_btn
+    )
+
 app.launch()