Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import numpy as np
 import zipfile
-import io
+from io import BytesIO
 from PIL import Image
 
 def split_image_grid(image, grid_size):
@@ -29,10 +29,10 @@ def split_image_grid(image, grid_size):
 
 def create_zip_file(frames):
     # Create an in-memory zip file
-    zip_buffer = io.BytesIO()
+    zip_buffer = BytesIO()
     with zipfile.ZipFile(zip_buffer, 'w') as zipf:
         for idx, frame in enumerate(frames):
-            frame_byte_array = io.BytesIO()
+            frame_byte_array = BytesIO()
             # Save the frame as a PNG file in the byte array
             frame_img = Image.fromarray(frame)
             frame_img.save(frame_byte_array, format="PNG")
@@ -43,18 +43,15 @@ def create_zip_file(frames):
 
 def create_gif(frames):
     # Create a GIF from the frames
-    gif_buffer = io.BytesIO()
+    gif_buffer = BytesIO()
     frames_pil = [Image.fromarray(frame) for frame in frames]
     frames_pil[0].save(gif_buffer, format="GIF", save_all=True, append_images=frames_pil[1:], loop=0)
-    gif_buffer.seek(0)  # Ensure the buffer is at the beginning
+    gif_buffer.seek(0)
     return gif_buffer
 
 def process_image(image, grid_size):
-    # Split the image into frames
     frames = split_image_grid(image, grid_size)
-    # Create a zip file from the frames
     zip_file = create_zip_file(frames)
-    # Create a GIF from the frames
     gif_file = create_gif(frames)
     return zip_file, gif_file
 
@@ -66,7 +63,6 @@ with gr.Blocks() as demo:
         zip_output = gr.File(label="Download ZIP")
         gif_output = gr.Image(label="Generated GIF")
     process_button = gr.Button("Process Image")
-
     process_button.click(process_image, inputs=[image_input, grid_size_slider], outputs=[zip_output, gif_output])
 
 demo.launch(show_error=True)