AdritRao commited on
Commit
ff4d8cd
·
verified ·
1 Parent(s): 94985b8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -0
app.py CHANGED
@@ -95,8 +95,21 @@ if st.button("Analyze"):
95
  st.success("Analysis started (expected time: 5 mins)")
96
  run_inference()
97
 
 
98
  outputs_directory = "/home/user/app/C2C/outputs"
99
 
 
 
 
 
 
 
 
 
 
 
 
 
100
  subdirectories = [subdir for subdir in os.listdir(outputs_directory)
101
  if os.path.isdir(os.path.join(outputs_directory, subdir))]
102
 
 
95
  st.success("Analysis started (expected time: 5 mins)")
96
  run_inference()
97
 
98
+
99
  outputs_directory = "/home/user/app/C2C/outputs"
100
 
101
+ def clear_directory(directory):
102
+ if os.path.exists(directory):
103
+ for item_name in os.listdir(directory):
104
+ item_path = os.path.join(directory, item_name)
105
+ if os.path.isfile(item_path):
106
+ os.remove(item_path)
107
+ elif os.path.isdir(item_path):
108
+ shutil.rmtree(item_path)
109
+ clear_directory(outputs_directory)
110
+
111
+
112
+
113
  subdirectories = [subdir for subdir in os.listdir(outputs_directory)
114
  if os.path.isdir(os.path.join(outputs_directory, subdir))]
115