utils.py

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