not save to local

app.py

@@ -3,15 +3,15 @@ import shutil
 import uuid
 import logging
 
-from flask import Flask, redirect, url_for, request, flash, session
+from flask import Flask, redirect, url_for, request, flash, session, after_this_request
 from flask import render_template
 from flask import send_file
 
 logger = logging.getLogger(__name__)
 logging.basicConfig(level=logging.INFO)
 
-UPLOAD_FOLDER = '/app/images'
-OUT_FOLDER = '/app/anno'
+UPLOAD_FOLDER = './images'
+OUT_FOLDER = './anno'
 
 app = Flask(__name__)
 app.config['SEND_FILE_MAX_AGE_DEFAULT'] = 0
@@ -142,10 +142,25 @@ def download():
     anno_path = get_anno_path(user_id)
     img_dir = get_images_directory(user_id)
     shutil.copyfile(anno_path, f'{img_dir}/annotations_pascal_voc.csv')
+    download_zip = os.path.join(img_dir, 'final')
+    print(download_zip)
     shutil.make_archive('final', 'zip', img_dir)
+    # shutil.make_archive(base_name=download_zip,
+    #                     format='zip',
+    #                     base_dir=img_dir)
+
+    @after_this_request
+    def remove_directory(response):
+        try:
+            shutil.rmtree(img_dir)
+            os.remove(anno_path)
+        except Exception as e:
+            print(f"Error deleting directory {img_dir}: {e}")
+        return response
+
     return send_file('final.zip',
                      mimetype='text/csv',
-                     download_name='final.zip',
+                     download_name='annotated_data.zip',
                      as_attachment=True)
 
 if __name__ == "__main__":

templates/tagger.html

@@ -59,9 +59,13 @@
             <h1 class="text-3xl sm:text-4xl font-bold text-gray-800 leading-tight">
                 🇰🇭 For the development of AI 🇰🇭
             </h1>
-            <h3>One annotation takes 2 steps: 1) Create bounding box by click and drag on the image, and 2) Enter label for the bounding box</h3>
-            <h3>Move to the next image by clicking on ✅ at the right</h3>
-            <h4>Loading images may take a few seconds. Please wait...</h4>
+            <h3>One annotation takes 2 steps:</h3>
+            <div class="mb-4">
+                <h4>1. Create bounding box by click and drag on the image,</h4>
+                <h4>2. Enter label for the bounding box</h4>
+            </div>
+            <h4>Move to the next image by clicking on ➡️ or ✅ at the right</h4>
+            <h5>Loading images may take a few seconds. Please wait...</h5>
         </div>
 
         <!-- Right Section: Placeholder (can be used for user avatar/menu later) -->