#!/usr/bin/env python
# -*- coding: UTF-8 -*-

"""
@Project: 影视解说-脚本修复
@File   : script_repair.py
@Description: 影视解说脚本校验失败后的JSON修复提示词
"""

from ..base import ParameterizedPrompt, PromptMetadata, ModelType, OutputFormat


class ScriptRepairPrompt(ParameterizedPrompt):
    """影视解说脚本修复提示词"""

    def __init__(self):
        metadata = PromptMetadata(
            name="script_repair",
            category="film_tv_narration",
            version="v1.0",
            description="根据确定性校验错误修复影视解说脚本JSON，优先修正时间戳、视频来源和格式问题",
            model_type=ModelType.TEXT,
            output_format=OutputFormat.JSON,
            tags=["影视", "解说脚本", "JSON修复", "时间戳校验", "多视频"],
            parameters=[
                "drama_name",
                "drama_genre",
                "plot_analysis",
                "subtitle_content",
                "invalid_script",
                "validation_errors",
                "narration_language",
            ],
        )
        super().__init__(
            metadata,
            required_parameters=["drama_name", "subtitle_content", "invalid_script", "validation_errors"],
        )

        self._system_prompt = (
            "你是一位影视解说脚本JSON修复器。你只能根据校验错误修复JSON，"
            "必须输出严格JSON，不能输出解释、Markdown或代码块。"
        )

    def get_template(self) -> str:
        return """# 影视解说脚本修复任务

## 修复目标
下面的影视作品《${drama_name}》解说脚本未通过剪辑校验。请只根据校验错误和字幕内容修复它，输出一个完整可剪辑的 JSON。

## 剧情理解材料
<plot>
${plot_analysis}
</plot>

## 校验错误
<validation_errors>
${validation_errors}
</validation_errors>

## 当前无效脚本
<invalid_script>
${invalid_script}
</invalid_script>

## 可用字幕窗口
<subtitles>
${subtitle_content}
</subtitles>

## 解说台词目标语言
<narration_language>
${narration_language}
</narration_language>

## 用户选择的影视类型
<drama_genre>
${drama_genre}
</drama_genre>

## 修复规则
1. 只输出 JSON，不要任何解释、标题、Markdown 或代码块。
2. 输出根对象必须是 {"items": [...]}。
3. 每个 item 必须包含 _id、video_id、video_name、timestamp、picture、narration、OST。
4. 必须删除片头、片尾、演职员表、版权声明、平台水印展示、下集预告、花絮、赞助口播、商品露出、贴片广告、中插广告、片中广告或任何与主线剧情无关的推广片段；这些内容绝对不能出现在修复后的 items 中。
5. 如果字幕或画面文字出现“广告”“赞助”“推广”“片头”“片尾”“预告”“下集”“扫码”“购买”“会员”“关注”等明显非剧情信号，必须删除对应 item，不得改写成解说片段。
6. video_id、video_name 和 timestamp 必须来自对应字幕窗口；不得把不同视频的同名时间戳混用。
7. 同一 video_id 内片段不得交叉或重叠。
8. OST=1 的 narration 必须是“播放原片+序号”；OST=0 的 narration 必须使用 ${narration_language}。
9. 禁止连续 3 个或更多 OST=1；必须插入或改写 OST=0 解说片段承接剧情。
10. 跨 video_id 切换前后不能都是 OST=1；必须至少有一个 OST=0 片段解释场景和剧情为什么切换。
11. OST=0 narration 要补足人物动机、信息承接和因果转折，不要只概括当前画面。
12. 第一段必须是 OST=0 解说钩子，按“人物困境 + 反常信息 + 悬念问题”写，不要直接播放原片。
13. OST=0 文案必须匹配画面时长，按“解说字数 / 5 = 所需视频秒数”估算；过密时要缩短文案、延长时间戳或拆成多个片段。
14. 不要自行改判影视类型；如需改写 narration，必须按用户选择的 ${drama_genre} 保持表达重点。
15. 尽量保留原脚本中没有错误的片段；无法修复的片段可以删除，但剩余片段必须重新按 1 开始编号。

请输出修复后的完整 JSON。"""