Add application file

app.py

@@ -1,27 +1,49 @@
 import streamlit as st
 import pydicom
 import matplotlib.pyplot as plt
+import zipfile
+import io
 
 # Streamlit app title
 st.title("DICOM Image Viewer")
 
-# Upload a DICOM file
-uploaded_file = st.file_uploader("Upload a DICOM file", type=["dcm"])
+# Upload a ZIP file containing DICOMs
+uploaded_zip = st.file_uploader("Upload a ZIP file containing DICOMs", type=["zip"])
 
-if uploaded_file is not None:
-    try:
-        # Read the uploaded DICOM file
-        dicom_data = pydicom.dcmread(uploaded_file)
+# Initialize variables for DICOM data and image list
+dicom_images = []
+num_slices = 0
 
-        # Display the DICOM image
-        plt.imshow(dicom_data.pixel_array, cmap=plt.cm.bone)
-        plt.axis("off")
-        plt.title("DICOM Image")
-        plt.tight_layout()
+if uploaded_zip is not None:
+    # Unzip the uploaded ZIP file and extract DICOMs
+    with zipfile.ZipFile(uploaded_zip, "r") as zip_ref:
+        zip_ref.extractall("temp_zip_folder")
 
-        # Show the image in the Streamlit app
-        st.pyplot(plt)
-    except Exception as e:
-        st.error(f"Error: {str(e)}")
+    # Get a list of DICOM files
+    dicom_files = [f for f in os.listdir("temp_zip_folder") if f.lower().endswith(".dcm")]
+    num_slices = len(dicom_files)
 
-st.write("Upload a DICOM file to view the image.")
+    # Read and display DICOM images
+    for dicom_file in dicom_files:
+        dicom_data = pydicom.dcmread(f"temp_zip_folder/{dicom_file}")
+        dicom_images.append(dicom_data.pixel_array)
+
+    # User selects the slice using a slider
+    selected_slice = st.slider("Select a DICOM slice", 0, num_slices - 1)
+
+    # Display the selected DICOM image
+    plt.imshow(dicom_images[selected_slice], cmap=plt.cm.bone)
+    plt.axis("off")
+    plt.title(f"DICOM Image - Slice {selected_slice + 1}/{num_slices}")
+    plt.tight_layout()
+
+    # Show the image in the Streamlit app
+    st.pyplot(plt)
+
+# Remove the temporary folder and its contents
+if num_slices > 0:
+    for dicom_file in dicom_files:
+        os.remove(f"temp_zip_folder/{dicom_file}")
+    os.rmdir("temp_zip_folder")
+
+st.write("Upload a ZIP file containing DICOMs to view and scroll through the slices.")