disable upload when requiring mask

app.py

@@ -54,10 +54,21 @@ HEAD_OPTIONS: List[Tuple[str, str]] = [
     ("luna16-3D", "LUNA16 3D"),
     ("neural_foraminal_narrowing", "Neural Foraminal Narrowing"),
     ("oasis", "OASIS"),
-    ("spinal_canal_stenosis", "Spinal Canal Stenosis"),
-    ("subarticular_stenosis", "Subarticular Stenosis"),
 ]
 
+# Heads that require masks - custom image upload will be disabled for these
+HEADS_REQUIRING_MASK: set[str] = {
+    "anatomy-ct",
+    "anatomy-mri",
+    "deep-lesion-site",
+    "emidec-classification-mask",
+    "kits",
+    "kneeMRI",
+    "luna16-3D",
+    "spinal_canal_stenosis",
+    "subarticular_stenosis",
+}
+
 
 DATASET_OPTIONS: Dict[str, Dict[str, Any]] = {
     "anatomy-ct": {"label": "Anatomy CT (test)", "head": "anatomy-ct"},
@@ -328,6 +339,16 @@ def update_dataset_from_head(head: str) -> Dict[str, Any]:
     return gr.update()
 
 
+def update_upload_component_state(head: str) -> Dict[str, Any]:
+    """Disable upload component for heads that require masks."""
+    if head in HEADS_REQUIRING_MASK:
+        return gr.update(
+            interactive=False,
+            info="Custom image upload is disabled for this task because it requires a mask from the dataset.",
+        )
+    return gr.update(interactive=True, info=None)
+
+
 def load_dataset_metadata(subset: str) -> Tuple[Dict[str, Any], str]:
     try:
         dataset = load_curia_dataset(subset)
@@ -524,6 +545,10 @@ def build_demo() -> gr.Blocks:
             fn=update_dataset_from_head,
             inputs=[head_dropdown],
             outputs=[dataset_dropdown],
+        ).then(
+            fn=update_upload_component_state,
+            inputs=[head_dropdown],
+            outputs=[upload_component],
         )
 
         dataset_dropdown.change(