File size: 2,312 Bytes
38d647f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 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
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, "") |