Update app.py

app.py

@@ -51,6 +51,51 @@ def search_file(start_path, target_file):
             return os.path.join(root, target_file)
     return None
 
+def fetch_outputs():
+    outputs_directory = "/home/user/app/C2C/outputs"
+
+    if os.path.exists(outputs_directory):
+        subdirectories = [subdir for subdir in os.listdir(outputs_directory)
+                          if os.path.isdir(os.path.join(outputs_directory, subdir))]
+
+        first_subdirectory = subdirectories[0] if subdirectories else None
+
+        if first_subdirectory:
+            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")
+        else:
+            st.warning("Output files not found")
+    else:
+        st.error("Output files not found")
+
 def main():
     st.write("Upload a ZIP file containing DICOM slices")
     uploaded_zip_file = st.file_uploader("Upload a .zip file", type=["zip"])
@@ -71,49 +116,8 @@ def main():
                 st.success("Analysis started (expected time: 5 mins)")
                 run_inference()
 
-                outputs_directory = "/home/user/app/C2C/outputs"
-
-                if os.path.exists(outputs_directory):
-                    subdirectories = [subdir for subdir in os.listdir(outputs_directory)
-                                      if os.path.isdir(os.path.join(outputs_directory, subdir))]
-
-                    first_subdirectory = subdirectories[0] if subdirectories else None
-
-                    if first_subdirectory:
-                        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(temp_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")
-                    else:
-                        st.warning("Output files not found")
-                else:
-                    st.error("Output files not found")
+            if st.button("Fetch Outputs"):
+                fetch_outputs()
         except Exception as e:
             st.error(f"Error: {str(e)}")