Update app.py

app.py

@@ -29,28 +29,25 @@ def visualize_slices(file_obj, slice_index):
     slice_coronal = data[:, slice_index_coronal, :]
     slice_sagittal = data[slice_index_sagittal, :, :]
 
-    # Plot the slices
+    # Plot the slices with the correct aspect ratios
     fig, axes = plt.subplots(1, 3, figsize=(15, 5))
 
-    # Axial view
     axes[0].imshow(slice_axial, cmap='gray', aspect='auto')
     axes[0].set_title("Axial View")
     axes[0].axis('off')
 
-    # Coronal view
-    axes[1].imshow(slice_coronal, cmap='gray', aspect='auto')
+    axes[1].imshow(slice_coronal.T, cmap='gray', aspect='auto', extent=[0, data.shape[0], 0, data.shape[2]])
     axes[1].set_title("Coronal View")
     axes[1].axis('off')
 
-    # Sagittal view
-    axes[2].imshow(slice_sagittal.T, cmap='gray', aspect='auto')
+    axes[2].imshow(slice_sagittal.T, cmap='gray', aspect='auto', extent=[0, data.shape[1], 0, data.shape[2]])
     axes[2].set_title("Sagittal View")
     axes[2].axis('off')
 
     # Convert matplotlib figure to PIL Image
     buf = io.BytesIO()
     plt.tight_layout()
-    fig.savefig(buf, format='png', dpi=100, bbox_inches='tight', pad_inches=0.1)
+    fig.savefig(buf, format='png')
     plt.close(fig)
     buf.seek(0)
     pil_img = Image.open(buf)