Update app.py

app.py

@@ -95,22 +95,19 @@ if uploaded_zip_file is not None:
             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)]
+            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]
+            # 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")
             
                 # Display the video generated by the inference script
-                if os.path.exists(video_path):
+                if os.path exists(video_path):
                     st.video(video_path, format="video/mp4")
             
                 # Display the image generated by the inference script
@@ -120,6 +117,7 @@ if uploaded_zip_file is not None:
                 st.warning("No subdirectories or folders found inside the 'outputs' directory.")
 
 
+
     except Exception as e:
         st.error(f"Error: {str(e)}")
     finally: