Update app.py

app.py

@@ -3,11 +3,11 @@ import os
 from utils import load_dicom
 
 def process_scan(dicom_file):
-    """
-    Takes uploaded DICOM file, returns image to display.
-    """
+    if dicom_file is None:
+        return None, gr.Button(interactive=False)
+    
     image = load_dicom(dicom_file.name)
-    return image
+    return image, gr.Button(interactive=True)
 
 # Build the UI
 with gr.Blocks(title="MRI Brain Report Generator") as app:
@@ -38,14 +38,18 @@ with gr.Blocks(title="MRI Brain Report Generator") as app:
     
     # Buttons row
     with gr.Row():
-        generate_btn = gr.Button("Generate Report", variant="primary")
+        generate_btn = gr.Button(
+            "Generate Report", 
+            variant="primary",
+            interactive=False    # disabled by default
+        )
         download_btn = gr.DownloadButton("Download PDF", visible=False)
 
-    # Connect upload → image preview
+    # When file is uploaded → show preview AND enable generate button
     upload_btn.change(
         fn=process_scan,
         inputs=upload_btn,
-        outputs=scan_display
+        outputs=[scan_display, generate_btn]
     )
 
 app.launch()