Update app.py

app.py

@@ -23,11 +23,9 @@ if page == "Old DICOM Viewer (May 2024)":
 
     st.header("Single DICOM Image from folder 00002067")
     if os.path.exists(single_image_directory):
-        single_image_files = [
-            os.path.join(single_image_directory, f)
-            for f in os.listdir(single_image_directory)
-            if os.path.isfile(os.path.join(single_image_directory, f))
-        ]
+        single_image_files = [os.path.join(single_image_directory, f)
+                              for f in os.listdir(single_image_directory)
+                              if os.path.isfile(os.path.join(single_image_directory, f))]
         if single_image_files:
             ds = pydicom.dcmread(single_image_files[0])
             single_image = ds.pixel_array
@@ -42,16 +40,14 @@ if page == "Old DICOM Viewer (May 2024)":
 
     st.header("Browse Multiple DICOM Images from 00007FA6")
     if os.path.exists(multiple_images_directory):
-        multiple_image_files = [
-            os.path.join(multiple_images_directory, f)
-            for f in os.listdir(multiple_images_directory)
-            if os.path.isfile(os.path.join(multiple_images_directory, f))
-        ]
+        multiple_image_files = [os.path.join(multiple_images_directory, f)
+                                 for f in os.listdir(multiple_images_directory)
+                                 if os.path.isfile(os.path.join(multiple_images_directory, f))]
         if multiple_image_files:
             multiple_image_files.sort(key=lambda x: get_slice_location(x))
             dicom_images = [pydicom.dcmread(file).pixel_array for file in multiple_image_files]
             st.subheader("Display with Image Slider:")
-            image_index = st.slider("Select Image Index", 0, len(dicom_images)-1, 0)
+            image_index = st.slider("Select Image Index", 0, len(dicom_images) - 1, 0)
             fig, ax = plt.subplots()
             ax.imshow(dicom_images[image_index], cmap='gray')
             ax.axis('off')
@@ -65,11 +61,9 @@ elif page == "New DICOM Viewer (October 2024)":
     st.title("CBCT Viewer for 2024-10-15")
     new_data_directory = "20241015/"
     if os.path.exists(new_data_directory):
-        new_image_files = [
-            os.path.join(new_data_directory, f)
-            for f in os.listdir(new_data_directory)
-            if os.path.isfile(os.path.join(new_data_directory, f)) and f.endswith('.dcm')
-        ]
+        new_image_files = [os.path.join(new_data_directory, f)
+                           for f in os.listdir(new_data_directory)
+                           if os.path.isfile(os.path.join(new_data_directory, f)) and f.endswith('.dcm')]
         if new_image_files:
             new_image_files.sort(key=lambda x: int(os.path.splitext(os.path.basename(x))[0]))
             dicom_datasets = [pydicom.dcmread(file) for file in new_image_files]
@@ -77,7 +71,7 @@ elif page == "New DICOM Viewer (October 2024)":
             volume = np.stack(image_arrays, axis=0)  # (num_slices, height, width)
 
             st.subheader("Axial View")
-            slice_index = st.slider("Select Axial Slice", 0, volume.shape[0]-1, volume.shape[0]//2)
+            slice_index = st.slider("Select Axial Slice", 0, volume.shape[0] - 1, volume.shape[0] // 2)
             slice_img = volume[slice_index, :, :]
             fig, ax = plt.subplots()
             ax.imshow(slice_img, cmap='gray')
@@ -89,21 +83,6 @@ elif page == "New DICOM Viewer (October 2024)":
         st.write("Directory 20241015 does not exist.")
 
     st.subheader("Videos")
-    # Use an absolute path relative to the location of this script
-    file_dir = os.path.dirname(os.path.abspath(__file__))
-    sagittal_path = os.path.join(file_dir, "..", "CBCT_2024_sagittal.mov")
-    coronal_path = os.path.join(file_dir, "..", "CBCT_2024_coronal.mov")
-
-    if os.path.exists(sagittal_path):
-        with open(sagittal_path, "rb") as f:
-            sagittal_bytes = f.read()
-        st.video(sagittal_bytes)
-    else:
-        st.error(f"Video file not found: {sagittal_path}")
-
-    if os.path.exists(coronal_path):
-        with open(coronal_path, "rb") as f:
-            coronal_bytes = f.read()
-        st.video(coronal_bytes)
-    else:
-        st.error(f"Video file not found: {coronal_path}")
+    # Use the MP4 versions of the videos for better browser compatibility.
+    st.video("CBCT_2024_sagittal.mp4")
+    st.video("CBCT_2024_coronal.mp4")