Update app.py

app.py

@@ -1,17 +1,7 @@
 import streamlit as st
-import pydicom
-import matplotlib.pyplot as plt
-import zipfile
 import os
+import zipfile
 import subprocess
-from datetime import datetime
-import shutil
-import moviepy.video.io.ImageSequenceClip
-from io import BytesIO
-from tkinter import Tcl
-from PIL import Image
-
-subprocess_executed = False
 
 logo_image_path = '1.png'
 
@@ -51,17 +41,19 @@ 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"
+def resolve_path(base_path):
+    return os.path.abspath(os.path.join(base_path, "C2C/comp2comp/utils/../../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))]
+def fetch_outputs():
+    outputs_base_path = resolve_path("/home/user/app")
+    if os.path.exists(outputs_base_path):
+        subdirectories = [subdir for subdir in os.listdir(outputs_base_path)
+                          if os.path.isdir(os.path.join(outputs_base_path, subdir))]
 
         first_subdirectory = subdirectories[0] if subdirectories else None
 
         if first_subdirectory:
-            subdirectory_path = os.path.join(outputs_directory, first_subdirectory)
+            subdirectory_path = os.path.join(outputs_base_path, first_subdirectory)
             temp_dicom_dir_path = os.path.join(subdirectory_path, "temp_dicom_dir")
 
             for root, dirs, _ in os.walk(temp_dicom_dir_path):
@@ -89,7 +81,7 @@ def fetch_outputs():
         else:
             st.warning("Output files not found")
     else:
-        st.error("Output files not found")
+        st.error("Output directory not found")
 
 def main():
     st.write("Upload a ZIP file containing DICOM slices")