Update app.py

app.py

@@ -3,29 +3,23 @@ import numpy as np
 import zipfile
 from io import BytesIO
 from PIL import Image
+import os
 
-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_cols
-    grid_height = grid_rows
-    # Calculate the size of each grid cell
-    cell_width = width // grid_width
-    cell_height = height // grid_height
-
-    # Split the image into individual frames
-    frames = []
-    for i in range(grid_height):
-        for j in range(grid_width):
-            left = j * cell_width
-            upper = i * cell_height
-            right = left + cell_width
-            lower = upper + cell_height
-            frame = img[upper:lower, left:right]
-            frames.append(frame)
-
-    return frames
+
+def cut_image(image, num_cuts_h, num_cuts_v):
+    width, height = image.size
+    cut_width = width // num_cuts_h
+    cut_height = height // num_cuts_v
+    parts = []
+        for i in range(num_cuts_v):
+        for j in range(num_cuts_h):
+        left = j * cut_width
+        upper = i * cut_height
+        right = left + cut_width
+        lower = upper + cut_height
+        parts.append(image.crop((left, upper, right, lower)))
+
+    return parts
 
 def zip_images(images):
     # Create a BytesIO object to hold the zip file
@@ -42,21 +36,25 @@ def zip_images(images):
     zip_buffer.seek(0)
     return zip_buffer
 
-def process_image(image, grid_cols_input, grid_rows_input):
+def process_image(image, num_cuts_h, num_cuts_v):
     # Split the image into a grid of frames
-    frames = split_image_grid(image, grid_cols_input, grid_rows_input)
+    cut_parts = cut_image(image, num_cuts_h, num_cuts_v)
+    #frames = split_image_grid(image, grid_cols_input, grid_rows_input)
     # Zip the frames into a single zip file
-    zip_file = zip_images(frames)
+    zip_file = zip_images(cut_parts)
     return zip_file
 
 with gr.Blocks() as demo:
     with gr.Row():
         image_input = gr.Image(label="Input Image", type="filepath")
-        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")
+        num_cuts_h = gr.Slider(minimum=2, maximum=8, default=3, step=1, label="Number of horizontal cuts/slices")
+        num_cuts_v = gr.Slider(minimum=2, maximum=8, default=3, step=1, label="Number of vertical cuts/slices")
+        gif_duration = gr.Slider(minimum=5, maximum=1000, default=150, step=5, label="GIF duration (ms)")
+        ping_pong_checkbox = gr.Checkbox(label="Ping-pong animation", default=True)
     zip_output = gr.File(label="Output Zip File")
     process_button = gr.Button("Process Image")
 
-    process_button.click(process_image, inputs=[image_input, grid_cols_input, grid_rows_input], outputs=zip_output)
+    # process_button.click(process_image, inputs=[image_input, num_cuts_h, num_cuts_v, gif_duration, ping_pong_checkbox], outputs=zip_output)
+    process_button.click(process_image, inputs=[image_input, num_cuts_h, num_cuts_v], outputs=zip_output)
 
 demo.launch(show_error=True)