Update app.py

app.py

@@ -32,22 +32,25 @@ def visualize_slices(file_obj, slice_index):
     # Plot the slices
     fig, axes = plt.subplots(1, 3, figsize=(15, 5))
 
-    axes[0].imshow(slice_axial, cmap='gray')
+    # Axial view
+    axes[0].imshow(slice_axial, cmap='gray', aspect='auto')
     axes[0].set_title("Axial View")
     axes[0].axis('off')
 
-    axes[1].imshow(slice_coronal, cmap='gray')
+    # Coronal view
+    axes[1].imshow(slice_coronal, cmap='gray', aspect='auto')
     axes[1].set_title("Coronal View")
     axes[1].axis('off')
 
-    axes[2].imshow(slice_sagittal, cmap='gray')
+    # Sagittal view
+    axes[2].imshow(slice_sagittal.T, cmap='gray', aspect='auto')
     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')
+    fig.savefig(buf, format='png', dpi=100, bbox_inches='tight', pad_inches=0.1)
     plt.close(fig)
     buf.seek(0)
     pil_img = Image.open(buf)