Update app.py

app.py

@@ -513,6 +513,29 @@ def _handle_drop(files_list, current_state):
 def clear_files():
     return [], "0 file(s) selected", []
 
+import random
+
+def load_sample_images():
+    sample_dir = Path("sample_images")
+
+    images = [
+        str(p)
+        for p in sample_dir.glob("*")
+        if p.suffix.lower() in [".png", ".jpg", ".jpeg"]
+    ]
+
+    if not images:
+        raise gr.Error("No sample MRI images found in the sample_images folder.")
+
+    # Randomly choose up to 4 images
+    selected = random.sample(images, min(4, len(images)))
+
+    return (
+        selected,
+        f"{len(selected)} sample MRI(s) loaded",
+        _gallery_from_paths(selected),
+    )
+
 # ==============================
 # UI wiring
 # ==============================
@@ -612,6 +635,13 @@ def build_app():
                             value=True,
                             label="Compact results (show only top label + confidence)"
                         )
+
+                        sample_btn = gr.Button(
+                            "🧪 Load Sample MRIs",
+                            variant="secondary",
+                            size="lg"
+                        )
+                        
                     with gr.Column():
                         # Multi-upload UX: button + drag-and-drop zone (both append)
                         files_state = gr.State([])
@@ -682,6 +712,12 @@ def build_app():
             outputs=[files_state, files_label, gallery, files_dnd]
         )
 
+        sample_btn.click(
+            fn=load_sample_images,
+            inputs=None,
+            outputs=[files_state, files_label, gallery]
+        )
+        
         clear_btn.click(
             fn=clear_files,
             inputs=None,