Update app.py

app.py

@@ -3,6 +3,8 @@ import pydicom
 import matplotlib.pyplot as plt
 import os
 import time
+import numpy as np
+from scipy.ndimage import rotate
 
 # Sidebar navigation
 page = st.sidebar.radio("Select page", ("New DICOM Viewer (October 2024)", "Old DICOM Viewer (May 2024)"))
@@ -102,7 +104,7 @@ if page == "Old DICOM Viewer (May 2024)":
         st.write("Directory 00007FA6 does not exist.")
 
 elif page == "New DICOM Viewer (October 2024)":
-    st.title("DICOM Viewer for 20241015")
+    st.title("CBCT Viewer for 2024-10-15")
     new_data_directory = "20241015/"
     if os.path.exists(new_data_directory):
         new_image_files = [
@@ -111,12 +113,31 @@ elif page == "New DICOM Viewer (October 2024)":
             if os.path.isfile(os.path.join(new_data_directory, f)) and f.endswith('.dcm')
         ]
         if new_image_files:
-            # Sort files based on the numeric part of the filename (e.g., "0000.dcm", "0001.dcm", etc.)
+            # Sort files based on the numeric part of the filename
             new_image_files.sort(key=lambda x: int(os.path.splitext(os.path.basename(x))[0]))
             dicom_images = [pydicom.dcmread(file).pixel_array for file in new_image_files]
-            image_index = st.slider("Select Image Index", 0, len(dicom_images) - 1, 0)
+            volume = np.stack(dicom_images, axis=0)  # Create a 3D volume from the stack
+
+            view_type = st.selectbox("Select View Type", ("Axial", "Sagittal", "Coronal", "Oblique"))
+            if view_type == "Axial":
+                slice_index = st.slider("Select Axial Slice", 0, volume.shape[0]-1, volume.shape[0]//2)
+                slice_img = volume[slice_index, :, :]
+            elif view_type == "Sagittal":
+                col_index = st.slider("Select Sagittal Column", 0, volume.shape[2]-1, volume.shape[2]//2)
+                slice_img = volume[:, :, col_index]
+                slice_img = np.flipud(slice_img.T)
+            elif view_type == "Coronal":
+                row_index = st.slider("Select Coronal Row", 0, volume.shape[1]-1, volume.shape[1]//2)
+                slice_img = volume[:, row_index, :]
+                slice_img = np.flipud(slice_img.T)
+            elif view_type == "Oblique":
+                angle = st.slider("Select Oblique Angle", -45, 45, 0)
+                slice_index = st.slider("Select Slice Index for Oblique View", 0, volume.shape[0]-1, volume.shape[0]//2)
+                # For a simplified oblique view, rotate the selected axial slice
+                slice_axial = volume[slice_index, :, :]
+                slice_img = rotate(slice_axial, angle, reshape=False)
             fig, ax = plt.subplots()
-            ax.imshow(dicom_images[image_index], cmap='gray')
+            ax.imshow(slice_img, cmap='gray')
             ax.axis('off')
             st.pyplot(fig)
         else: