Update app.py

app.py

@@ -15,24 +15,11 @@ 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"])
+st.write("Upload a ZIP file containing DICOM slices to view the images.")
 
-# 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
-
-
+@st.cache_resource
+def install_dependencies():
+    if not install_script_executed:
         command = "chmod +x install.sh"
         try:
             subprocess.run(command, shell=True, check=True)
@@ -44,8 +31,25 @@ if uploaded_zip_file is not None:
         try:
             subprocess.run(command, shell=True, check=True)
             print("Script 'install.sh' has started running.")
+            install_script_executed = True
         except subprocess.CalledProcessError as e:
             print(f"Error while running the script: {e}")
+
+# Function to read and display the DICOM image
+@st.cache_resource
+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:
+        install_dependencies()
             
         temp_dir = "/home/user/app/C2C/temp_dicom_dir"
         
@@ -70,9 +74,11 @@ if uploaded_zip_file is not None:
             # Display the DICOM image using the cached function
             plt = display_dicom_image(selected_slice, dicom_files)
             st.pyplot(plt)
+            
+    except Exception as e:
+        st.error(f"Error: {str(e)}")
 
-        if not subprocess_executed:
-
+if st.button("Analyze"):
             command = "chmod +x inference.sh"
     
             try:
@@ -124,12 +130,6 @@ if uploaded_zip_file is not None:
             else:
                 st.warning("No subdirectories or folders found inside the 'outputs' directory.")
 
-
-
-
-    except Exception as e:
-        st.error(f"Error: {str(e)}")
     # finally:
     #     shutil.rmtree(temp_dir)
 
-st.write("Upload a ZIP file containing DICOM slices to view the images.")