Update utils.py

utils.py

@@ -19,6 +19,20 @@ def get_pexels_video(query):
             return random.choice(data['videos'])['video_files'][0]['link']
     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ộ.
+    """
+    response = requests.get(url, stream=True)
+    if response.status_code == 200:
+        with open(save_path, 'wb') as f:
+            for chunk in response.iter_content(chunk_size=1024):
+                f.write(chunk)
+        return save_path
+    else:
+        logger.error(f"Failed to download video from {url}")
+        return None
+
 def get_bgm_file(bgm_type: str = "random", bgm_file: str = ""):
     """
     Lấy file nhạc nền.
@@ -36,6 +50,17 @@ 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.
+    """
+    video = VideoFileClip(video_path)
+    bgm = AudioFileClip(bgm_file).subclip(0, video.duration)
+    final_audio = CompositeAudioClip([video.audio, bgm])
+    final_video = video.set_audio(final_audio)
+    final_video.write_videofile(output_path, audio_codec="aac")
+    return output_path
+
 def combine_videos(combined_video_path: str,
                    video_paths: list[str],
                    audio_file: str,