Update app.py

app.py

@@ -15,7 +15,6 @@ subprocess_executed = False
 
 logo_image_path = '1.png'
 
-
 st.image(logo_image_path, width=150)
 st.title("Automated Abdominal Aortic Aneurysm Detection")
 
@@ -49,6 +48,13 @@ def zip_and_download(directory, zip_filename):
         bytes_data = f.read()
     st.download_button(label="Download Outputs", data=bytes_data, file_name=zip_filename, mime="application/zip")
 
+# Function to search for files in all subdirectories
+def search_file(start_path, target_file):
+    for root, _, files in os.walk(start_path):
+        if target_file in files:
+            return os.path.join(root, target_file)
+    return None
+
 # Main Streamlit code
 def main():
     st.write("Upload a ZIP file containing DICOM slices")
@@ -82,11 +88,15 @@ def main():
                         subdirectory_path = os.path.join(outputs_directory, first_subdirectory)
                         temp_dicom_dir_path = os.path.join(subdirectory_path, "temp_dicom_dir")
 
-                        video_path = os.path.join(temp_dicom_dir_path, "images/summary/aaa.mp4")
-                        image_path = os.path.join(temp_dicom_dir_path, "images/summary/diameter_graph.png")
-                        largest_slice = os.path.join(temp_dicom_dir_path, "images/summary/out.png")
+                        video_path = search_file(temp_dicom_dir_path, "aaa.mp4")
+                        image_path = search_file(temp_dicom_dir_path, "diameter_graph.png")
+                        largest_slice = search_file(temp_dicom_dir_path, "out.png")
+
+                        st.write(f"Video Path: {video_path}")
+                        st.write(f"Image Path: {image_path}")
+                        st.write(f"Largest Slice Path: {largest_slice}")
 
-                        if os.path.exists(video_path) and os.path.exists(image_path) and os.path.exists(largest_slice):
+                        if video_path and image_path and largest_slice:
                             zip_filename = os.path.join(temp_dir, "outputs.zip")
                             zip_and_download(temp_dicom_dir_path, zip_filename)