Update app.py

app.py

@@ -72,54 +72,56 @@ if uploaded_zip_file is not None:
             st.pyplot(plt)
 
 if st.button("Analyze"):
-            command = "chmod +x inference.sh"
+    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}")
+    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
+    command = "./inference.sh"
 
-            # Assuming the 'outputs' directory is located at "/home/user/app/C2C/comp2comp/outputs"
-            outputs_directory = "/home/user/app/C2C/outputs"
-            
-            # Get a list of subdirectories inside the 'outputs' directory
-            subdirectories = [subdir for subdir in os.listdir(outputs_directory)
-                  if os.path.isdir(os.path.join(outputs_directory, subdir))]
+    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}")
 
-            # Use the first (or only) subdirectory, or None if there are no subdirectories
-            first_subdirectory = subdirectories[0] if subdirectories else None
-            
-            if first_subdirectory:
-                # Construct the paths to the video and image files inside the first subdirectory
-                video_path = os.path.join(outputs_directory, first_subdirectory, "temp_dicom_dir/images/summary/aaa.mp4")
-                image_path = os.path.join(outputs_directory, first_subdirectory, "temp_dicom_dir/images/summary/diameter_graph.png")
-                largest_slice = os.path.join(outputs_directory, first_subdirectory, "temp_dicom_dir/images/summary/out.png")
-            
-                if os.path.exists(largest_slice):
-                    st.title("Largest Slice")
-                    st.image(largest_slice, use_column_width=True)
-                
-                # Display the video generated by the inference script
-                if os.path.exists(video_path):
-                    st.title("Video")
-                    st.video(video_path, format="video/mp4")
-            
-                # Display the image generated by the inference script
-                if os.path.exists(image_path):
-                    st.title("Diameter Graph")
-                    st.image(image_path, use_column_width=True)
+    # Set the flag to indicate that subprocess commands have been executed
+    subprocess_executed = True
+
+    # Assuming the 'outputs' directory is located at "/home/user/app/C2C/comp2comp/outputs"
+    outputs_directory = "/home/user/app/C2C/outputs"
+    
+    # Get a list of subdirectories inside the 'outputs' directory
+    subdirectories = [subdir for subdir in os.listdir(outputs_directory)
+          if os.path.isdir(os.path.join(outputs_directory, subdir))]
+
+    # Use the first (or only) subdirectory, or None if there are no subdirectories
+    first_subdirectory = subdirectories[0] if subdirectories else None
+    
+    if first_subdirectory:
+        # Construct the paths to the video and image files inside the first subdirectory
+        video_path = os.path.join(outputs_directory, first_subdirectory, "temp_dicom_dir/images/summary/aaa.mp4")
+        image_path = os.path.join(outputs_directory, first_subdirectory, "temp_dicom_dir/images/summary/diameter_graph.png")
+        largest_slice = os.path.join(outputs_directory, first_subdirectory, "temp_dicom_dir/images/summary/out.png")
+    
+        if os.path.exists(largest_slice):
+            st.title("Largest Slice")
+            st.image(largest_slice, use_column_width=True)
+        
+        # Display the video generated by the inference script
+        if os.path exists(video_path):
+            st.title("Video")
+            st.video(video_path, format="video/mp4")
+        
+        # Display the image generated by the inference script
+        if os.path.exists(image_path):
+            st.title("Diameter Graph")
+            st.image(image_path, use_column_width=True)
+    except Exception as e:
+        st.error(f"Error: {str(e)}")