Update app.py

app.py

@@ -1,11 +1,14 @@
 import gradio as gr
 import numpy as np
 from io import BytesIO
-from PIL import Image
+from PIL import Image, ImageOps
 import zipfile
 import os
 import atexit
 import shutil
+import cv2
+import imageio
+import torchvision.transforms.functional as TF
 
 # Create a persistent directory to store generated files
 GENERATED_FILES_DIR = "generated_files"
@@ -37,21 +40,31 @@ def split_image_grid(image, grid_cols, grid_rows):
             frames.append(np.array(frame))
     return frames
 
-def zip_images(images):
-    zip_path = os.path.join(GENERATED_FILES_DIR, "output.zip")
-    with zipfile.ZipFile(zip_path, 'w') as zipf:
-        for idx, img in enumerate(images):
-            img_buffer = BytesIO()
-            img = Image.fromarray(img)
-            img.save(img_buffer, format='PNG')
-            img_buffer.seek(0)
-            zipf.writestr(f'image_{idx}.png', img_buffer.getvalue())
-    return zip_path
+def interpolate_frames(frames, 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
+            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:
+        img = ImageOps.autocontrast(Image.fromarray(img))
+        img = img.convert("RGB")  # Ensure the image is in RGB mode
+        enhanced_images.append(np.array(img))
+    return enhanced_images
 
-def create_gif(images):
-    gif_path = os.path.join(GENERATED_FILES_DIR, "output.gif")
+def create_gif_imageio(images, duration=50, loop=0):
+    gif_path = os.path.join(GENERATED_FILES_DIR, "output_enhanced.gif")
     images_pil = [Image.fromarray(img) for img in images]
-    images_pil[0].save(gif_path, format='GIF', save_all=True, append_images=images_pil[1:], duration=100, loop=0)
+    imageio.mimsave(gif_path, images_pil, duration=duration, loop=loop)
     return gif_path
 
 def process_image(image, grid_cols_input, grid_rows_input):
@@ -61,9 +74,22 @@ def process_image(image, grid_cols_input, grid_rows_input):
 
 def process_image_to_gif(image, grid_cols_input, grid_rows_input):
     frames = split_image_grid(image, grid_cols_input, grid_rows_input)
-    gif_file = create_gif(frames)
+    interpolated_frames = interpolate_frames(frames, factor=2)
+    enhanced_frames = enhance_gif(interpolated_frames)
+    gif_file = create_gif_imageio(enhanced_frames, duration=50, loop=0)
     return gif_file
 
+def zip_images(images):
+    zip_path = os.path.join(GENERATED_FILES_DIR, "output.zip")
+    with zipfile.ZipFile(zip_path, 'w') as zipf:
+        for idx, img in enumerate(images):
+            img_buffer = BytesIO()
+            img = Image.fromarray(img)
+            img.save(img_buffer, format='PNG')
+            img_buffer.seek(0)
+            zipf.writestr(f'image_{idx}.png', img_buffer.getvalue())
+    return zip_path
+
 with gr.Blocks() as demo:
     with gr.Row():
         image_input = gr.Image(label="Input Image", type="pil")