Update app.py

app.py

@@ -3,10 +3,14 @@ import numpy as np
 from io import BytesIO
 from PIL import Image
 import zipfile
-import cv2
+import os
+
+class NamedBytesIO(BytesIO):
+    def __init__(self, *args, name="output.zip", **kwargs):
+        super(NamedBytesIO, self).__init__(*args, **kwargs)
+        self.name = name
 
 def split_image_grid(image, grid_cols, grid_rows):
-    # Ensure image is a PIL Image
     if isinstance(image, np.ndarray):
         image = Image.fromarray(image)
     
@@ -25,7 +29,7 @@ def split_image_grid(image, grid_cols, grid_rows):
     return frames
 
 def zip_images(images):
-    zip_buffer = BytesIO()
+    zip_buffer = NamedBytesIO(name="output.zip")
     with zipfile.ZipFile(zip_buffer, 'w') as zipf:
         for idx, img in enumerate(images):
             img_buffer = BytesIO()
@@ -37,7 +41,7 @@ def zip_images(images):
     return zip_buffer
 
 def create_gif(images):
-    gif_buffer = BytesIO()
+    gif_buffer = NamedBytesIO(name="output.gif")
     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)
@@ -55,7 +59,7 @@ def process_image_to_gif(image, grid_cols_input, grid_rows_input):
 
 with gr.Blocks() as demo:
     with gr.Row():
-        image_input = gr.Image(label="Input Image", type="pil")  # Changed to "pil" for easier handling
+        image_input = gr.Image(label="Input Image", type="pil")
         grid_cols_input = gr.Slider(1, 10, value=2, step=1, label="Grid Columns")
         grid_rows_input = gr.Slider(1, 10, value=2, step=1, label="Grid Rows")
     with gr.Row():