1

app.py

@@ -2,7 +2,9 @@ import os
 import cv2
 import numpy as np
 import gradio as gr
+from tqdm import tqdm
 from stability_sdk.api import Context
+from stability_sdk.utils import create_video_from_frames
 from stability_sdk.animation import AnimationArgs, Animator
 
 STABILITY_HOST = "grpc.stability.ai:443" 
@@ -103,32 +105,13 @@ def anim(f_promt, s_promt, stability_key, cadence_interp, width ,height ,sampler
         api_context=context,
         animation_prompts=animation_prompts,
         negative_prompt=negative_prompt,
-        args=args
+        args=args,
+        out_dir="video_01"
     )
+    for _ in tqdm(animator.render(), total=args.max_frames):
+        pass
 
-    # Define output folder path
-    image_path = "/tmp/frames/"
-    output_dir = os.path.join(image_path, "output")
-
-    if not os.path.exists(output_dir):
-        os.makedirs(output_dir)
-
-    # Define the codec using VideoWriter_fourcc and create a VideoWriter object
-    # We specify output file name as output.avi, codec as MJPG, fps as 25.0, and frame size as (width, height)
-    fourcc = cv2.VideoWriter_fourcc(*'mp4v')
-    video = cv2.VideoWriter('output.mp4', fourcc, 25.0, (width, height))
-
-    for idx, frame in enumerate(animator.render()):
-        # Convert the image from BGR to RGB format
-        frame = cv2.cvtColor(np.array(frame), cv2.COLOR_RGB2BGR)
-        # Write the RGB image to file
-        video.write(frame)
-        print(f"Added frame {idx} to video.")
-
-    # Release the VideoWriter
-    video.release()
-    print("Video created successfully!")
-
+    return create_video_from_frames(animator.out_dir, "video.mp4", fps=24)
 
       
 with gr.Blocks() as demo: