Update app.py

app.py

@@ -4,7 +4,20 @@ from io import BytesIO
 from PIL import Image
 import zipfile
 import os
-import tempfile
+import atexit
+import shutil
+
+# Create a persistent directory to store generated files
+GENERATED_FILES_DIR = "generated_files"
+if not os.path.exists(GENERATED_FILES_DIR):
+    os.makedirs(GENERATED_FILES_DIR)
+
+def cleanup_generated_files():
+    if os.path.exists(GENERATED_FILES_DIR):
+        shutil.rmtree(GENERATED_FILES_DIR)
+
+# Register the cleanup function to run when the script exits
+atexit.register(cleanup_generated_files)
 
 def split_image_grid(image, grid_cols, grid_rows):
     if isinstance(image, np.ndarray):
@@ -25,23 +38,21 @@ def split_image_grid(image, grid_cols, grid_rows):
     return frames
 
 def zip_images(images):
-    with tempfile.TemporaryDirectory() as temp_dir:
-        zip_path = os.path.join(temp_dir, "output.zip")
-        with zipfile.ZipFile(zip_path, 'w') as zipf:
-            for idx, img in enumerate(images):
-                img_buffer = BytesIO()
-                img = Image.fromarray(img)
-                img.save(img_buffer, format='PNG')
-                img_buffer.seek(0)
-                zipf.writestr(f'image_{idx}.png', img_buffer.getvalue())
-        return zip_path
+    zip_path = os.path.join(GENERATED_FILES_DIR, "output.zip")
+    with zipfile.ZipFile(zip_path, 'w') as zipf:
+        for idx, img in enumerate(images):
+            img_buffer = BytesIO()
+            img = Image.fromarray(img)
+            img.save(img_buffer, format='PNG')
+            img_buffer.seek(0)
+            zipf.writestr(f'image_{idx}.png', img_buffer.getvalue())
+    return zip_path
 
 def create_gif(images):
-    with tempfile.TemporaryDirectory() as temp_dir:
-        gif_path = os.path.join(temp_dir, "output.gif")
-        images_pil = [Image.fromarray(img) for img in images]
-        images_pil[0].save(gif_path, format='GIF', save_all=True, append_images=images_pil[1:], duration=100, loop=0)
-        return gif_path
+    gif_path = os.path.join(GENERATED_FILES_DIR, "output.gif")
+    images_pil = [Image.fromarray(img) for img in images]
+    images_pil[0].save(gif_path, format='GIF', save_all=True, append_images=images_pil[1:], duration=100, loop=0)
+    return gif_path
 
 def process_image(image, grid_cols_input, grid_rows_input):
     frames = split_image_grid(image, grid_cols_input, grid_rows_input)