Update app.py

app.py

@@ -18,7 +18,7 @@ OUTPUT_DIR = Path("outputs")
 OUTPUT_DIR.mkdir(exist_ok=True)
 
 # Fixed subject ID — no user input
-SUBJECT_ID = "sub-01"
+SUBJECT_ID = "sub"
 
 def preprocess_eeg(
     eeg_file,
@@ -39,9 +39,23 @@ def preprocess_eeg(
     subject_dir.mkdir(parents=True)
     (subject_dir / "plots").mkdir(exist_ok=True)
 
-    # Copy EEG file
-    eeg_path = subject_dir / Path(eeg_file.name).name
-    shutil.copy(eeg_file.name, eeg_path)
+    uploaded_file = Path(eeg_file.name)
+
+    if input_format == "mff":
+        # Validate: must be a ZIP file
+        if not zipfile.is_zipfile(uploaded_file):
+            raise ValueError("MFF input must be a ZIP file containing the .mff folder.")
+        # Extract directly into subject_dir as "sub.mff"
+        eeg_path = subject_dir / f"{SUBJECT_ID}.mff"
+        with zipfile.ZipFile(uploaded_file, 'r') as zf:
+            zf.extractall(eeg_path)
+        # Optional: validate
+        if not (eeg_path / "info.xml").exists():
+            raise ValueError("Invalid MFF: missing 'info.xml' in the ZIP.")
+    else:
+        # For FIF: copy the file into subject_dir with its original name
+        eeg_path = subject_dir / uploaded_file.name
+        shutil.copy(uploaded_file, eeg_path)
 
     try:
         # Run pipeline
@@ -73,7 +87,7 @@ def preprocess_eeg(
             f.write(f"=== EEG Info Summary for {SUBJECT_ID} ===\n")
             f.write(str(raw.info))
 
-        # Create ZIP
+        # Create ZIP of entire subject_dir
         zip_path = OUTPUT_DIR / f"{SUBJECT_ID}_preprocessing_output.zip"
         with zipfile.ZipFile(zip_path, "w", zipfile.ZIP_DEFLATED) as zf:
             for root, _, files in os.walk(subject_dir):