Update app.py

app.py

@@ -40,19 +40,19 @@ def split_image_grid(image, grid_cols, grid_rows):
             frames.append(np.array(frame))
     return frames
 
-def interpolate_frames(frames, factor=2):
+def interpolate_frames(frames, interpolation_factor=2):
     interpolated_frames = []
     for i in range(len(frames) - 1):
         frame1 = frames[i]
         frame2 = frames[i + 1]
         interpolated_frames.append(frame1)
-        for j in range(1, factor):
-            t = j / factor
+        for j in range(1, interpolation_factor):
+            t = j / interpolation_factor
             frame_t = cv2.addWeighted(frame1, 1 - t, frame2, t, 0)
             interpolated_frames.append(frame_t)
     interpolated_frames.append(frames[-1])
     return interpolated_frames
-
+    
 def enhance_gif(images):
     enhanced_images = []
     for img in images:
@@ -61,23 +61,25 @@ def enhance_gif(images):
         enhanced_images.append(np.array(img))
     return enhanced_images
 
-def create_gif_imageio(images, fps=10, loop=0):
+def create_gif_imageio(images, fps=10, ping_pong_animation=False, loop=0):
     duration = 1000 / fps  # Convert FPS to milliseconds
+    if ping_pong_animation:
+        images = images + images[-2:0:-1]  # Create a ping-pong sequence
     gif_path = os.path.join(GENERATED_FILES_DIR, "output_enhanced.gif")
     images_pil = [Image.fromarray(img) for img in images]
     imageio.mimsave(gif_path, images_pil, duration=duration, loop=loop)
     return gif_path
-    
+
 def process_image(image, grid_cols_input, grid_rows_input):
     frames = split_image_grid(image, grid_cols_input, grid_rows_input)
     zip_file = zip_images(frames)
     return zip_file
 
-def process_image_to_gif(image, grid_cols_input, grid_rows_input, fps_input):
+def process_image_to_gif(image, grid_cols_input, grid_rows_input, fps_input, ping_pong_toggle, interpolation_factor):
     frames = split_image_grid(image, grid_cols_input, grid_rows_input)
-    interpolated_frames = interpolate_frames(frames, factor=2)
+    interpolated_frames = interpolate_frames(frames, interpolation_factor)
     enhanced_frames = enhance_gif(interpolated_frames)
-    gif_file = create_gif_imageio(enhanced_frames, fps=fps_input, loop=0)
+    gif_file = create_gif_imageio(enhanced_frames, fps=fps_input, ping_pong_animation=ping_pong_toggle, loop=0)
     
     # Preview the first frame of the GIF
     # preview_image = Image.fromarray(enhanced_frames[0])
@@ -103,6 +105,9 @@ with gr.Blocks() as demo:
         grid_cols_input = gr.Slider(1, 10, value=2, step=1, label="Grid Columns")
         grid_rows_input = gr.Slider(1, 10, value=2, step=1, label="Grid Rows")
         fps_input = gr.Slider(1, 30, value=10, step=1, label="FPS (Frames per Second)")
+        ping_pong_toggle = gr.Checkbox(label="Ping-Pong Effect")
+        interpolation_factor_input = gr.Slider(1, 10, value=2, step=1, label="Interpolation Factor")
+
     with gr.Row():
         zip_button = gr.Button("Create Zip File")
         gif_button = gr.Button("Create GIF")
@@ -113,7 +118,7 @@ with gr.Blocks() as demo:
     zip_button.click(process_image, inputs=[image_input, grid_cols_input, grid_rows_input], outputs=zip_output)
     gif_button.click(
         process_image_to_gif, 
-        inputs=[image_input, grid_cols_input, grid_rows_input, fps_input], 
+        inputs=[image_input, grid_cols_input, grid_rows_input, fps_input, ping_pong_toggle, interpolation_factor_input], 
         outputs=[preview_image, gif_output]
     )