Update app.py

app.py

@@ -28,10 +28,28 @@ def extract_slice(data, view, slice_index):
     elif view == 'Sagittal':
         return data[slice_index, :, :]
 
+def resize_slice(slice_image, view, data_shape):
+    if view == 'Axial':
+        return slice_image  # No resizing needed for Axial view
+    else:
+        # For Coronal and Sagittal views, find the two largest dimensions
+        if view == 'Coronal':
+            resize_dims = (data_shape[0], data_shape[2])  # (x, z)
+        elif view == 'Sagittal':
+            resize_dims = (data_shape[1], data_shape[2])  # (y, z)
+        
+        # Resize the slice image while maintaining the aspect ratio
+        resized_image = Image.fromarray(slice_image).resize(resize_dims, Image.ANTIALIAS)
+        return np.array(resized_image)
+
 def visualize_slice(file_obj, view, slice_index):
     data = load_nrrd(file_obj)
+    data_shape = data.shape  # Get the original x, y, z dimensions
     slice_image = extract_slice(data, view, slice_index)
-
+    
+    # Resize the slice image if necessary
+    slice_image = resize_slice(slice_image, view, data_shape)
+    
     # Plot the slice
     fig, ax = plt.subplots()
     ax.imshow(slice_image, cmap='gray')