Spaces:
Running
on
Zero
Running
on
Zero
File size: 3,924 Bytes
a12f521 bfa7d21 a12f521 bfa7d21 a12f521 bfa7d21 a12f521 bfa7d21 a12f521 bfa7d21 a12f521 bfa7d21 a12f521 |
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 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 |
#!/usr/bin/env python
import argparse
from pathlib import Path
from PIL import Image
import numpy as np
def create_comparison_gif(before_img_path, after_img_path, output_path=None, duration=100, frames=20, crop_pixels=0, bottom_bias=0.50):
"""
Create a comparison GIF with a vertical bar revealing the 'after' image.
Args:
before_img_path (str): Path to the 'before' image
after_img_path (str): Path to the 'after' image
output_path (str, optional): Path for the output GIF. Defaults to 'comparison.gif'
duration (int, optional): Duration of each frame in ms. Defaults to 100.
frames (int, optional): Number of frames in the GIF. Defaults to 20.
crop_pixels (int, optional): Number of pixels to crop from each side. Defaults to 0.
bottom_bias (float, optional): Percentage of additional cropping from the bottom. Defaults to 0.25 (25% more from the bottom).
Returns:
str: Path to the created GIF
"""
# Default output path
if output_path is None:
output_path = 'comparison.gif'
# Open images and ensure they have the same size
before_img = Image.open(before_img_path)
after_img = Image.open(after_img_path)
# Resize if they're different
if before_img.size != after_img.size:
after_img = after_img.resize(before_img.size, Image.LANCZOS)
width, height = before_img.size
# Crop the images if crop_pixels is specified
if crop_pixels > 0:
left = crop_pixels
right = width - crop_pixels
top = int(crop_pixels * (1 - bottom_bias))
bottom = height - int(crop_pixels * (1 + bottom_bias))
before_img = before_img.crop((left, top, right, bottom))
after_img = after_img.crop((left, top, right, bottom))
width, height = before_img.size # Update width and height after cropping
# Create frames for the GIF
gif_frames = []
for i in range(frames + 1):
# Calculate position of the reveal bar
bar_position = int((width * i) / frames)
# Create a new frame
frame = Image.new('RGBA', (width, height))
# Paste the 'before' image
frame.paste(before_img, (0, 0))
# Create a mask for the 'after' image
mask = Image.new('L', (width, height), 0)
mask_array = np.array(mask)
mask_array[:, :bar_position] = 255
mask = Image.fromarray(mask_array)
# Paste the 'after' image with the mask
frame.paste(after_img, (0, 0), mask)
# Add to frames
gif_frames.append(frame.convert('RGBA'))
# Save as GIF
gif_frames[0].save(
output_path,
save_all=True,
append_images=gif_frames[1:],
optimize=False,
duration=duration,
loop=0,
disposal=2
)
return output_path
def main():
parser = argparse.ArgumentParser(description='Create a before/after comparison GIF with a sliding reveal effect')
parser.add_argument('before_image', help='Path to the before image')
parser.add_argument('after_image', help='Path to the after image')
parser.add_argument('--output', '-o', help='Output path for the GIF', default='comparison.gif')
parser.add_argument('--duration', '-d', type=int, help='Duration of each frame in ms', default=100)
parser.add_argument('--frames', '-f', type=int, help='Number of frames to generate', default=20)
parser.add_argument('--crop', '-c', type=int, help='Number of pixels to crop from each side', default=0)
args = parser.parse_args()
output_path = create_comparison_gif(
args.before_image,
args.after_image,
args.output,
args.duration,
args.frames,
args.crop
)
print(f"Created comparison GIF: {output_path}")
if __name__ == "__main__":
main() |