Add application file

app.py

@@ -7,54 +7,46 @@ import os
 # Streamlit app title
 st.title("DICOM Image Viewer")
 
-# Upload a ZIP file containing DICOMs
-uploaded_zip = st.file_uploader("Upload a ZIP file containing DICOMs", type=["zip"])
-
-# Initialize variables for DICOM data and image list
-dicom_images = []
-num_slices = 0
-error_message = ""
-
-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")
-
-    # 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)
-
-    # Initialize a list to store successfully processed DICOMs
-    processed_dicoms = []
-
-    # Process DICOM files and skip problematic ones
-    for dicom_file in dicom_files:
-        try:
-            dicom_data = pydicom.dcmread(f"temp_zip_folder/{dicom_file}")
-            dicom_images.append(dicom_data.pixel_array)
-            processed_dicoms.append(dicom_data)
-        except Exception as e:
-            error_message += f"Skipping problematic DICOM ({dicom_file}): {str(e)}\n"
-
-    # User selects the slice using a slider
-    selected_slice = st.slider("Select a DICOM slice", 0, len(processed_dicoms) - 1)
-
-    if selected_slice < len(processed_dicoms):
-        # 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}/{len(processed_dicoms)}")
-        plt.tight_layout()
-
-        # Show the image in the Streamlit app
-        st.pyplot(plt)
-    else:
-        error_message += "No DICOM slices available for the selected slice number."
-
-    # Remove the temporary folder and its contents
-    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.")
-st.write(error_message)
+# Upload a ZIP file containing DICOM slices
+uploaded_zip_file = st.file_uploader("Upload a ZIP file containing DICOM slices", type=["zip"])
+
+if uploaded_zip_file is not None:
+    try:
+        # Create a temporary directory to unzip the files
+        temp_dir = "temp_dicom_dir"
+        os.makedirs(temp_dir, exist_ok=True)
+
+        with zipfile.ZipFile(uploaded_zip_file, "r") as zip_ref:
+            zip_ref.extractall(temp_dir)
+
+        # Get a list of DICOM files in the directory
+        dicom_files = [os.path.join(temp_dir, f) for f in os.listdir(temp_dir) if f.endswith(".dcm")]
+        dicom_files.sort()  # Sort the files
+
+        if len(dicom_files) == 0:
+            st.error("No DICOM files found in the ZIP archive.")
+        else:
+            # Display a slider for selecting the slice
+            selected_slice = st.slider("Select a slice", 0, len(dicom_files) - 1, 0)
+
+            # Read the selected DICOM file
+            dicom_data = pydicom.dcmread(dicom_files[selected_slice])
+
+            # Display the DICOM image
+            plt.imshow(dicom_data.pixel_array, cmap=plt.cm.bone)
+            plt.axis("off")
+            plt.title("DICOM Image")
+            plt.tight_layout()
+
+            # Show the image in the Streamlit app
+            st.pyplot(plt)
+
+    except Exception as e:
+        st.error(f"Error: {str(e)}")
+    finally:
+        # Clean up by removing the temporary directory
+        for file in dicom_files:
+            os.remove(file)
+        os.rmdir(temp_dir)
+
+st.write("Upload a ZIP file containing DICOM slices to view the images.")