Update app.py

app.py

@@ -25,19 +25,39 @@ def create_gif(editor1_output, editor2_output, transition_type):
             frame = Image.new('RGBA', size)
             frame.paste(img1, (0, 0))
             frame.paste(img2.crop((0, 0, i, size[1])), (full_width - i, 0), mask=img2.crop((0, 0, i, size[1])))
+            draw = ImageDraw.Draw(frame)
+            draw.line((full_width - i, 0, full_width - i, size[1]), fill=(0, 255, 0), width=2)
             frames.append(frame.convert('P', palette=Image.ADAPTIVE))
 
         for i in range(full_width, 0, -step):
             frame = Image.new('RGBA', size)
             frame.paste(img2, (0, 0))
             frame.paste(img1.crop((0, 0, i, size[1])), (full_width - i, 0), mask=img1.crop((0, 0, i, size[1])))
+            draw = ImageDraw.Draw(frame)
+            draw.line((full_width - i, 0, full_width - i, size[1]), fill=(0, 255, 0), width=2)
             frames.append(frame.convert('P', palette=Image.ADAPTIVE))
     else:  # rotate transition
+        mask_size = (size[0] * 2, size[1] * 2)
+        mask = Image.new('L', mask_size, 0)
+        draw = ImageDraw.Draw(mask)
+        draw.rectangle([size[0], 0, mask_size[0], mask_size[1]], fill=255)
+
+        center_x, center_y = size[0] // 2, size[1] // 2
+
         for angle in range(0, 360, 360 // total_frames):
-            mask = Image.new('L', size, 0)
-            draw = ImageDraw.Draw(mask)
-            draw.pieslice([0, 0, size[0], size[1]], -90, angle - 90, fill=255)
-            frame = Image.composite(img2, img1, mask)
+            rotated_mask = mask.rotate(angle, center=(mask_size[0] // 2, mask_size[1] // 2), expand=False)
+            cropped_mask = rotated_mask.crop((size[0] // 2, size[1] // 2, size[0] // 2 + size[0], size[1] // 2 + size[1]))
+            frame = Image.composite(img1, img2, cropped_mask)
+
+            draw = ImageDraw.Draw(frame)
+            reverse_angle = -angle + 90
+            end_x1 = center_x + int(size[0] * 1.5 * cos(radians(reverse_angle)))
+            end_y1 = center_y + int(size[1] * 1.5 * sin(radians(reverse_angle)))
+            end_x2 = center_x - int(size[0] * 1.5 * cos(radians(reverse_angle)))
+            end_y2 = center_y - int(size[1] * 1.5 * sin(radians(reverse_angle)))
+            draw.line([center_x, center_y, end_x1, end_y1], fill=(0, 255, 0), width=3)
+            draw.line([center_x, center_y, end_x2, end_y2], fill=(0, 255, 0), width=3)
+
             frames.append(frame.convert('P', palette=Image.ADAPTIVE))
 
     # Save as GIF
@@ -94,4 +114,4 @@ with gr.Blocks() as iface:
     )
 
 # Launch the interface
-iface.launch(share=True)
+iface.launch(share=True)