Update app.py

app.py

@@ -34,6 +34,22 @@ def install_dependencies():
         except subprocess.CalledProcessError as e:
             print(f"Error while running the script: {e}")
 
+@st.cache_resource
+def run_inference():
+        command = "chmod +x inference.sh"
+        try:
+            subprocess.run(command, shell=True, check=True)
+            print("Script 'inference.sh' has been made executable.")
+        except subprocess.CalledProcessError as e:
+            print(f"Error while making the script executable: {e}")
+
+        command = "./inference.sh"
+        try:
+            subprocess.run(command, shell=True, check=True)
+            print("Script 'inference.sh' has started running.")
+        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):
@@ -78,24 +94,7 @@ if uploaded_zip_file is not None:
         st.error(f"Error: {str(e)}")
 
 if st.button("Analyze"):
-            command = "chmod +x inference.sh"
-    
-            try:
-                subprocess.run(command, shell=True, check=True)
-                print("Script 'inference.sh' has started running.")
-            except subprocess.CalledProcessError as e:
-                print(f"Error while making the script executable: {e}")
-    
-            command = "./inference.sh"
-    
-            try:
-                subprocess.run(command, shell=True, check=True)
-                print("Script 'inference.sh' has started running.")
-            except subprocess.CalledProcessError as e:
-                print(f"Error while making the script executable: {e}")
-
-            # Set the flag to indicate that subprocess commands have been executed
-            subprocess_executed = True
+            run_inference()
 
             # Assuming the 'outputs' directory is located at "/home/user/app/C2C/comp2comp/outputs"
             outputs_directory = "/home/user/app/C2C/outputs"
@@ -127,8 +126,4 @@ if st.button("Analyze"):
                     st.title("Diameter Graph")
                     st.image(image_path, use_column_width=True)
             else:
-                st.warning("No subdirectories or folders found inside the 'outputs' directory.")
-
-    # finally:
-    #     shutil.rmtree(temp_dir)
-
+                st.warning("No subdirectories or folders found inside the 'outputs' directory.")