Update app.py

app.py

@@ -3,6 +3,11 @@ from PIL import Image, ImageDraw
 import io
 import numpy as np
 from math import cos, radians, sin
+import os
+import time
+
+# Global variable to store generated GIFs
+generated_gifs = []
 
 def create_gif(editor1_output, editor2_output, transition_type):
     img1 = editor1_output["composite"].convert('RGBA')
@@ -12,7 +17,7 @@ def create_gif(editor1_output, editor2_output, transition_type):
     duration = 100  # Duration for each frame in milliseconds
     total_frames = 18  # Total number of frames
 
-    size = (256, 256)  # Fixed size for both images
+    size = (512, 512)  # Increased size for better quality
     img1 = img1.resize(size, Image.LANCZOS)
     img2 = img2.resize(size, Image.LANCZOS)
 
@@ -26,7 +31,7 @@ def create_gif(editor1_output, editor2_output, transition_type):
             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)
+            draw.line((full_width - i, 0, full_width - i, size[1]), fill=(0, 255, 0), width=4)
             frames.append(frame.convert('P', palette=Image.ADAPTIVE))
 
         for i in range(full_width, 0, -step):
@@ -34,7 +39,7 @@ def create_gif(editor1_output, editor2_output, transition_type):
             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)
+            draw.line((full_width - i, 0, full_width - i, size[1]), fill=(0, 255, 0), width=4)
             frames.append(frame.convert('P', palette=Image.ADAPTIVE))
     else:  # rotate transition
         mask_size = (size[0] * 2, size[1] * 2)
@@ -55,8 +60,8 @@ def create_gif(editor1_output, editor2_output, transition_type):
             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)
+            draw.line([center_x, center_y, end_x1, end_y1], fill=(0, 255, 0), width=6)
+            draw.line([center_x, center_y, end_x2, end_y2], fill=(0, 255, 0), width=6)
 
             frames.append(frame.convert('P', palette=Image.ADAPTIVE))
 
@@ -65,12 +70,19 @@ def create_gif(editor1_output, editor2_output, transition_type):
     frames[0].save(output, format='GIF', save_all=True, append_images=frames[1:], duration=duration, loop=0, optimize=True)
     output.seek(0)
 
-    # Save the GIF to a temporary file
-    temp_output_path = "output.gif"
-    with open(temp_output_path, "wb") as f:
+    # Save the GIF to a file with a unique name
+    os.makedirs("outputs", exist_ok=True)
+    timestamp = int(time.time())
+    output_path = f"outputs/output_{timestamp}.gif"
+    with open(output_path, "wb") as f:
         f.write(output.getvalue())
     
-    return temp_output_path
+    # Add the new GIF to the list of generated GIFs
+    global generated_gifs
+    generated_gifs.append((output_path, f"{transition_type.capitalize()} transition"))
+    
+    # Return the updated list of generated GIFs
+    return generated_gifs
 
 # Gradio interface
 with gr.Blocks() as iface:
@@ -105,12 +117,20 @@ with gr.Blocks() as iface:
         generate_button = gr.Button("Generate GIF")
     
     with gr.Row():
-        output_gif = gr.Image(type="filepath", label="Generated GIF", height=256, width=256)
+        output_gallery = gr.Gallery(
+            label="Generated GIFs",
+            show_label=True,
+            elem_id="output_gallery",
+            columns=3,
+            rows=2,
+            height=400,
+            object_fit="contain"
+        )
 
     generate_button.click(
         create_gif,
         inputs=[image_editor1, image_editor2, transition_type],
-        outputs=[output_gif]
+        outputs=[output_gallery]
     )
 
 # Launch the interface