Update app.py

app.py

@@ -33,7 +33,7 @@ def extract_middle_slices(nifti_path, output_image_path, slice_size=180):
     Extracts slices centered around the center of mass of non-zero voxels in a 3D NIfTI image.
     The slices are taken along axial, coronal, and sagittal planes and saved as a single PNG.
     """
-    # Load NIfTI image and get the data
+     # Load NIfTI image and get the data
     img = nib.load(nifti_path)
     data = img.get_fdata()
 
@@ -63,21 +63,23 @@ def extract_middle_slices(nifti_path, output_image_path, slice_size=180):
     coronal_slice = extract_2d_slice(data, center, axis=1)  # Coronal (y-axis)
     sagittal_slice = extract_2d_slice(data, center, axis=0)  # Sagittal (x-axis)
 
+    # Apply rotations to each slice
+    axial_slice = np.rot90(axial_slice, k=-1)  # 90 degrees clockwise
+    coronal_slice = np.rot90(coronal_slice, k=1)  # 90 degrees anticlockwise
+    sagittal_slice = np.rot90(sagittal_slice, k=1)  # 90 degrees anticlockwise
+
     # Create subplots
     fig, axes = plt.subplots(1, 3, figsize=(12, 4))
 
-    # Plot each slice
+    # Plot each rotated slice
     axes[0].imshow(axial_slice, cmap="gray", origin="lower")
     axes[0].axis("off")
-    axes[0].set_title("Axial")
 
     axes[1].imshow(coronal_slice, cmap="gray", origin="lower")
     axes[1].axis("off")
-    axes[1].set_title("Coronal")
 
     axes[2].imshow(sagittal_slice, cmap="gray", origin="lower")
     axes[2].axis("off")
-    axes[2].set_title("Sagittal")
 
     # Save the figure
     plt.tight_layout()