Update app.py

app.py

@@ -6,18 +6,9 @@ import os
 # Set up the Streamlit app title
 st.title("DICOM Viewer for CBCT Scans")
 
-# Directories where DICOM files are stored
-dicom_directories = {
-    "00002067": "./00002067/",
-    "00007FA6": "./00007FA6/"
-}
-
-# Sidebar for selecting DICOM folder
-st.sidebar.header("Select DICOM Folder")
-selected_folder = st.sidebar.selectbox("Choose a folder to view DICOM files", list(dicom_directories.keys()))
-
-# Get the selected directory path
-dicom_directory = dicom_directories[selected_folder]
+# Directory paths where DICOM files are stored
+single_image_directory = "./00002067/"
+multiple_images_directory = "./00007FA6/"
 
 # Function to get the slice location or image position for sorting
 def get_slice_location(dicom_file):
@@ -31,18 +22,47 @@ def get_slice_location(dicom_file):
     else:
         return 0  # Default to 0 if neither tag is available
 
-# Get list of DICOM files in the selected directory
-dicom_files = [os.path.join(dicom_directory, f) for f in os.listdir(dicom_directory) if f.endswith('.dcm')]
+# Display the single DICOM image from the first folder
+st.header("Single DICOM Image:")
+
+# 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')]
+
+if single_image_files:
+    # Read and display the single DICOM image
+    ds = pydicom.dcmread(single_image_files[0])
+    single_image = ds.pixel_array
+
+    # Display the image
+    fig, ax = plt.subplots()
+    ax.imshow(single_image, cmap='gray')
+    ax.axis('off')
+    st.pyplot(fig)
+else:
+    st.write("No DICOM files found in the directory 00002067.")
+
+# Display the multiple DICOM images from the second folder using a slider
+st.header("Browse Multiple DICOM Images from 00007FA6")
+
+# 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')]
 
-if dicom_files:
+if multiple_image_files:
     # Sort DICOM files by slice location or image position
-    dicom_files.sort(key=lambda x: get_slice_location(x))
+    multiple_image_files.sort(key=lambda x: get_slice_location(x))
     
     # Read all DICOM images (sorted)
-    dicom_images = [pydicom.dcmread(file).pixel_array for file in dicom_files]
+    dicom_images = [pydicom.dcmread(file).pixel_array for file in multiple_image_files]
 
     # Interactive slider to select image index
     st.sidebar.header("Image Slider")
     image_index = st.sidebar.slider("Select Image Index", 0, len(dicom_images) - 1, 0)
 
-    #
+    # Display the selected DICOM image
+    st.header(f"Displaying Image {image_index + 1} of {len(dicom_images)} from folder 00007FA6")
+    fig, ax = plt.subplots()
+    ax.imshow(dicom_images[image_index], cmap='gray')
+    ax.axis('off')
+    st.pyplot(fig)
+else:
+    st.write("No DICOM files found in the directory 00007FA6.")