Update app.py

app.py

@@ -6,8 +6,18 @@ import os
 # Set up the Streamlit app title
 st.title("DICOM Viewer for CBCT Scans")
 
-# Directory where DICOM files are stored in the repository
-dicom_directory = "./dicom_files/"
+# 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]
 
 # Function to get the slice location or image position for sorting
 def get_slice_location(dicom_file):
@@ -21,7 +31,7 @@ def get_slice_location(dicom_file):
     else:
         return 0  # Default to 0 if neither tag is available
 
-# Read all DICOM files from the directory
+# 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')]
 
 if dicom_files:
@@ -35,11 +45,4 @@ if dicom_files:
     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)}")
-    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. Please check the folder and try again.")
+    #