Update utils.py

utils.py

@@ -66,13 +66,15 @@ def convert_dicom_zip_to_nifti(zip_file):
 
         try:
             # Load the 3D volume
-            img_nib = nib.load(nifti_path)
-            img_nib = nib.as_closest_canonical(img_nib)
-            volume = img_nib.get_fdata()
+            img = nib.as_closest_canonical(nifti_path)
+            volume = img.get_fdata()
 
             # Extract middle axial slice
             mid = volume.shape[2] // 2
-            slice_2d = volume[:, :, mid]
+            slice_2d = volume[:, :, mid
+
+            # Rotate to correct display orientation
+            slice_2d = np.rot90(slice_2d) 
 
             # Normalize to 0-255
             s_min, s_max = slice_2d.min(), slice_2d.max()
@@ -80,8 +82,6 @@ def convert_dicom_zip_to_nifti(zip_file):
                 continue
 
             normalized = (slice_2d - s_min) / (s_max - s_min) * 255
-            # Rotate to correct display orientation
-            slice_2d_rotated = np.rot90(normalized)
 
             image = Image.fromarray(
                 slice_2d_rotated.astype(np.uint8)