Update app.py

app.py

@@ -3,11 +3,12 @@ import torch
 import os
 import uuid
 from diffusers import AnimateDiffPipeline, EulerDiscreteScheduler
-from diffusers.utils import export_to_video
 from huggingface_hub import hf_hub_download
 from safetensors.torch import load_file
+from moviepy.editor import ImageSequenceClip
 import logging
 import importlib.util
+import numpy as np
 
 # Thiết lập logging
 logging.basicConfig(level=logging.INFO)
@@ -15,13 +16,11 @@ logger = logging.getLogger(__name__)
 
 # Kiểm tra backend
 def check_backend():
-    if importlib.util.find_spec("imageio") and importlib.util.find_spec("imageio_ffmpeg"):
-        logger.info("imageio and imageio-ffmpeg are installed. Using recommended backend.")
-    elif importlib.util.find_spec("cv2"):
-        logger.warning("imageio/imageio-ffmpeg not found. Using legacy OpenCV backend.")
+    if importlib.util.find_spec("moviepy"):
+        logger.info("moviepy is installed. Using moviepy for video export.")
     else:
-        logger.error("No suitable backend found for export_to_video. Please install imageio and imageio-ffmpeg.")
-        raise ImportError("Missing backend for export_to_video. Install with: pip install imageio imageio-ffmpeg")
+        logger.error("moviepy is required for video export. Install with: pip install moviepy")
+        raise ImportError("Missing moviepy. Install with: pip install moviepy")
 
 check_backend()
 
@@ -95,17 +94,35 @@ def generate_image(prompt, base="Realistic", motion="", step=1, progress=gr.Prog
                 prompt=prompt,
                 guidance_scale=1.2,
                 num_inference_steps=step,
+                num_frames=32,  # Tạo 32 khung hình (sẽ lặp lại để đạt 192)
                 callback=progress_callback,
                 callback_steps=1,
                 width=256,
                 height=256
             )
 
-        # Xuất video
+        # Chuẩn bị khung hình
+        frames = output.frames[0]  # frames là list các numpy arrays
+        target_fps = 24
+        target_duration = 8  # 8 giây
+        target_frames = target_fps * target_duration  # 192 khung hình
+
+        # Lặp lại khung hình để đạt 192
+        current_frames = len(frames)
+        if current_frames < target_frames:
+            repeat_factor = (target_frames + current_frames - 1) // current_frames
+            frames = np.repeat(frames, repeat_factor, axis=0)[:target_frames]
+        elif current_frames > target_frames:
+            frames = frames[:target_frames]
+
+        logger.info(f"Generated {len(frames)} frames for video.")
+
+        # Xuất video với moviepy
         name = str(uuid.uuid4()).replace("-", "")
         path = os.path.join(output_dir, f"{name}.mp4")
         logger.info(f"Saving video to {path}")
-        export_to_video(output.frames[0], path, fps=10)
+        clip = ImageSequenceClip(list(frames), fps=target_fps)
+        clip.write_videofile(path, codec="libx264", audio=False)
 
         # Kiểm tra file video
         if not os.path.exists(path):