app.py

from fastapi import FastAPI, File, UploadFile, HTTPException, Form
from fastapi.responses import Response
import tempfile
import os
import traceback
import sys
from pathlib import Path
from typing import Optional, List

from modules.video_processor import generate_video
from modules.lut import LUTS
from modules.particles import EFFECTS
from modules.titles import TITLE_STYLES, TITLE_ANIMATIONS, TITLE_POSITIONS

app = FastAPI(title="Video Generator with Effects API")

@app.get("/")
async def root():
    return {
        "message": "Video Generator API",
        "effects": EFFECTS,
        "luts": {k: v["name"] for k, v in LUTS.items()},
        "title_styles": TITLE_STYLES,
        "title_animations": TITLE_ANIMATIONS,
        "title_positions": TITLE_POSITIONS,
        "features": ["ken_burns", "typewriter", "cinemascope", "title", "video_input"],
    }

@app.get("/health")
async def health():
    return {"status": "healthy"}

@app.post("/generate")
async def generate(
    images: List[UploadFile] = File(None),
    video: UploadFile = File(None),
    audio: UploadFile = File(...),
    srt_text: str = Form(""),
    bgm: Optional[UploadFile] = File(None),
    effect: str = Form("snow"),
    lut: str = Form("none"),
    ken_burns: bool = Form(False),
    typewriter: bool = Form(False),
    cinemascope: bool = Form(False),
    title_text: str = Form(""),
    title_style: str = Form("3d"),
    title_color: str = Form("white"),
    title_animation: str = Form("fade_in"),
    title_duration: float = Form(4.0),
    title_position: str = Form("center"),
    title_start: float = Form(0.0),
    title_font_size: int = Form(0),
    typewriter_font_size: int = Form(0),
    typewriter_color: str = Form("white"),
    subtitle_color: str = Form("white"),
    subtitle_effect: str = Form("none"),
    fps: int = Form(25),
):
    # 验证
    if not images and not video:
        raise HTTPException(400, "必须提供 images 或 video 中的一个")
    if images and video:
        raise HTTPException(400, "不能同时提供 images 和 video")
    
    # 创建临时目录
    tmp_dir = tempfile.mkdtemp()
    try:
        # 保存文件
        image_paths = None
        video_path = None
        if images:
            image_paths = []
            for i, img in enumerate(images):
                suffix = Path(img.filename).suffix if img.filename else ".jpg"
                path = os.path.join(tmp_dir, f"image_{i}{suffix}")
                with open(path, "wb") as f:
                    f.write(await img.read())
                image_paths.append(path)
        elif video:
            suffix = Path(video.filename).suffix if video.filename else ".mp4"
            video_path = os.path.join(tmp_dir, f"video{suffix}")
            with open(video_path, "wb") as f:
                f.write(await video.read())

        audio_path = os.path.join(tmp_dir, "audio" + (Path(audio.filename).suffix or ".wav"))
        with open(audio_path, "wb") as f:
            f.write(await audio.read())

        bgm_path = None
        if bgm:
            bgm_path = os.path.join(tmp_dir, "bgm" + (Path(bgm.filename).suffix or ".mp3"))
            with open(bgm_path, "wb") as f:
                f.write(await bgm.read())

        output_path = os.path.join(tmp_dir, "output.mp4")
        
        # 调用主合成函数
        generate_video(
            image_paths=image_paths,
            video_path=video_path,
            audio_path=audio_path,
            srt_text=srt_text,
            bgm_path=bgm_path,
            effect=effect,
            output_path=output_path,
            lut=lut,
            ken_burns=ken_burns,
            typewriter=typewriter,
            cinemascope=cinemascope,
            title_text=title_text,
            title_style=title_style,
            title_color=title_color,
            title_animation=title_animation,
            title_duration=title_duration,
            title_position=title_position,
            title_start=title_start,
            title_font_size=title_font_size,
            typewriter_font_size=typewriter_font_size,
            typewriter_color=typewriter_color,
            subtitle_color=subtitle_color,
            subtitle_effect=subtitle_effect,
            fps=fps,
        )

        with open(output_path, "rb") as f:
            video_bytes = f.read()

        import shutil
        shutil.rmtree(tmp_dir, ignore_errors=True)
        return Response(content=video_bytes, media_type="video/mp4",
                        headers={"Content-Disposition": "attachment; filename=output.mp4"})

    except Exception as e:
        print(traceback.format_exc(), file=sys.stderr)
        import shutil
        shutil.rmtree(tmp_dir, ignore_errors=True)
        raise HTTPException(500, detail=str(e))

if __name__ == "__main__":
    import uvicorn
    uvicorn.run(app, host="0.0.0.0", port=7860)