Update app.py

app.py

@@ -2,24 +2,25 @@ import gradio as gr
 import os
 from utils import load_dicom
 from model import generate_report
-from utils import load_dicom, generate_pdf
+from utils import load_dicoms, generate_pdf
 
 def process_scan(dicom_files):
     if not dicom_files:
-        return None, gr.Button(interactive=False)
+        return gr.Gallery(visible=False), gr.Button(interactive=False)
     
-    images = load_dicom(dicom_files)
-    return image, gr.Button(interactive=True)
+    images = load_dicoms(dicom_files)
+    if not images:
+        return gr.Gallery(visible=False), gr.Button(interactive=False)
+        
+    return gr.Gallery(value=images, visible=True), gr.Button(interactive=True)
 
 def run_generate(dicom_files):
     if not dicom_files:
         return "", gr.DownloadButton(visible=False)
     
-    images = load_dicom(dicom_files)
+    images = load_dicoms(dicom_files)
     if not images:
-        return gr.Gallery(visible=False), gr.Button(interactive=False)
-        
-    return gr.Gallery(value=images, visible=True), gr.Button(interactive=True)
+        return gr.DownloadButton(visible=False)
     
     report = generate_report(images)
     
@@ -58,7 +59,7 @@ with gr.Blocks(title="MRI Brain Report Generator") as app:
                 label="Scan Preview",
                 columns=3,
                 height="auto",
-                visible=True
+                visible=False
             )
         
         # Right column - report (empty for now)