Update app.py

app.py

@@ -22,41 +22,25 @@ def get_num_slices(data, view):
 
 def extract_slice(data, view, slice_index):
     if view == 'Axial':
-        return data[:, :, slice_index]
+        # Axial view: XY plane
+        slice_image = data[:, :, slice_index]
     elif view == 'Coronal':
-        return data[:, slice_index, :]
+        # Coronal view: XZ plane
+        slice_image = data[:, slice_index, :]
+        # Flip the image vertically for correct orientation
+        slice_image = np.flipud(slice_image).T
     elif view == 'Sagittal':
-        return data[slice_index, :, :]
-
-def resize_slice(slice_image, max_size=(500, 200)):
-    # Convert the NumPy array to a PIL image
-    slice_pil = Image.fromarray(slice_image)
-    
-    # Get original dimensions
-    original_width, original_height = slice_pil.size
-    
-    # Calculate aspect ratio
-    aspect_ratio = original_width / original_height
-    
-    # Determine new dimensions based on the aspect ratio
-    if original_width > original_height:
-        new_width = min(max_size[0], original_width)
-        new_height = int(new_width / aspect_ratio)
-    else:
-        new_height = min(max_size[1], original_height)
-        new_width = int(new_height * aspect_ratio)
+        # Sagittal view: YZ plane
+        slice_image = data[slice_index, :, :]
+        # Rotate the image for correct orientation
+        slice_image = np.rot90(slice_image, k=3)
     
-    # Resize the image
-    resized_image = slice_pil.resize((new_width, new_height), Image.ANTIALIAS)
-    return np.array(resized_image)
+    return slice_image
 
 def visualize_slice(file_obj, view, slice_index):
     data = load_nrrd(file_obj)
     slice_image = extract_slice(data, view, slice_index)
-    
-    # Resize the slice image while maintaining aspect ratio
-    slice_image = resize_slice(slice_image)
-    
+
     # Plot the slice
     fig, ax = plt.subplots()
     ax.imshow(slice_image, cmap='gray')