Update app.py

app.py

@@ -5,11 +5,6 @@ from PIL import Image
 import zipfile
 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):
     if isinstance(image, np.ndarray):
         image = Image.fromarray(image)
@@ -29,7 +24,7 @@ def split_image_grid(image, grid_cols, grid_rows):
     return frames
 
 def zip_images(images):
-    zip_buffer = NamedBytesIO(name="output.zip")
+    zip_buffer = BytesIO()
     with zipfile.ZipFile(zip_buffer, 'w') as zipf:
         for idx, img in enumerate(images):
             img_buffer = BytesIO()
@@ -38,14 +33,14 @@ def zip_images(images):
             img_buffer.seek(0)
             zipf.writestr(f'image_{idx}.png', img_buffer.getvalue())
     zip_buffer.seek(0)
-    return zip_buffer
+    return {"name": "output.zip", "data": zip_buffer.getvalue()}
 
 def create_gif(images):
-    gif_buffer = NamedBytesIO(name="output.gif")
+    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 gif_buffer
+    return {"name": "output.gif", "data": gif_buffer.getvalue()}
 
 def process_image(image, grid_cols_input, grid_rows_input):
     frames = split_image_grid(image, grid_cols_input, grid_rows_input)