Update app.py

app.py

@@ -1,53 +1,53 @@
-import os.path
-import modules.scripts as scripts
 import gradio as gr
-from modules import sd_samplers, shared
-from modules.processing import Processed, process_images, StableDiffusionProcessing, create_infotext
-import modules.images as images
-from modules.shared import opts, cmd_opts, state
+import numpy as np
+import zipfile
+import io
 from PIL import Image
-import os
-
-class Script(scripts.Script):
-
-    def title(self):
-        return "GIF creator from Image Slice"
-
-    def show(self, is_txt2img):
-        return True
-
-    def ui(self, is_txt2img):
-        num_cuts = gr.inputs.Slider(minimum=2, maximum=8, default=2, step=1, label="Number of cuts/slices")
-        gif_duration = gr.inputs.Slider(minimum=5, maximum=1000, default=150, step=5, label="GIF duration (ms)")
-        ping_pong_checkbox = gr.inputs.Checkbox(label="Ping-pong animation", default=True)
-        return [num_cuts, gif_duration, ping_pong_checkbox]
-    
-    def cut_image(self, image, num_cuts):
-        width, height = image.size
-        cut_width = width // num_cuts
-        cut_height = height // num_cuts
-        parts = []
-        for i in range(num_cuts):
-            for j in range(num_cuts):
-                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 run(self, p, num_cuts, gif_duration, ping_pong_animation):
-        proc = process_images(p)
-        gens = proc.images
-        save_image_tuple = images.save_image(gens[0], p.outpath_samples, "", 0, "-Original", opts.samples_format)
-        save_image = save_image_tuple[0]
-        image = Image.open(save_image) # Cut the image into equal parts and create a GIF with optional ping-pong animation
-        cut_parts = self.cut_image(image, num_cuts)
-        gif_filename = os.path.splitext(save_image)[0] + "_gif.gif"
-        if ping_pong_animation:
-            cut_parts = cut_parts + cut_parts[-2:0:-1]  # Create a ping-pong sequence
-        cut_parts[0].save(gif_filename, format='GIF', append_images=cut_parts[1:], save_all=True, duration=gif_duration, loop=0)
-        print(f"GIF created: {gif_filename}")
-        os.remove(save_image) # Delete the original saved image
-    
-        return Processed(p, gens, 0, "")
+
+def split_image_grid(image, grid_size):
+    # 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_size
+
+    # 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 = (j + 1) * cell_width
+            lower = (i + 1) * cell_height
+            frame = img[upper:lower, left:right]
+            frames.append(frame)
+
+    return frames
+
+def create_zip_file(frames):
+    # Create an in-memory zip file
+    zip_buffer = io.BytesIO()
+    with zipfile.ZipFile(zip_buffer, 'w') as zipf:
+        for idx, frame in enumerate(frames):
+            frame_byte_array = io.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")
+            zipf.writestr(f"frame_{idx}.png", frame_byte_array.getvalue())
+
+    zip_buffer.seek(0)
+    return zip_buffer
+
+def create_gif(frames):
+    # Create a GIF from the frames
+    gif_buffer = io.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
+    return gif_buffer
+
+def process_image(image, grid_size):
+    frames = split_image