Update app.py

app.py

@@ -1,8 +1,6 @@
 import gradio as gr
 import os
-from utils import load_dicoms
 from model import generate_report
-from utils import load_dicoms, generate_pdf
 from utils import convert_dicom_zip_to_nifti, generate_pdf
 
 # def process_scan(dicom_files):
@@ -59,6 +57,12 @@ def run_generate(sequence_images):
 
     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)
+
 def reset_on_clear():
     return (
         gr.Gallery(value=None, visible=False),  # scan_display