# Director Agent 伪代码逻辑
class DirectorAgent:
    def __init__(self, gemini_client):
        self.gemini = gemini_client
    
    def run(self, brief: CreativeBrief, footage_index: FootageIndex) -> EditPlan:
        # 1. 解析创意简报
        product = brief.product
        audience = brief.audience
        tone = brief.tone
        duration = brief.duration_seconds
        
        # 2. 搜索素材
        relevant_shots = self.gemini.search(
            footage_index,
            criteria=f"{product} for {audience} with {tone} tone"
        )
        
        # 3. 生成剪辑计划
        edit_plan = self.gemini.generate_edit_plan(
            shots=relevant_shots,
            duration=duration,
            style=brief.style_ref
        )
        
        return edit_plan

class EditPlan(BaseModel):
    shots: List[Shot]           # 镜头列表（有序）
    text_overlays: List[TextOverlay]  # 文字叠加
    transitions: List[str]       # 转场效果
    music_mood: str             # 音乐风格
    
class Shot(BaseModel):
    footage_id: str             # 素材ID
    start_time: float           # 入点
    end_time: float             # 出点
    reason: str                 # 选择理由

class TrimRefinerAgent:
    def run(self, edit_plan: EditPlan) -> EditPlan:
        refined_plan = EditPlan()
        
        for shot in edit_plan.shots:
            # 精细化裁剪边界
            new_start = find_best_cut_point(
                shot.start_time, 
                mode='forward'  # 向后找到最佳切点
            )
            new_end = find_best_cut_point(
                shot.end_time,
                mode='backward'  # 向前找到最佳切点
            )
            
            refined_plan.shots.append(Shot(
                footage_id=shot.footage_id,
                start_time=new_start,
                end_time=new_end,
                reason=f"Refined: {shot.reason}"
            ))
        
        return refined_plan

class EditorAgent:
    def __init__(self, ffmpeg_path="/usr/bin/ffmpeg"):
        self.ffmpeg = ffmpeg_path
    
    def run(self, edit_plan: EditPlan, footage_dir: Path) -> Path:
        output_path = Path("output") / f"{edit_plan.name}_v{version}.mp4"
        
        # 构建FFmpeg命令
        ffmpeg_cmd = self.build_ffmpeg_cmd(edit_plan, footage_dir, output_path)
        
        # 执行渲染
        subprocess.run(ffmpeg_cmd, check=True)
        
        return output_path
    
    def build_ffmpeg_cmd(self, plan, footage_dir, output):
        # 1. 拼接视频片段
        # 2. 应用裁剪
        # 3. 添加转场
        # 4. 叠加文字
        # 5. 混音
        pass

class ReviewerAgent:
    def __init__(self, gemini_client):
        self.gemini = gemini_client
    
    def run(self, video_path: Path, brief: CreativeBrief) -> ReviewScore:
        # 让Gemini观看视频并评分
        scores = self.gemini.evaluate_video(
            video_path=video_path,
            criteria={
                "adherence": "是否符合创意简报要求",
                "pacing": "节奏是否流畅",
                "visual_quality": "视觉质量是否达标",
                "watchability": "观看体验"
            }
        )
        
        return ReviewScore(**scores)
    
    def should_retry(self, score: ReviewScore, threshold: float) -> bool:
        return score.overall < threshold

# pipelines/ugc-ad.yaml
name: ugc-ad
description: "用户生成内容广告流水线"

steps:
  - agent: director          # 第一步：导演选择镜头
  - agent: trim_refiner      # 第二步：优化裁剪点
  - agent: editor            # 第三步：FFmpeg渲染
  - agent: reviewer          # 第四步：质量评审
    retry_if:
      metric: overall        # 基于综合评分重试
      threshold: 0.65        # 阈值：0.65
      max_retries: 2         # 最多重试2次（3次尝试）
      feedback_target: director  # 反馈给导演改进

# pipelines/fast-promo.yaml
name: fast-promo
description: "快速促销视频流水线（无TrimRefiner）"

steps:
  - agent: director
  - agent: editor
  - agent: reviewer
    retry_if:
      metric: adherence
      threshold: 0.8
      max_retries: 1

{
  "product": "Smart Water Bottle",
  "audience": "Health-conscious women 25-45",
  "tone": "authentic",
  "duration_seconds": 30,
  "style_ref": "styles/dtc-testimonial.yaml"
}

{
  "footage_dir": "/path/to/footage",
  "shots": [
    {
      "id": "shot_001",
      "footage_file": "raw_clips/clip_001.mp4",
      "start_time": 0.0,
      "end_time": 15.3,
      "transcript": "This water bottle tracks my hydration...",
      "visual_features": {
        "scene_complexity": "low",
        "camera_motion": "static",
        "lighting": "natural"
      },
      "audio_features": {
        "energy": "medium",
        "has_speech": true,
        "has_music": false
      }
    }
  ],
  "generated_at": "2026-04-17T12:00:00Z"
}

# 只有素材变化时才重新处理
if footage_index.exists() and is_cache_valid(footage_dir, footage_index):
    logger.info("Using cached footage index")
else:
    logger.info("Generating new footage index")
    footage_index = preprocess(footage_dir)
    footage_index.save()

# 1. 克隆仓库
git clone https://github.com/poseljacob/agentic-video-editor.git
cd agentic-video-editor

# 2. 创建虚拟环境
python -m venv .venv
source .venv/bin/activate  # Windows: .venv\Scripts\activate

# 3. 安装依赖
pip install -e ".[dev]"

# 4. 配置API Key
cp .env.example .env
# 编辑.env，添加：GOOGLE_AI_API_KEY=your_key_here

# 5. 验证FFmpeg
ffmpeg -version

# 基本用法
ave edit \
  --footage-dir /path/to/your/footage \
  --brief '{"product": "My Product", "audience": "Women 25-45", "tone": "authentic", "duration_seconds": 30}' \
  --pipeline pipelines/ugc-ad.yaml \
  --style styles/dtc-testimonial.yaml

# 终端1：启动FastAPI后端
source .venv/bin/activate
cd agentic-video-editor
uvicorn src.web.app:app --reload --port 8000

# 终端2：启动Next.js前端
cd src/web/studio
pnpm install  # 首次需要安装依赖
pnpm dev --port 3000

# src/agents/my_agent.py
from .base import BaseAgent

class MyCustomAgent(BaseAgent):
    name = "my_custom_agent"
    description = "我的自定义Agent"
    
    def run(self, context: dict) -> dict:
        # 实现Agent逻辑
        result = self.do_something(context)
        return result

# src/tools/my_tools.py
from google.adk.tools import FunctionTool

@FunctionTool
def search_footage_by_emotion(footage_index: dict, emotion: str) -> list:
    """根据情感搜索素材片段"""
    # 实现搜索逻辑
    pass