Upload utils.py

utils.py

@@ -5,10 +5,17 @@ import requests
 from loguru import logger
 from moviepy.editor import VideoFileClip, concatenate_videoclips, AudioFileClip, CompositeAudioClip
 
+def check_file_exists(file_path):
+    if not os.path.exists(file_path):
+        raise FileNotFoundError(f"Tệp {file_path} không tồn tại.")
+
 # Đường dẫn đầy đủ tới các file
-video_file = "input/video.mp4"  # Đảm bảo có tên file và định dạng
-bgm_file = "data/bg_music.mp3"  # Đảm bảo có tên file và định dạng
-output_path = "output/final_video.mp4"  # Đảm bảo có tên file và định dạng
+video_file = "input/video.mp4"
+bgm_file = "data/bg_music.mp3"
+output_path = "output/final_video.mp4"
+
+check_file_exists(video_file)
+check_file_exists(bgm_file)
 
 video = VideoFileClip(video_file)
 bgm = AudioFileClip(bgm_file).subclip(0, video.duration)
@@ -17,10 +24,10 @@ final_video = video.set_audio(final_audio)
 final_video.write_videofile(output_path, audio_codec="aac")
 
 def get_pexels_video(query):
-    """
-    Lấy video từ Pexels API dựa trên query.
-    """
     api_key = os.getenv('Pexels_API_KEY')
+    if not api_key:
+        raise ValueError("API key không được tìm thấy trong biến môi trường.")
+    
     url = f"https://api.pexels.com/videos/search?query={query}&per_page=30"
     headers = {"Authorization": api_key}
     response = requests.get(url, headers=headers)
@@ -28,12 +35,10 @@ def get_pexels_video(query):
         data = response.json()
         if data['videos']:
             return random.choice(data['videos'])['video_files'][0]['link']
+    logger.error("Không tìm thấy video nào.")
     return None
 
 def download_video(url, save_path="downloaded_video.mp4"):
-    """
-    Tải video từ URL và lưu vào tệp cục bộ.
-    """
     try:
         response = requests.get(url, stream=True)
         response.raise_for_status()
@@ -46,14 +51,11 @@ def download_video(url, save_path="downloaded_video.mp4"):
         return None
 
 def get_bgm_file(bgm_type: str = "random", bgm_file: str = ""):
-    """
-    Lấy file nhạc nền.
-    """
     if not bgm_type:
         return ""
     if bgm_type == "random":
         suffix = "*.mp3"
-        song_dir = "/data/bg-music"  # Thay thế bằng đường dẫn thực tế
+        song_dir = "/data/bg-music"
         files = glob.glob(os.path.join(song_dir, suffix))
         return random.choice(files) if files else ""
 
@@ -63,9 +65,9 @@ def get_bgm_file(bgm_type: str = "random", bgm_file: str = ""):
     return ""
 
 def add_background_music(video_path, bgm_file, output_path="video_with_bgm.mp4"):
-    """
-    Thêm nhạc nền vào video.
-    """
+    check_file_exists(video_path)
+    check_file_exists(bgm_file)
+
     video = VideoFileClip(video_path)
     bgm = AudioFileClip(bgm_file).subclip(0, video.duration)
     final_audio = CompositeAudioClip([video.audio, bgm])
@@ -79,9 +81,8 @@ def combine_videos(combined_video_path: str,
                    max_clip_duration: int = 5,
                    threads: int = 2,
                    ) -> str:
-    """
-    Kết hợp nhiều video thành một video duy nhất.
-    """
+    check_file_exists(audio_file)
+
     audio_clip = AudioFileClip(audio_file)
     audio_duration = audio_clip.duration
     logger.info(f"Max duration of audio: {audio_duration} seconds")
@@ -93,8 +94,8 @@ def combine_videos(combined_video_path: str,
         random.shuffle(video_paths)
 
         for video_path in video_paths:
+            check_file_exists(video_path)
             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:
@@ -114,9 +115,6 @@ def generate_video(video_paths: list[str],
                    audio_path: str,
                    output_file: str,
                    ) -> str:
-    """
-    Tạo video cuối cùng bằng cách kết hợp video và âm thanh.
-    """
     logger.info(f"Start generating video")
     combined_video_path = "temp_combined_video.mp4"