update

app.py

@@ -39,6 +39,17 @@ def zip_result_files(result_folder):
     
     return zip_buffer
 
+def clear_folder(folder_path):
+    for filename in os.listdir(folder_path):
+        file_path = os.path.join(folder_path, filename)
+        try:
+            if os.path.isfile(file_path) or os.path.islink(file_path):
+                os.unlink(file_path)  # Remove file or symlink
+            elif os.path.isdir(file_path):
+                shutil.rmtree(file_path, ignore_errors=True)  # Remove directory and its contents
+        except Exception as e:
+            print(f'Failed to delete {file_path}. Reason: {e}')
+
 st.title("Text Detection App")
 
 uploaded_file = st.file_uploader("Choose an image file", type=["jpg", "jpeg", "png"])
@@ -51,9 +62,9 @@ if uploaded_file is not None:
     # Save the uploaded file temporarily
     input_path = "test_folder"
     output_path = "target_test"
-    os.makedirs(input_path, exist_ok=True)
-    os.makedirs(osp(output_path, "result"), exist_ok=True)
-    os.makedirs(osp(output_path, "mask"), exist_ok=True)
+    # os.makedirs(input_path, exist_ok=True)
+    # os.makedirs(osp(output_path, "result"), exist_ok=True)
+    # os.makedirs(osp(output_path, "mask"), exist_ok=True)
 
     input_file_path = os.path.join(input_path, uploaded_file.name)
     with open(input_file_path, "wb") as f:
@@ -65,9 +76,9 @@ if uploaded_file is not None:
         
         try:
             status_text.text("Running text detection...")
-            os.makedirs(input_path, exist_ok=True)
-            os.makedirs(osp(output_path, "result"), exist_ok=True)
-            os.makedirs(osp(output_path, "mask"), exist_ok=True)
+            # os.makedirs(input_path, exist_ok=True)
+            # os.makedirs(osp(output_path, "result"), exist_ok=True)
+            # os.makedirs(osp(output_path, "mask"), exist_ok=True)
             rc, stderr_output = run_bash_script(input_path, output_path, progress_placeholder, status_text)
             
             if rc == 0:
@@ -95,7 +106,7 @@ if uploaded_file is not None:
             st.error(f"An error occurred: {str(e)}")
         finally:
             # Clean up temporary files
-            shutil.rmtree(osp(output_path, "mask"), ignore_errors=True)
+            clear_folder(osp(output_path, "mask"))
             progress_placeholder.empty()
             status_text.empty()