@fifine/aim-studio 0.0.2 → 0.0.3

package/dist/templates/aim/scripts/export.py

@@ -0,0 +1,427 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+Export script for generating Seedance-ready prompts
+
+Usage:
+    python export.py --ep 1                    # Export episode 1
+    python export.py --ep 1-3                   # Export episodes 1-3
+    python export.py --all                      # Export all episodes
+    python export.py --scene 1                  # Export scene 1 of current episode
+    python export.py --format seedance          # Seedance format (default)
+    python export.py --format simple             # Simple format for quick copy
+"""
+
+import argparse
+import json
+import os
+import sys
+from pathlib import Path
+from datetime import datetime
+
+# Add parent directory to path for imports
+sys.path.insert(0, str(Path(__file__).parent))
+from common.paths import get_project_root, get_tasks_dir
+
+
+def read_file(file_path: Path) -> str:
+    """Read file content."""
+    try:
+        return file_path.read_text(encoding="utf-8")
+    except (FileNotFoundError, PermissionError):
+        return ""
+
+
+def get_character_prompt(character_name: str, spec_dir: Path) -> str:
+    """Generate character prompt for Seedance."""
+    character_file = spec_dir / "story" / "character.md"
+    if not character_file.exists():
+        return ""
+
+    content = read_file(character_file)
+
+    # Extract character section
+    lines = content.split("\n")
+    in_character = False
+    character_lines = []
+
+    for line in lines:
+        if f"## {character_name}" in line or f"## {character_name}：" in line:
+            in_character = True
+            continue
+        if in_character:
+            if line.startswith("## "):
+                break
+            character_lines.append(line)
+
+    if character_lines:
+        # Extract key information: appearance, personality, clothing
+        prompt_parts = []
+
+        # Extract appearance
+        for line in character_lines:
+            if "外观" in line or "外貌" in line or "Appearance" in line:
+                prompt_parts.append(line.split(":", 1)[-1].strip())
+
+            if "服装" in line or "Clothing" in line:
+                prompt_parts.append(line.split(":", 1)[-1].strip())
+
+        return ", ".join(prompt_parts) if prompt_parts else ""
+
+    return ""
+
+
+def get_world_prompt(spec_dir: Path) -> str:
+    """Generate world prompt for Seedance."""
+    world_file = spec_dir / "story" / "world.md"
+    if not world_file.exists():
+        return ""
+
+    content = read_file(world_file)
+
+    # Extract style keywords
+    keywords = []
+
+    # Look for style/tone keywords
+    for line in content.split("\n"):
+        if "风格" in line or "Style" in line or "视觉" in line or "Visual" in line:
+            # Extract keywords after colon
+            if ":" in line:
+                keywords.append(line.split(":", 1)[-1].strip())
+
+    return ", ".join(keywords)[:200] if keywords else "cinematic, high quality"
+
+
+def parse_scene_file(scene_path: Path) -> dict:
+    """Parse scene file and extract Seedance-relevant content."""
+    content = read_file(scene_path)
+
+    if not content:
+        return {}
+
+    scene_data = {
+        "content": content,
+        "has_seedance_format": False,
+    }
+
+    # Check for Seedance 2.0 format (六要素)
+    required_elements = ["主体描述", "动作", "环境", "镜头", "音频", "风格"]
+    if any(elem in content for elem in required_elements):
+        scene_data["has_seedance_format"] = True
+
+    return scene_data
+
+
+def generate_seedance_prompt(
+    scene_content: str,
+    character_prompts: dict,
+    world_prompt: str,
+    scene_name: str,
+    previous_context: str = "",
+) -> str:
+    """
+    Generate a clean, copy-paste ready prompt for Seedance.
+
+    Format: Pure text, no markdown, ready for video generation.
+    """
+
+    # Extract key information from scene content
+    lines = scene_content.split("\n")
+
+    output = []
+
+    # Scene header
+    output.append(f"=== {scene_name} ===")
+    output.append("")
+
+    # Character prompts
+    if character_prompts:
+        output.append("[CHARACTERS]")
+        for name, prompt in character_prompts.items():
+            if prompt:
+                output.append(f"{name}: {prompt}")
+        output.append("")
+
+    # Environment
+    output.append("[ENVIRONMENT]")
+    # Extract environment description
+    for line in lines:
+        if "环境" in line or "场景" in line:
+            env_text = line.split(":", 1)[-1].strip() if ":" in line else line
+            if env_text:
+                output.append(env_text)
+                break
+    output.append("")
+
+    # Previous context
+    if previous_context:
+        output.append("[PREVIOUS CONTEXT]")
+        output.append(previous_context[:200])
+        output.append("")
+
+    # Main content - extract the actual scene description
+    output.append("[SCENE]")
+
+    # Extract main scene content (skip headers and metadata)
+    in_main_content = False
+    for line in lines:
+        # Skip metadata lines
+        if any(marker in line for marker in ["#", "##", "---", "**", "日期", "时间", "时长"]):
+            continue
+
+        # Skip empty lines at start
+        if not in_main_content and not line.strip():
+            continue
+
+        # Start main content
+        if line.strip() and not in_main_content:
+            in_main_content = True
+
+        if in_main_content:
+            # Clean the line - remove markdown
+            cleaned_line = line.strip()
+            cleaned_line = cleaned_line.replace("**", "").replace("*", "")
+            cleaned_line = cleaned_line.replace("[", "").replace("]", "")
+
+            if cleaned_line:
+                output.append(cleaned_line)
+
+    output.append("")
+
+    # Style prompt
+    if world_prompt:
+        output.append("[STYLE]")
+        output.append(world_prompt)
+
+    # Return as pure text
+    return "\n".join(output)
+
+
+def generate_simple_prompt(scene_content: str, scene_name: str) -> str:
+    """
+    Generate a simple, minimal prompt for quick copy.
+    Just the essential scene description.
+    """
+
+    lines = scene_content.split("\n")
+    output = []
+
+    output.append(f"Scene: {scene_name}")
+    output.append("")
+
+    # Extract just the dialogue and action
+    for line in lines:
+        # Skip headers and metadata
+        if line.startswith("#") or line.startswith("##"):
+            continue
+        if "日期" in line or "时间" in line or "时长" in line:
+            continue
+        if "主体描述" in line or "动作" in line:
+            continue
+
+        # Clean and add
+        cleaned = line.strip().replace("**", "").replace("*", "")
+        if cleaned:
+            output.append(cleaned)
+
+    return "\n".join(output)
+
+
+def find_episodes(tasks_dir: Path) -> list:
+    """Find all episode directories."""
+    episodes = []
+
+    if not tasks_dir.exists():
+        return episodes
+
+    for item in tasks_dir.iterdir():
+        if item.is_dir() and "EP" in item.name.upper():
+            episodes.append((item.name, item))
+
+    # Sort by episode number
+    episodes.sort(key=lambda x: x[0])
+    return episodes
+
+
+def find_scenes(episode_dir: Path) -> list:
+    """Find all scene files in an episode directory."""
+    scenes = []
+
+    if not episode_dir.exists():
+        return scenes
+
+    for item in episode_dir.iterdir():
+        if item.is_file() and (item.suffix == ".md" or item.suffix == ".txt"):
+            if "task.json" in item.name or "prd" in item.name:
+                continue
+            scenes.append(item)
+
+    scenes.sort(key=lambda x: x.name)
+    return scenes
+
+
+def export_episode(
+    episode_name: str,
+    episode_dir: Path,
+    spec_dir: Path,
+    output_dir: Path,
+    format_type: str = "seedance",
+) -> list:
+    """Export a single episode."""
+
+    # Get character and world prompts
+    world_prompt = get_world_prompt(spec_dir)
+
+    # Find scenes
+    scenes = find_scenes(episode_dir)
+
+    output_files = []
+
+    for scene_file in scenes:
+        scene_name = scene_file.stem
+        content = read_file(scene_file)
+
+        # Get character prompts for this scene (simplified - use all characters)
+        character_prompts = {}
+        character_file = spec_dir / "story" / "character.md"
+        if character_file.exists():
+            char_content = read_file(character_file)
+            # Extract character names from ## headers
+            for line in char_content.split("\n"):
+                if line.startswith("## ") and not line.startswith("###"):
+                    char_name = line.replace("##", "").strip()
+                    # Generate prompt for this character
+                    char_prompt = get_character_prompt(char_name, spec_dir)
+                    if char_prompt:
+                        character_prompts[char_name] = char_prompt
+
+        # Generate prompt based on format
+        if format_type == "simple":
+            prompt = generate_simple_prompt(content, scene_name)
+        else:
+            prompt = generate_seedance_prompt(
+                content,
+                character_prompts,
+                world_prompt,
+                scene_name,
+            )
+
+        # Write to output directory
+        output_file = output_dir / f"{episode_name}_{scene_name}.txt"
+        output_file.write_text(prompt, encoding="utf-8")
+        output_files.append(output_file)
+
+    return output_files
+
+
+def main():
+    parser = argparse.ArgumentParser(
+        description="Export script for Seedance video generation"
+    )
+    parser.add_argument(
+        "--ep",
+        type=str,
+        help="Episode number(s). Examples: 1, 1-3, all",
+    )
+    parser.add_argument(
+        "--scene",
+        type=int,
+        help="Scene number within episode",
+    )
+    parser.add_argument(
+        "--format",
+        type=str,
+        default="seedance",
+        choices=["seedance", "simple"],
+        help="Export format: seedance (full) or simple (minimal)",
+    )
+    parser.add_argument(
+        "--output",
+        type=str,
+        default="export",
+        help="Output directory name",
+    )
+    parser.add_argument(
+        "--open",
+        action="store_true",
+        help="Open output directory after export",
+    )
+
+    args = parser.parse_args()
+
+    # Get project paths
+    project_root = get_project_root()
+    tasks_dir = get_tasks_dir(project_root)
+    spec_dir = project_root / ".aim-studio" / "spec"
+
+    # Check if story project
+    if not (spec_dir / "story").exists():
+        print("Error: This is not a story project. No spec/story/ directory found.")
+        sys.exit(1)
+
+    # Create output directory
+    output_dir = project_root / args.output
+    output_dir.mkdir(exist_ok=True)
+
+    print(f"Exporting to: {output_dir}")
+    print(f"Format: {args.format}")
+    print("")
+
+    exported_files = []
+
+    # Determine which episodes to export
+    if args.ep == "all":
+        episodes = find_episodes(tasks_dir)
+        for ep_name, ep_dir in episodes:
+            print(f"Exporting {ep_name}...")
+            files = export_episode(ep_name, ep_dir, spec_dir, output_dir, args.format)
+            exported_files.extend(files)
+    elif "-" in str(args.ep):
+        # Range like 1-3
+        start, end = map(int, args.ep.split("-"))
+        for ep_num in range(start, end + 1):
+            ep_name = f"EP{ep_num:02d}"
+            ep_dir = tasks_dir / ep_name
+            if ep_dir.exists():
+                print(f"Exporting {ep_name}...")
+                files = export_episode(ep_name, ep_dir, spec_dir, output_dir, args.format)
+                exported_files.extend(files)
+    elif args.ep:
+        # Single episode
+        ep_num = int(args.ep)
+        ep_name = f"EP{ep_num:02d}"
+        ep_dir = tasks_dir / ep_name
+
+        if ep_dir.exists():
+            print(f"Exporting {ep_name}...")
+            exported_files = export_episode(ep_name, ep_dir, spec_dir, output_dir, args.format)
+        else:
+            print(f"Error: Episode {ep_name} not found in {tasks_dir}")
+            sys.exit(1)
+    else:
+        print("Error: Please specify --ep or --all")
+        parser.print_help()
+        sys.exit(1)
+
+    print("")
+    print(f"Exported {len(exported_files)} files:")
+    for f in exported_files:
+        print(f"  - {f.relative_to(project_root)}")
+
+    # Optionally open output directory
+    if args.open:
+        import subprocess
+
+        if sys.platform == "win32":
+            os.startfile(output_dir)
+        elif sys.platform == "darwin":
+            subprocess.run(["open", str(output_dir)])
+        else:
+            subprocess.run(["xdg-open", str(output_dir)])
+
+    print("")
+    print("Done! Files are ready for Seedance.")
+
+
+if __name__ == "__main__":
+    main()

package/dist/templates/claude/commands/aim/export.md

@@ -1,187 +1,161 @@
 ---
 name: export
-description: 导出剧本用于AI视频生成
+description: 导出剧本用于AI视频生成（纯文本格式）
 ---
 
 # 导出剧本
 
-将剧本导出为可直接粘贴到 AI 视频生成工具（如 Seedance）的格式。
+将剧本导出为可直接粘贴到 AI 视频生成工具（如 Seedance）的**纯文本格式**。
 
-## 导出流程
+---
+
+## 核心功能
+
+本命令生成**不带任何 Markdown 符号**的纯文本文件，可直接复制粘贴到 Seedance 的视频生成输入框。
+
+---
 
-### 1. 选择导出范围
+## 使用方法
+
+### 基本用法
 
 ```bash
-# 导出单集
-/aim:export --ep 1
+# 导出第1集
+python3 .aim-studio/scripts/export.py --ep 1
 
 # 导出多集
-/aim:export --ep 1-5
+python3 .aim-studio/scripts/export.py --ep 1-3
+
+# 导出全部集数
+python3 .aim-studio/scripts/export.py --all
 
-# 导出全部
-/aim:export --all
+# 导出后打开文件夹
+python3 .aim-studio/scripts/export.py --ep 1 --open
 ```
 
-### 2. 选择导出格式
+### 导出格式
 
-| 格式 | 用途 | 说明 |
+| 格式 | 说明 | 用途 |
 |------|------|------|
-| `seedance` | Seedance Web 端 | 单场次独立 prompt，包含上下文 |
-| `batch` | 批量生成 | 所有场次合并，适合批量处理 |
-| `review` | 预览检查 | 带注释格式，便于检查 |
-| `text` | 纯文本发布 | 去除备注，统一标点 |
+| `seedance` | 完整格式，包含角色、环境、上下文（默认） | 推荐用于 Seedance |
+| `simple` | 极简格式，只有场景描述和对话 | 快速复制 |
 
-### 3. 导出内容结构
+```bash
+# 使用简单格式
+python3 .aim-studio/scripts/export.py --ep 1 --format simple
+```
 
-每次导出会生成以下文件：
+---
+
+## 输出文件
+
+导出会生成纯文本文件到 `export/` 目录：
 
 ```
 export/
-├── EP01/
-│   ├── scene_01.txt      # 场次1 - 可直接粘贴
-│   ├── scene_02.txt      # 场次2 - 可直接粘贴
-│   ├── context.txt       # 本集上下文摘要
-│   └── characters.txt    # 本集角色参考
-├── EP02/
-│   └── ...
-└── global_context.txt    # 全局设定（角色、世界观）
+├── EP01_场景1.txt      # 可直接粘贴
+├── EP01_场景2.txt      # 可直接粘贴
+├── EP02_场景1.txt
+└── ...
 ```
 
 ---
 
-## Seedance 导出格式
-
-### 单场次 Prompt 模板
+## Seedance 纯文本格式
 
-每个场次文件包含完整的上下文信息，可直接粘贴：
+导出的文件内容示例（无任何 Markdown 符号）：
 
 ```
-【角色设定】
-[角色名]：[外貌描述]，[性格关键词]
-（如有多位角色，逐一列出）
+=== 场景1: 青云峰主殿 ===
 
-【场景设定】
-[场景名称]：[环境描述]，[氛围]
+[CHARACTERS]
+沈安在: 白衣中年男子, 两鬓霜白, 面容俊俏
+慕容天: 少年身形, 目光坚定
 
-【前情提要】（如适用）
-[上一场次的关键事件，1-2句话]
+[ENVIRONMENT]
+青云峰主殿: 破旧大殿, 阳光洒落, 萧条冷清
 
-【本场次剧本】
-场次 X：[场景] - [时间] - [气氛] - [时长]
+[PREVIOUS CONTEXT]
+慕容天决定下山离开
 
-[画面描述]
+[SCENE]
+场次 1: 青云峰主殿内 - 清晨 - 萧条落寞 - 30秒
 
-[角色名]：(情绪) 台词
+白衣中年沈安在瘫坐在太师椅上，神情苦涩。
 
-[画面描述]
+慕容天：(低沉) 师父，弟子不孝，今日便要下山！
 
-...
-
-【视觉风格】
-[风格关键词，如：xianxia, fantasy, cinematic lighting]
-```
-
-### 示例输出
+沈安在听到声音，却懒得出门。镜头推进他的脸，苦涩中带着自嘲。
 
+[STYLE]
+xianxia, chinese fantasy, cinematic lighting, melancholic atmosphere
 ```
-【角色设定】
-沈安在：白衣中年男子，两鬓霜白，面容俊俏，气质高深莫测。性格：腹黑、护短、爱演。
-慕容天：少年身形，目光从落寞到坚定。性格：勤奋、尊师重道、意志坚定。
 
-【场景设定】
-青云峰主殿：破旧的大殿，阳光透过残破窗棂洒落，萧条冷清的氛围。
+---
 
-【前情提要】
-慕容天决定下山离开，沈安在内心吐槽自己的穿越遭遇。
+## 快速使用流程
 
-【本场次剧本】
-场次 1：青云峰主殿内 - 清晨 - 萧条落寞 - 30秒
+### 步骤 1：导出
 
-白衣中年沈安在瘫坐在太师椅上，神情苦涩。殿外，少年慕容天背着大包小包，站在门口。
+```bash
+python3 .aim-studio/scripts/export.py --ep 1 --open
+```
 
-慕容天：(低沉) 师父，弟子不孝，今日便要下山继承家业，还望师父珍重！
+### 步骤 2：复制
 
-沈安在听到声音，却懒得出门。镜头推进他的脸，苦涩中带着自嘲。
+打开 `export/` 文件夹，选择对应场次的 `.txt` 文件，**直接全选复制**。
 
-沈安在：(内心独白) 老天啊，你是让我来这边受折磨的吗？
+### 步骤 3：粘贴到 Seedance
 
-【视觉风格】
-xianxia, chinese fantasy, cinematic lighting, melancholic atmosphere
-```
+将复制的内容**直接粘贴**到 Seedance 的 prompt 输入框中，点击生成。
 
 ---
 
-## 保持一致性的策略
+## 保持一致性的技巧
 
-### 1. 全局上下文文件
+### 1. 角色描述一致性
 
-`global_context.txt` 包含：
-- 核心角色设定（外貌、性格、说话风格）
-- 世界观设定（力量体系、地理环境）
-- 视觉风格关键词
+每次导出时，脚本会自动从 `spec/story/character.md` 提取角色信息。确保角色设定文件中的描述保持一致。
 
-**使用方式**：
-- 首次使用时，先粘贴全局上下文
-- 或将全局上下文保存为 Seedance 的"项目设定"
+### 2. 视觉风格关键词
 
-### 2. 集间上下文文件
+在世界观文件 `spec/story/world.md` 中定义统一的视觉风格关键词，导出会自动包含这些关键词。
 
-每集的 `context.txt` 包含：
-- 本集角色状态快照
-- 本集关键事件摘要
-- 上一集结尾状态
+### 3. 上下文衔接
 
-### 3. 角色参考文件
-
-每集的 `characters.txt` 包含：
-- 本集出场角色的完整设定
-- SD Prompt 片段
-- 服装变化记录
+每集导出时会包含 `[PREVIOUS CONTEXT]` 部分，帮助 Seedance 理解剧情连续性。
 
 ---
 
-## 执行方式
+## 常见问题
 
-```bash
-# 交互式导出
-/aim:export
+### Q: 导出文件有乱码？
 
-# 快速导出（使用默认设置）
-/aim:export --ep 1 --format seedance
+确保使用 `--format seedance`（默认）格式，这是为 Seedance 优化的纯文本格式。
 
-# 导出纯文本格式
-/aim:export --ep 1 --format text
+### Q: 如何只导出一个场景？
 
-# 导出并打开目录
-/aim:export --ep 1 --open
-```
-
----
+目前脚本按集导出。如果需要单场景，可以手动编辑导出的文件。
 
-## 导出后检查清单
+### Q: 导出的内容太长？
 
-- [ ] 每个场次文件是否可独立使用？
-- [ ] 角色设定是否完整？
-- [ ] 前情提要是否准确？
-- [ ] 视觉风格是否一致？
-- [ ] 时长是否符合配置？
+Seedance 对 prompt 长度有限制。可以使用 `--format simple` 获得更简洁的版本。
 
 ---
 
-## 注意事项
+## 与其他命令配合
 
-1. **首次使用**：建议先粘贴 `global_context.txt` 到 Seedance，建立项目基准
-2. **连续剧**：每集开始前，先粘贴上一集的 `context.txt`
-3. **角色一致性**：使用相同的角色描述和 SD Prompt 片段
-4. **时长控制**：根据视频格式配置，调整每场次的字数
+```
+1. /aim:story              → 创作剧本
+2. /aim:check-story        → 检查剧情一致性
+3. /aim:export --ep 1      → 导出第1集
+4. 复制到 Seedance         → 生成视频
+```
 
 ---
 
-## 纯文本导出（传统格式）
-
-如需导出为传统发布格式，直接读取当前剧本并整理为适合发布的格式：
-- 去除构建过程中的备注
-- 统一标点符号
-- 优化段落间距
+## 注意事项
 
-输出目标：纯文本 / 长图文案 / 剧本杀格式（请根据用户需求调整）
+1. **纯文本格式**：导出的文件不包含任何 `#`、`*`、`[]` 等 Markdown 符号
+2. **字符编码**：文件使用 UTF-8 编码，确保 Seedance 能正确识别
+3. **首次使用**：建议先导出单集测试，确认格式符合预期