Update app.py

app.py

@@ -34,7 +34,7 @@ def zip_images(images):
         for idx, img in enumerate(images):
             # Save each image to the zip file
             img_buffer = BytesIO()
-            img = Image.fromarray(img)
+            img = Image.fromarray(img)  # Convert NumPy array to PIL Image
             img.save(img_buffer, format='PNG')
             img_buffer.seek(0)
             zipf.writestr(f'image_{idx}.png', img_buffer.getvalue())
@@ -42,20 +42,20 @@ def zip_images(images):
     zip_buffer.seek(0)
     return zip_buffer
 
+def process_image(image, grid_size):
+    # Split the image into a grid of frames
+    frames = split_image_grid(image, grid_size)
+    # Zip the frames into a single zip file
+    zip_file = zip_images(frames)
+    return zip_file
+
 with gr.Blocks() as demo:
     with gr.Row():
         image_input = gr.Image(label="Input Image")
         grid_size_input = gr.Slider(1, 10, value=2, step=1, label="Grid Size")
-    split_button = gr.Button("Split Image")
-    zip_output = gr.File(label="Download Zipped Images")
-
-    def process_image(image, grid_size):
-        # Split the image into a grid of frames
-        frames = split_image_grid(image, (grid_size, grid_size))
-        # Zip the frames into a single file
-        zip_file = zip_images(frames)
-        return zip_file
+    zip_output = gr.File(label="Output Zip File")
+    process_button = gr.Button("Process Image")
 
-    split_button.click(process_image, inputs=[image_input, grid_size_input], outputs=zip_output)
+    process_button.click(process_image, inputs=[image_input, grid_size_input], outputs=zip_output)
 
 demo.launch(show_error=True)