AdritRao commited on
Commit
c5601af
·
verified ·
1 Parent(s): 356b9e4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -18
app.py CHANGED
@@ -1,17 +1,7 @@
1
  import streamlit as st
2
- import pydicom
3
- import matplotlib.pyplot as plt
4
- import zipfile
5
  import os
 
6
  import subprocess
7
- from datetime import datetime
8
- import shutil
9
- import moviepy.video.io.ImageSequenceClip
10
- from io import BytesIO
11
- from tkinter import Tcl
12
- from PIL import Image
13
-
14
- subprocess_executed = False
15
 
16
  logo_image_path = '1.png'
17
 
@@ -51,17 +41,19 @@ def search_file(start_path, target_file):
51
  return os.path.join(root, target_file)
52
  return None
53
 
54
- def fetch_outputs():
55
- outputs_directory = "/home/user/app/C2C/outputs"
56
 
57
- if os.path.exists(outputs_directory):
58
- subdirectories = [subdir for subdir in os.listdir(outputs_directory)
59
- if os.path.isdir(os.path.join(outputs_directory, subdir))]
 
 
60
 
61
  first_subdirectory = subdirectories[0] if subdirectories else None
62
 
63
  if first_subdirectory:
64
- subdirectory_path = os.path.join(outputs_directory, first_subdirectory)
65
  temp_dicom_dir_path = os.path.join(subdirectory_path, "temp_dicom_dir")
66
 
67
  for root, dirs, _ in os.walk(temp_dicom_dir_path):
@@ -89,7 +81,7 @@ def fetch_outputs():
89
  else:
90
  st.warning("Output files not found")
91
  else:
92
- st.error("Output files not found")
93
 
94
  def main():
95
  st.write("Upload a ZIP file containing DICOM slices")
 
1
  import streamlit as st
 
 
 
2
  import os
3
+ import zipfile
4
  import subprocess
 
 
 
 
 
 
 
 
5
 
6
  logo_image_path = '1.png'
7
 
 
41
  return os.path.join(root, target_file)
42
  return None
43
 
44
+ def resolve_path(base_path):
45
+ return os.path.abspath(os.path.join(base_path, "C2C/comp2comp/utils/../../outputs"))
46
 
47
+ def fetch_outputs():
48
+ outputs_base_path = resolve_path("/home/user/app")
49
+ if os.path.exists(outputs_base_path):
50
+ subdirectories = [subdir for subdir in os.listdir(outputs_base_path)
51
+ if os.path.isdir(os.path.join(outputs_base_path, subdir))]
52
 
53
  first_subdirectory = subdirectories[0] if subdirectories else None
54
 
55
  if first_subdirectory:
56
+ subdirectory_path = os.path.join(outputs_base_path, first_subdirectory)
57
  temp_dicom_dir_path = os.path.join(subdirectory_path, "temp_dicom_dir")
58
 
59
  for root, dirs, _ in os.walk(temp_dicom_dir_path):
 
81
  else:
82
  st.warning("Output files not found")
83
  else:
84
+ st.error("Output directory not found")
85
 
86
  def main():
87
  st.write("Upload a ZIP file containing DICOM slices")