Update app.py

app.py

@@ -4,6 +4,7 @@ from io import BytesIO
 from PIL import Image
 import zipfile
 import os
+import tempfile
 
 def split_image_grid(image, grid_cols, grid_rows):
     if isinstance(image, np.ndarray):
@@ -24,23 +25,23 @@ def split_image_grid(image, grid_cols, grid_rows):
     return frames
 
 def zip_images(images):
-    zip_buffer = BytesIO()
-    with zipfile.ZipFile(zip_buffer, '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())
-    zip_buffer.seek(0)
-    return {"name": "output.zip", "data": zip_buffer.getvalue()}
+    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
 
 def create_gif(images):
-    gif_buffer = BytesIO()
-    images_pil = [Image.fromarray(img) for img in images]
-    images_pil[0].save(gif_buffer, format='GIF', save_all=True, append_images=images_pil[1:], duration=100, loop=0)
-    gif_buffer.seek(0)
-    return {"name": "output.gif", "data": gif_buffer.getvalue()}
+    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
 
 def process_image(image, grid_cols_input, grid_rows_input):
     frames = split_image_grid(image, grid_cols_input, grid_rows_input)