1

app.py

@@ -1,5 +1,6 @@
 import os
 import cv2
+import numpy as np
 import gradio as gr
 from stability_sdk.api import Context
 from stability_sdk.animation import AnimationArgs, Animator
@@ -111,28 +112,24 @@ 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 = []
+
+    # 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()):
-        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)
-
-    # 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)
+        # 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
-    out.release()
+    video.release()
+    print("Video created successfully!")
+
+
       
 with gr.Blocks() as demo:
     gr.Markdown("Stability Animation 1")