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()
 
@@ -66,19 +66,21 @@ def extract_middle_slices(nifti_path, output_image_path, slice_size=180):
     # 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
+    coronal_slice = np.rot90(coronal_slice, k=2)  # Additional 180 degrees
     sagittal_slice = np.rot90(sagittal_slice, k=1)  # 90 degrees anticlockwise
+    sagittal_slice = np.rot90(sagittal_slice, k=2)  # Additional 180 degrees
 
     # Create subplots
     fig, axes = plt.subplots(1, 3, figsize=(12, 4))
 
     # Plot each rotated slice
-    axes[0].imshow(axial_slice, cmap="gray", origin="lower")
+    axes[0].imshow(axial_slice, cmap="gray", origin="lower", aspect='auto')
     axes[0].axis("off")
 
-    axes[1].imshow(coronal_slice, cmap="gray", origin="lower")
+    axes[1].imshow(coronal_slice, cmap="gray", origin="lower", aspect='auto')
     axes[1].axis("off")
 
-    axes[2].imshow(sagittal_slice, cmap="gray", origin="lower")
+    axes[2].imshow(sagittal_slice, cmap="gray", origin="lower", aspect='auto')
     axes[2].axis("off")
 
     # Save the figure