Update app.py

app.py

@@ -1,4 +1,5 @@
 import os
+import cv2
 import gradio as gr
 from stability_sdk.api import Context
 from stability_sdk.animation import AnimationArgs, Animator
@@ -111,14 +112,27 @@ def anim(f_promt, s_promt, stability_key, cadence_interp, width ,height ,sampler
     if not os.path.exists(output_dir):
         os.makedirs(output_dir)
     # Render each frame of animation
+    # Render each frame of animation
     images = []
     for idx, frame in enumerate(animator.render()):
         file_path = os.path.join(output_dir, f"frame_{idx:05d}.png") 
         frame.save(file_path)
         print("Created frame at:"+file_path)
         images.append(file_path)
-    
-    return images
+
+    # Define the codec using VideoWriter_fourcc() and create a VideoWriter object
+    fourcc = cv2.VideoWriter_fourcc(*'mp4v')  # or use 'XVID'
+    height, width, _ = cv2.imread(images[0]).shape  # get the frame size from the first image
+    video_path = os.path.join(output_dir, 'output.mp4')  # specify your output video file path here
+    out = cv2.VideoWriter(video_path, fourcc, fps, (width, height))
+
+    # Write each frame to the video file
+    for image_path in images:
+        frame = cv2.imread(image_path)
+        out.write(frame)
+
+    # Release the VideoWriter
+    out.release()
       
 with gr.Blocks() as demo:
     gr.Markdown("Stability Animation")