Auto cleanup temporary files

Images and zip files are kept 24 hours.

app.py

@@ -1,11 +1,16 @@
+import os
+
 import gradio as gr
 import spaces
 from PIL import Image
 
 from dataset import ImageDataset
+from documentation import DOC_CONTENT
 from labelizer import get_task_response
 from packager import create_dataset_zip
-from documentation import DOC_CONTENT
+
+# Drop files after 24 hours
+CACHE_TTL = 24 * 60 * 60
 
 
 @spaces.GPU
@@ -73,7 +78,8 @@ def labelize_all_images(
 def create_dataset_zipfile(dataset: ImageDataset, organize_in_folders: bool):
     """Create and return zip file for download."""
     zip_path = create_dataset_zip(dataset, organize_in_folders)
-    return gr.update(visible=True, value=zip_path)
+    output = gr.update(visible=True, value=zip_path)
+    return output, zip_path
 
 
 def update_buttons_states(dataset: ImageDataset, labeling_in_progress=False):
@@ -99,7 +105,9 @@ def finish_labeling(dataset: ImageDataset):
     return update_buttons_states(dataset, labeling_in_progress=False)
 
 
-with gr.Blocks(title="Labelizer", fill_width=True) as demo:
+with gr.Blocks(
+    title="Labelizer", fill_width=True, delete_cache=(CACHE_TTL, CACHE_TTL)
+) as demo:
     dataset = gr.State()
     with gr.Sidebar():
         gr.Markdown("# 🖼️ Image Labeling Tool")
@@ -128,16 +136,18 @@ with gr.Blocks(title="Labelizer", fill_width=True) as demo:
             )
 
         with gr.Group():
-            organize_files = gr.Checkbox(
-                label="📂 Organize in folders", value=False, render=False
-            )
+            # original zip file to drop, see the click envent of download_button
+            to_delete = gr.State()
+            # should create a zip file
             download_button = gr.Button(
                 "💾 Create zip file to download",
                 interactive=False,
                 size="lg",
             )
+            # the download section
             download_file = gr.File(label="Generated datasets", visible=False)
-            organize_files.render()
+            # to organize dataset in folders or not
+            organize_files = gr.Checkbox(label="📂 Organize in folders", value=False)
 
     @gr.render(inputs=[dataset, is_labeling_in_progress])
     def render_grid(ds, is_labeling_in_progress):
@@ -284,7 +294,11 @@ with gr.Blocks(title="Labelizer", fill_width=True) as demo:
     ).then(
         create_dataset_zipfile,
         inputs=[dataset, organize_files],
-        outputs=download_file,
+        outputs=[download_file, to_delete],
+    ).then(
+        # delete the generated files from /tmp as it is now coppied in gradio cache
+        lambda x: os.remove(x),
+        inputs=[to_delete],
     )