import logging
import os
import subprocess
from celery import shared_task
from Ansjer.config import LOGGER
import django

# 设置 Django 的环境变量
os.environ.setdefault("DJANGO_SETTINGS_MODULE", 'Ansjer.cn_config.test_settings')  # 确保替换为实际的 settings 模块路径
django.setup()


@shared_task
def generate_video(image_files, output_path):
    video_files = []
    music_path = "static/temp/music.mp3"
    # 定义特效列表
    transitions = [
        "BowTieHorizontal.glsl",
        "burn.glsl",
        "cube.glsl",
        "pinwheel.glsl",
        "windowslice.glsl",
        "Radial.glsl",
        "rotateTransition.glsl",
        "wind.glsl",
        "squeeze.glsl"
    ]

    # 生成相邻图片的视频
    for i in range(len(image_files) - 1):
        output_video = os.path.join(output_path, f'{i + 1}_{i + 2}.mp4')
        video_files.append(output_video)

        cmd = [
            'ffmpeg', '-loop', '1', '-i', image_files[i],
            '-loop', '1', '-i', image_files[i + 1],
            '-filter_complex',
            f'gltransition=duration=1:offset=0.7:source=gl-transitions/{transitions[i]}',
            '-t', '2', output_video
        ]

        subprocess.run(cmd, check=True)

    # 生成拼接列表文件
    with open(os.path.join(output_path, 'filelist.txt'), 'w') as f:
        for video in video_files:
            f.write(f"file '{video}'\n")

    # 拼接视频
    final_output = os.path.join(output_path, 'final_output.mp4')
    subprocess.run(['ffmpeg', '-f', 'concat', '-safe', '0', '-i', 'filelist.txt', '-c', 'copy', final_output],
                   check=True)

    # 添加背景音乐
    final_output_with_music = os.path.join(output_path, 'final_output_with_music.mp4')
    subprocess.run(['ffmpeg', '-i', final_output, '-i', music_path, '-c:v', 'copy', '-c:a', 'aac', '-b:a', '192k',
                    '-shortest', final_output_with_music], check=True)

    # 清理临时文件
    for video in video_files:
        os.remove(video)
    os.remove('filelist.txt')