Add application file

app.py

@@ -10,6 +10,18 @@ st.title("DICOM Image Viewer")
 # Upload a ZIP file containing DICOM slices
 uploaded_zip_file = st.file_uploader("Upload a ZIP file containing DICOM slices", type=["zip"])
 
+# Function to read and display the DICOM image
+@st.cache(suppress_st_warning=True, show_spinner=False)
+def display_dicom_image(selected_slice, dicom_files):
+    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()
+    return plt
+
 if uploaded_zip_file is not None:
     try:
         # Create a temporary directory to unzip the files
@@ -29,16 +41,8 @@ if uploaded_zip_file is not None:
             # 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
+            # Display the DICOM image using the cached function
+            plt = display_dicom_image(selected_slice, dicom_files)
             st.pyplot(plt)
 
     except Exception as e: