Update utils.py

utils.py

@@ -1,25 +1,22 @@
 import glob
 import os
 import random
-
 import requests
 from loguru import logger
 from moviepy.editor import AudioFileClip, VideoFileClip, concatenate_videoclips
 
-def get_pexels_image(query):
+def get_pexels_video(query):
     """
-    Lấy ảnh từ Pexels API dựa trên query.
+    Lấy video từ Pexels API dựa trên query.
     """
     api_key = os.getenv('Pexels_API_KEY')
-    # Thêm từ khóa "Vietnamese" vào truy vấn và tăng số lượng ảnh lên 30
-    url = f"https://api.pexels.com/v1/search?query={query}%20Vietnamese&per_page=30"
+    url = f"https://api.pexels.com/videos/search?query={query}&per_page=30"
     headers = {"Authorization": api_key}
     response = requests.get(url, headers=headers)
     if response.status_code == 200:
         data = response.json()
-        if data['photos']:
-            # Chọn ngẫu nhiên một ảnh từ kết quả
-            return random.choice(data['photos'])['src']['medium']
+        if data['videos']:
+            return random.choice(data['videos'])['video_files'][0]['link']
     return None
 
 def get_bgm_file(bgm_type: str = "random", bgm_file: str = ""):
@@ -30,7 +27,7 @@ def get_bgm_file(bgm_type: str = "random", bgm_file: str = ""):
         return ""
     if bgm_type == "random":
         suffix = "*.mp3"
-        song_dir = utils.song_dir()  # Đảm bảo utils.song_dir() trả về đường dẫn đúng
+        song_dir = "/data/bg-music"  # Thay thế bằng đường dẫn thực tế
         files = glob.glob(os.path.join(song_dir, suffix))
         return random.choice(files) if files else ""
 
@@ -60,6 +57,7 @@ def combine_videos(combined_video_path: str,
 
         for video_path in video_paths:
             clip = VideoFileClip(video_path).without_audio()
+            # Cắt video nếu dài hơn thời lượng tối đa
             if (audio_duration - video_duration) < clip.duration:
                 clip = clip.subclip(0, (audio_duration - video_duration))
             elif max_clip_duration < clip.duration:
@@ -87,7 +85,6 @@ def generate_video(video_paths: list[str],
 
     combine_videos(combined_video_path, video_paths, audio_path)
 
-    # Add audio to the final video
     final_video = VideoFileClip(combined_video_path)
     audio_clip = AudioFileClip(audio_path)
     final_video = final_video.set_audio(audio_clip)
@@ -95,7 +92,6 @@ def generate_video(video_paths: list[str],
     logger.info(f"Writing final video to {output_file}")
     final_video.write_videofile(output_file, audio_codec="aac")
 
-    # Remove temporary video
     os.remove(combined_video_path)
     logger.success(f"Completed generating video: {output_file}")
     return output_file