Update utils.py

utils.py

@@ -13,18 +13,6 @@ import zipfile
 import subprocess
 import nibabel as nib
 
-SKIP_KEYWORDS = ["localizer", "scout", "loc"]
-
-for nifti_file in sorted(nifti_files):
-    sequence_name = nifti_file.replace(".nii.gz", "")
-    
-    # Skip localizer images
-    if any(skip in sequence_name.lower() for skip in SKIP_KEYWORDS):
-        print(f"Skipping localizer: {sequence_name}")
-        continue
-        
-    nifti_path = os.path.join(nifti_dir, nifti_file)
-
 def convert_dicom_zip_to_nifti(zip_file):
     """
     Takes a uploaded ZIP file containing DICOM files.
@@ -32,6 +20,8 @@ def convert_dicom_zip_to_nifti(zip_file):
     Returns list of (sequence_name, PIL Image) tuples.
     """
 
+    SKIP_KEYWORDS = ["localizer", "scout", "loc"]
+
     # Step 1: Setup temp directories
     unique_id = uuid.uuid4().hex
     tmp_dir = f"/tmp/dicom_{unique_id}"
@@ -66,6 +56,12 @@ def convert_dicom_zip_to_nifti(zip_file):
 
     for nifti_file in sorted(nifti_files):
         sequence_name = nifti_file.replace(".nii.gz", "")
+        
+        # Skip localizer images
+        if any(skip in sequence_name.lower() for skip in SKIP_KEYWORDS):
+            print(f"Skipping localizer: {sequence_name}")
+            continue
+            
         nifti_path = os.path.join(nifti_dir, nifti_file)
 
         try: