File size: 3,504 Bytes
1e06115 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 |
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):
"""
Lấy ảnh 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"
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']
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 = utils.song_dir() # Đảm bảo utils.song_dir() trả về đường dẫn đúng
files = glob.glob(os.path.join(song_dir, suffix))
return random.choice(files) if files else ""
if os.path.exists(bgm_file):
return bgm_file
return ""
def combine_videos(combined_video_path: str,
video_paths: list[str],
audio_file: 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.
"""
audio_clip = AudioFileClip(audio_file)
audio_duration = audio_clip.duration
logger.info(f"Max duration of audio: {audio_duration} seconds")
clips = []
video_duration = 0
while video_duration < audio_duration:
random.shuffle(video_paths)
for video_path in video_paths:
clip = VideoFileClip(video_path).without_audio()
if (audio_duration - video_duration) < clip.duration:
clip = clip.subclip(0, (audio_duration - video_duration))
elif max_clip_duration < clip.duration:
clip = clip.subclip(0, max_clip_duration)
clip = clip.set_fps(30)
clips.append(clip)
video_duration += clip.duration
final_clip = concatenate_videoclips(clips)
final_clip = final_clip.set_fps(30)
logger.info(f"Writing combined video to {combined_video_path}")
final_clip.write_videofile(combined_video_path, threads=threads)
logger.success(f"Completed combining videos")
return combined_video_path
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"
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)
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 |