Update app.py

app.py

@@ -73,54 +73,58 @@ if st.button("Analyze"):
     except subprocess.CalledProcessError as e:
         print(f"Error while running the script: {e}")
 
-    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
 
-    # 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))]
+    if subprocess_executed:
 
-    # Use the first (or only) subdirectory, or None if there are no subdirectories
-    first_subdirectory = subdirectories[0] if subdirectories else None
+        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}")
     
-    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")
+        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
+    
+        # 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))]
     
-        if os.path.exists(largest_slice):
-            st.title("Largest Slice")
-            st.image(largest_slice, use_column_width=True)
+        # Use the first (or only) subdirectory, or None if there are no subdirectories
+        first_subdirectory = subdirectories[0] if subdirectories else None
         
-        # Display the video generated by the inference script
-        if os.path.exists(video_path):
-            st.title("Video")
-            st.video(video_path, format="video/mp4")
+        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")
         
-        # 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)
+            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)