my

app.py

@@ -47,19 +47,32 @@ def calculate_iou(bbox1, bbox2):
     return iou
 
 
-cropped_dir = "./app/cropped_images/"
-if os.path.exists(cropped_dir):
+def recreate_directories():
+    # Paths for the directories
+    cropped_dir = "./app/cropped_images/"
+    output_dir1 = "./app/Folder1"
+    output_dir2 = "./app/Folder2"
+    output_dir3 = "./app/Folder3"
+    UPLOAD_FOLDER = "./app/data1"
+
+    # Remove existing directories
+    if os.path.exists(cropped_dir):
         shutil.rmtree(cropped_dir)
-os.makedirs(cropped_dir, exist_ok=True)
-
-output_dir1 = "./app/Folder1"
-output_dir2 = "./app/Folder2"
-output_dir3 = "./app/Folder3"
-UPLOAD_FOLDER = "./app/data1"
-os.makedirs(output_dir1, exist_ok=True)
-os.makedirs(output_dir2, exist_ok=True)
-os.makedirs(output_dir3, exist_ok=True)
-os.makedirs(UPLOAD_FOLDER, exist_ok=True)
+    if os.path.exists(output_dir1):
+        shutil.rmtree(output_dir1)
+    if os.path.exists(output_dir2):
+        shutil.rmtree(output_dir2)
+    if os.path.exists(output_dir3):
+        shutil.rmtree(output_dir3)
+    if os.path.exists(UPLOAD_FOLDER):
+        shutil.rmtree(UPLOAD_FOLDER)
+
+    # Recreate directories
+    os.makedirs(cropped_dir, exist_ok=True)
+    os.makedirs(output_dir1, exist_ok=True)
+    os.makedirs(output_dir2, exist_ok=True)
+    os.makedirs(output_dir3, exist_ok=True)
+    os.makedirs(UPLOAD_FOLDER, exist_ok=True)
 
 @app.route('/')
 def index():
@@ -67,13 +80,14 @@ def index():
 
 @app.route('/upload', methods=['POST'])
 def upload_file():
+    recreate_directories()
     if 'invoice-upload' not in request.files:
         return jsonify({'error': 'No file part'}), 400
     file = request.files['invoice-upload']
     if file.filename == '':
         return jsonify({'error': 'No selected file'}), 400
     if file:
-        file_path = os.path.join(UPLOAD_FOLDER, file.filename)
+        file_path = os.path.join('./app/data1', file.filename)
         file.save(file_path)
         output_image, output_csv = process_image()