Update app.py

app.py

@@ -40,7 +40,9 @@ logging.basicConfig(level=logging.WARNING)
 
 WEIGHTS_PATH = 'best.pt' 
 SCALE_FACTOR = 2.0 
-OUTPUT_DIR = "yolo_extracted_regions"
+# OUTPUT_DIR = "yolo_extracted_regions"
+OUTPUT_DIR = os.path.join(tempfile.gettempdir(), "yolo_extracted_regions")
+
 
 # Detection parameters
 CONF_THRESHOLD = 0.2
@@ -350,6 +352,13 @@ def run_single_pdf_preprocessing(pdf_path: str) -> Tuple[int, int, int, str, flo
     GLOBAL_FIGURE_COUNT = 0
     GLOBAL_EQUATION_COUNT = 0
 
+
+
+    if os.path.exists(OUTPUT_DIR):
+    shutil.rmtree(OUTPUT_DIR)
+    os.makedirs(OUTPUT_DIR, exist_ok=True)
+
+
     # 1. Validation and Model Loading
     t0 = time.time()
     if not os.path.exists(pdf_path):
@@ -527,6 +536,10 @@ if __name__ == "__main__":
     )
 
     print("\nStarting Gradio application...")
-    interface.launch(inbrowser=True)
+    # interface.launch(inbrowser=True)
+    interface.launch(
+    inbrowser=True,
+    allowed_paths=[OUTPUT_DIR]
+)