Update app.py

app.py

@@ -91,17 +91,34 @@ if uploaded_zip_file is not None:
             # Set the flag to indicate that subprocess commands have been executed
             subprocess_executed = True
 
-            current_datetime = datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
-            video_path = f"/home/user/app/C2C/comp2comp/outputs/{current_datetime}/temp_dicom_dir/images/summary/aaa.mp4"
-            image_path = f"/home/user/app/C2C/comp2comp/outputs/{current_datetime}/temp_dicom_dir/images/summary/diameter_graph.png"
-        
-            # Display the video generated by the inference script
-            if os.path.exists(video_path):
-                st.video(video_path, format="video/mp4")
-        
-            # Display the image generated by the inference script
-            if os.path.exists(image_path):
-                st.image(image_path, caption="Diameter Graph", use_column_width=True)
+            # Assuming the 'outputs' directory is located at "/home/user/app/C2C/comp2comp/outputs"
+            outputs_directory = "/home/user/app/C2C/comp2comp/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 not subdirectories:
+                # If there are no subdirectories, use the only folder inside the 'outputs' directory
+                subdirectories = [d for d in os.listdir(outputs_directory) if os.path.isdir(os.path.join(outputs_directory, d))]
+            
+            if subdirectories:
+                # Get the first (or only) subdirectory
+                first_subdirectory = subdirectories[0]
+            
+                # 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")
+            
+                # Display the video generated by the inference script
+                if os.path.exists(video_path):
+                    st.video(video_path, format="video/mp4")
+            
+                # Display the image generated by the inference script
+                if os.path.exists(image_path):
+                    st.image(image_path, caption="Diameter Graph", use_column_width=True)
+            else:
+                st.warning("No subdirectories or folders found inside the 'outputs' directory.")
+
 
     except Exception as e:
         st.error(f"Error: {str(e)}")