Update app.py

app.py

@@ -23,6 +23,7 @@ import gc
 from openai import OpenAI
 import base64
 from io import BytesIO
+from moviepy.editor import ImageSequenceClip
 
 # Load Hugging Face token if needed
 hf_token = os.getenv("HF_TOKEN")
@@ -287,12 +288,10 @@ def generate_video_from_text(
     video_np = images.squeeze(0).permute(1, 2, 3, 0).cpu().float().numpy()
     video_np = (video_np * 255).astype(np.uint8)
     height, width = video_np.shape[1:3]
-    out = cv2.VideoWriter(
-        output_path, cv2.VideoWriter_fourcc(*"mp4v"), frame_rate, (width, height)
-    )
-    for frame in video_np[..., ::-1]:
-        out.write(frame)
-    out.release()
+
+    clip = ImageSequenceClip(list(video_np), fps=frame_rate)
+    clip = clip.resize(width=width, height=height)
+    clip.write_videofile(output_path, codec="libx264", audio=False)
     # Explicitly delete tensors and clear cache
     del images
     del video_np
@@ -382,12 +381,10 @@ def generate_video_from_image(
         video_np = images.squeeze(0).permute(1, 2, 3, 0).cpu().float().numpy()
         video_np = (video_np * 255).astype(np.uint8)
         height, width = video_np.shape[1:3]
-        out = cv2.VideoWriter(
-            output_path, cv2.VideoWriter_fourcc(*"mp4v"), frame_rate, (width, height)
-        )
-        for frame in video_np[..., ::-1]:
-            out.write(frame)
-        out.release()
+    
+        clip = ImageSequenceClip(list(video_np), fps=frame_rate)
+        clip = clip.resize(width=width, height=height)
+        clip.write_videofile(output_path, codec="libx264", audio=False)
     except Exception as e:
         raise gr.Error(
             f"An error occurred while generating the video. Please try again. Error: {e}",