Update app.py

app.py

@@ -4,8 +4,7 @@ import matplotlib.pyplot as plt
 import os
 
 # Set up the Streamlit app title
-st.title("CBCT Scan DICOM Viewer")
-
+st.title("DICOM Viewer for CBCT Scans")
 
 # Directory paths where DICOM files are stored
 single_image_directory = "00002067/"
@@ -23,16 +22,20 @@ def get_slice_location(dicom_file):
     else:
         return 0  # Default to 0 if neither tag is available
 
+# Debug: List contents of the directories
+st.write("Contents of directory 00002067:", os.listdir(single_image_directory))
+st.write("Contents of directory 00007FA6:", os.listdir(multiple_images_directory))
+
 # Display the single DICOM image from the first folder
-st.header("Single DICOM Image from 00002067")
+st.header("Single DICOM Image")
 
 # Check if the directory exists
 if os.path.exists(single_image_directory):
-    # Get the only DICOM file in the single_image_directory
-    single_image_files = [os.path.join(single_image_directory, f) for f in os.listdir(single_image_directory) if f.endswith('.dcm')]
+    # Get all files in the single_image_directory (no extension filtering)
+    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:
-        # Read and display the single DICOM image
+        # Read and display the first DICOM image found
         ds = pydicom.dcmread(single_image_files[0])
         single_image = ds.pixel_array
 
@@ -47,12 +50,12 @@ else:
     st.write("Directory 00002067 does not exist.")
 
 # Display the multiple DICOM images from the second folder using a slider
-st.header("Browse Multiple DICOM Images from 00007FA6")
+st.header("Browse Multiple DICOM Images")
 
 # Check if the directory exists
 if os.path.exists(multiple_images_directory):
-    # Get list of DICOM files in the multiple_images_directory
-    multiple_image_files = [os.path.join(multiple_images_directory, f) for f in os.listdir(multiple_images_directory) if f.endswith('.dcm')]
+    # Get all files in the multiple_images_directory (no extension filtering)
+    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:
         # Sort DICOM files by slice location or image position