Update app.py

app.py

@@ -1,4 +1,86 @@
 import gradio as gr
+import numpy as np
+import cv2
 
-i = gr.Interface.load("models/morph-labs/morph-prover-v0-7b")
-i.launch()
+class ImageMorpher:
+    def __init__(self, num_frames=30, fps=10):
+        self.num_frames = num_frames
+        self.fps = fps
+
+    def morph_images(self, image_A, image_B):
+        """Morphs two images."""
+        image_A = np.array(image_A)
+        image_B = np.array(image_B)
+
+        if image_A.shape != image_B.shape:
+            raise ValueError("Images must have the same dimensions.")
+        if image_A.dtype != image_B.dtype:
+            raise ValueError("Images must have the same data type.")
+
+        height, width, _ = image_A.shape
+        morphed_images = []
+
+        for i in range(self.num_frames + 1):
+            alpha = i / self.num_frames
+            beta = 1 - alpha
+            morphed_image = cv2.addWeighted(image_A, alpha, image_B, beta, 0)
+            morphed_images.append(morphed_image)
+
+        return morphed_images
+
+    def create_mp4(self, images, output_path="output.mp4"):
+        """Creates an MP4 video from a list of images."""
+        height, width, _ = images[0].shape
+        fourcc = cv2.VideoWriter_fourcc(*'mp4v')
+        video_writer = cv2.VideoWriter(output_path, fourcc, self.fps, (width, height))
+
+        for image in images:
+            video_writer.write(image)
+
+        video_writer.release()
+
+    def morph_image_chain(self, images):
+        """Morphs a chain of images and returns the video path."""
+        morphed_images = []
+        for i in range(len(images) - 1):
+            morphed_images.extend(self.morph_images(images[i], images[i + 1]))
+        self.create_mp4(morphed_images)
+        return "output.mp4"
+
+# Create an instance of the ImageMorpher class
+morpher = ImageMorpher()
+
+# Define the Gradio interface
+iface = gr.Interface(
+    fn=morpher.morph_image_chain,
+    inputs=gr.Image(label="Images", type="numpy", image_mode="RGB", tool="select", multiple=True),
+    outputs=gr.Video(label="Output Video"),
+    title="Image Chain Morphing App",
+)
+
+iface.launch()
+'''
+Explanation:
+
+ImageMorpher class:
+
+Encapsulates the morphing logic and video creation.
+Has num_frames and fps attributes for customization.
+morph_images method handles morphing two images.
+create_mp4 method creates the MP4 video.
+morph_image_chain method:
+Iterates through the input image list and morphs each pair.
+Creates the final MP4 video using create_mp4.
+Returns the path to the video file.
+
+Interface:
+    inputs: Changed to gr.Image with multiple=True for multi-image input.
+    outputs: Changed to gr.Video to display the video directly within the interface.
+    
+    The fn is set to morpher.morph_image_chain to handle the morphing process.
+
+How to use:
+    Select two or more images using the "select" tool in the "Images" input field.
+    Click "Submit".
+    The "Output Video" block will display the morphed video, which you can watch and download.
+'''