Update app.py

app.py

@@ -50,4 +50,23 @@ def create_gif(frames):
     return gif_buffer
 
 def process_image(image, grid_size):
-    frames = split_image
+    # 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
+
+with gr.Blocks() as demo:
+    with gr.Row():
+        image_input = gr.Image(type="pil", label="Input Image")
+        grid_size_slider = gr.Slider(1, 10, value=2, step=1, label="Grid Size")
+    with gr.Row():
+        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)