Update app.py

app.py

@@ -4,12 +4,12 @@ import zipfile
 from io import BytesIO
 from PIL import Image
 
-def split_image_grid(image, grid_size):
+def split_image_grid(image, grid_cols, grid_rows):
     # Convert the image to a NumPy array
     img = np.array(image)
     width, height = img.shape[1], img.shape[0]
-    grid_width, grid_height = grid_size
-
+    grid_width = grid_cols
+    grid_height = grid_rows
     # Calculate the size of each grid cell
     cell_width = width // grid_width
     cell_height = height // grid_height
@@ -42,9 +42,9 @@ def zip_images(images):
     zip_buffer.seek(0)
     return zip_buffer
 
-def process_image(image, grid_size):
+def process_image(image, grid_cols_input, grid_rows_input):
     # Split the image into a grid of frames
-    frames = split_image_grid(image, grid_size)
+    frames = split_image_grid(image, grid_cols_input, grid_rows_input)
     # Zip the frames into a single zip file
     zip_file = zip_images(frames)
     return zip_file
@@ -52,10 +52,11 @@ def process_image(image, grid_size):
 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")
+        grid_cols_input = gr.Slider(1, 10, value=2, step=1, label="Grid Cols")
+        grid_rows_input = gr.Slider(1, 10, value=2, step=1, label="Grid Rows")
     zip_output = gr.File(label="Output Zip File")
     process_button = gr.Button("Process Image")
 
-    process_button.click(process_image, inputs=[image_input, grid_size_input], outputs=zip_output)
+    process_button.click(process_image, inputs=[image_input, grid_cols_input, grid_rows_input], outputs=zip_output)
 
 demo.launch(show_error=True)