from moviepy.editor import ImageSequenceClip, AudioFileClip
from PIL import Image
import numpy as np
import os

def resize_images(image_folder, target_size=(1280, 720)):
    """統(tǒng)一縮放文件夾內(nèi)所有圖片"""
    resized_images = []
    for img_name in sorted(os.listdir(image_folder)):
        if img_name.lower().endswith(('.jpg', '.png')):
            img_path = os.path.join(image_folder, img_name)
            try:
                with Image.open(img_path) as img:
                    print(f"{img_name} - 平均亮度: {np.array(img).mean()}")
                    img = img.resize(target_size, Image.LANCZOS)
                    resized_images.append(np.array(img.copy()))  # ?? 返回 np.array
            except Exception as e:
                print(f"? 圖像讀取失敗：{img_path}, 錯誤信息: {e}")
    return resized_images

def images_to_video(image_arrays, duration_per_image, fps, audio_path=None, output_name="output.mp4"):
    """
    使用 ImageSequenceClip(fps=...) 嚴格控制幀率與幀數(shù)
    每張圖顯示 duration 秒
    """
    # 計算重復(fù)每張圖片的幀數(shù)
    frames_per_image = int(duration_per_image * fps)

    # 重復(fù)圖片，使得每張圖片持續(xù)多幀
    repeated_frames = []
    for img in image_arrays:
        repeated_frames.extend([img] * frames_per_image)

    # 生成視頻剪輯
    clip = ImageSequenceClip(repeated_frames, fps=fps)

    # 寫出視頻
    clip.write_videofile(output_name, fps=fps)

    # 添加背景音樂（可選）
    if audio_path and os.path.exists(audio_path):
        print("?? 添加背景音樂...")
        audio = AudioFileClip(audio_path).set_duration(clip.duration)
        final_clip = clip.set_audio(audio)
        final_clip.write_videofile("final_" + output_name, fps=fps)
    else:
        print("?? 未添加背景音樂（未找到或未提供）")

# 參數(shù)配置
image_folder = 'image_test/'
fps = 1
duration = 5
audio_path = 'bgm.mp3'

# 執(zhí)行處理
resized_images = resize_images(image_folder, target_size=(1280, 720))

if resized_images:
    images_to_video(
        image_arrays=resized_images,
        duration_per_image=duration,
        fps=fps,
        audio_path=audio_path,
        output_name="output.mp4"
    )
else:
    print("?? 沒有可用的圖片，視頻未生成。")