Manga Drama

#!/usr/bin/env python3
"""
漫剧生成器 - 基于 Seedance 的漫画风格短剧生成工具
支持图生视频，以主角图片为基础生成漫剧分镜
"""

import os
import sys
import json
import argparse
import subprocess
from pathlib import Path
from typing import List, Dict, Optional

# 添加 common 模块到路径
COMMON_DIR = Path(__file__).parent.parent.parent / "common"
sys.path.insert(0, str(COMMON_DIR))

# 导入环境变量工具
try:
    from env_utils import load_env, require_env_key
except ImportError:
    print("错误: 无法加载 env_utils 模块", file=sys.stderr)
    sys.exit(1)

# 加载环境变量
load_env()

# 默认配置
DEFAULT_MODEL = "doubao-seedance-1-5-pro-251215"
DEFAULT_RATIO = "9:16"  # 漫剧常用竖屏
DEFAULT_RESOLUTION = "1080p"
DEFAULT_DURATION = 5  # 每个分镜5秒

# 漫剧风格预设
MANGA_STYLE_PROMPT = "漫画风格，手绘质感，柔和的色彩过渡，线条清晰，日式或国漫风格，温馨治愈，电影级构图，高细节"

# 漫剧分镜模板
DRAMA_TEMPLATES = {
    "introduction": {
        "name": "主角登场",
        "description": "介绍主角，展示角色特征",
        "default_prompt": "{character}站在画面中央，微笑看向镜头，背景是柔和的光晕，漫画风格，温馨氛围，{style}"
    },
    "action": {
        "name": "动作场景",
        "description": "主角进行某个动作",
        "default_prompt": "{character}正在{action}，表情生动，动作流畅，漫画风格，{style}"
    },
    "emotion": {
        "name": "情感表达",
        "description": "表达某种情感",
        "default_prompt": "{character}露出{emotion}的表情，眼神传达情感，漫画风格，{style}"
    },
    "interaction": {
        "name": "互动场景",
        "description": "与环境或其他元素互动",
        "default_prompt": "{character}与{object}互动，场景温馨，漫画风格，{style}"
    },
    "ending": {
        "name": "结尾定格",
        "description": "漫剧结尾，定格画面",
        "default_prompt": "{character}的定格画面，{ending_scene}，漫画风格，温馨治愈，{style}"
    }
}


def generate_scene_prompt(
    template_key: str,
    character_desc: str,
    style: str = MANGA_STYLE_PROMPT,
    **kwargs
) -> str:
    """根据模板生成分镜提示词"""
    template = DRAMA_TEMPLATES.get(template_key, DRAMA_TEMPLATES["introduction"])
    prompt = template["default_prompt"].format(
        character=character_desc,
        style=style,
        **kwargs
    )
    return prompt


def create_drama_script(
    title: str,
    character_desc: str,
    scenes: List[Dict],
    output_file: str = None
) -> str:
    """
    创建漫剧脚本
    
    Args:
        title: 漫剧标题
        character_desc: 主角描述
        scenes: 分镜列表，每个分镜包含 type, prompt, duration
        output_file: 输出文件路径
    
    Returns:
        脚本内容（JSON 格式）
    """
    script = {
        "title": title,
        "character": character_desc,
        "style": "漫画风格",
        "total_scenes": len(scenes),
        "scenes": []
    }
    
    for i, scene in enumerate(scenes, 1):
        scene_data = {
            "scene_number": i,
            "type": scene.get("type", "introduction"),
            "name": DRAMA_TEMPLATES.get(scene.get("type", "introduction"), {}).get("name", "自定义场景"),
            "prompt": scene.get("prompt", ""),
            "duration": scene.get("duration", DEFAULT_DURATION),
            "ratio": scene.get("ratio", DEFAULT_RATIO),
            "resolution": scene.get("resolution", DEFAULT_RESOLUTION)
        }
        script["scenes"].append(scene_data)
    
    script_json = json.dumps(script, indent=2, ensure_ascii=False)
    
    if output_file:
        with open(output_file, 'w', encoding='utf-8') as f:
            f.write(script_json)
        print(f"✅ 脚本已保存: {output_file}")
    
    return script_json


def generate_scene_video(
    scene: Dict,
    character_image: str,
    model: str = DEFAULT_MODEL,
    output_dir: str = "~/Desktop",
    send_feishu: bool = False
) -> str:
    """
    生成分镜视频
    
    Args:
        scene: 分镜数据
        character_image: 主角图片路径
        model: 模型 ID
        output_dir: 输出目录
        send_feishu: 是否发送到飞书
    
    Returns:
        生成的视频路径
    """
    from seedance_video import generate_video_task
    
    print(f"\n🎬 生成分镜 {scene['scene_number']}: {scene['name']}")
    print(f"📝 提示词: {scene['prompt'][:80]}...")
    
    # 构建完整提示词（结合主角图片）
    full_prompt = scene['prompt']
    
    # 生成视频
    video_path = generate_video_task(
        prompt=full_prompt,
        image_path=character_image,