Update app.py

app.py

@@ -64,33 +64,28 @@ def fetch_outputs():
             subdirectory_path = os.path.join(outputs_directory, first_subdirectory)
             temp_dicom_dir_path = os.path.join(subdirectory_path, "temp_dicom_dir")
 
-            dicom_subdirectories = [subdir for subdir in os.listdir(temp_dicom_dir_path)
-                                    if os.path.isdir(os.path.join(temp_dicom_dir_path, subdir))]
-
-            dicom_folder = dicom_subdirectories[0] if dicom_subdirectories else None
-
-            if dicom_folder:
-                dicom_folder_path = os.path.join(temp_dicom_dir_path, dicom_folder)
-                video_path = search_file(dicom_folder_path, "aaa.mp4")
-                image_path = search_file(dicom_folder_path, "diameter_graph.png")
-                largest_slice = search_file(dicom_folder_path, "out.png")
-
-                if video_path and image_path and largest_slice:
-                    zip_filename = os.path.join("/home/user/app/C2C/temp_dicom_dir", "outputs.zip")
-                    zip_and_download(dicom_folder_path, zip_filename)
-
-                    st.title("Largest Slice")
-                    st.image(largest_slice, use_column_width=True)
-
-                    st.title("Video")
-                    st.video(video_path, format="video/mp4")
-
-                    st.title("Diameter Graph")
-                    st.image(image_path, use_column_width=True)
-                else:
-                    st.error("Output files not found")
-            else:
-                st.error("DICOM folder not found")
+            for root, dirs, _ in os.walk(temp_dicom_dir_path):
+                for subdir in dirs:
+                    dicom_folder_path = os.path.join(root, subdir)
+                    video_path = search_file(dicom_folder_path, "aaa.mp4")
+                    if video_path:  # Use this folder if it contains the mp4 file
+                        image_path = search_file(dicom_folder_path, "diameter_graph.png")
+                        largest_slice = search_file(dicom_folder_path, "out.png")
+
+                        if video_path and image_path and largest_slice:
+                            zip_filename = os.path.join("/home/user/app/C2C/temp_dicom_dir", "outputs.zip")
+                            zip_and_download(dicom_folder_path, zip_filename)
+
+                            st.title("Largest Slice")
+                            st.image(largest_slice, use_column_width=True)
+
+                            st.title("Video")
+                            st.video(video_path, format="video/mp4")
+
+                            st.title("Diameter Graph")
+                            st.image(image_path, use_column_width=True)
+                            return
+            st.error("Output files not found in any subdirectory")
         else:
             st.warning("Output files not found")
     else: