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

"""
@Project: 短剧解说-解说文案
@File   : narration_copy.py
@Description: 生成可供用户审核修改的短剧解说正文
"""

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


class NarrationCopyPrompt(ParameterizedPrompt):
    """短剧解说正文生成提示词"""

    def __init__(self):
        metadata = PromptMetadata(
            name="narration_copy",
            category="short_drama_narration",
            version="v1.0",
            description="基于剧情理解和字幕生成可审核修改的短剧解说正文，不绑定时间戳",
            model_type=ModelType.TEXT,
            output_format=OutputFormat.TEXT,
            tags=["短剧", "解说文案", "爆款开头", "叙事连续性", "用户审核"],
            parameters=["drama_name", "drama_genre", "plot_analysis", "subtitle_content", "narration_language"],
        )
        super().__init__(metadata, required_parameters=["drama_name", "plot_analysis", "subtitle_content"])

        self._system_prompt = (
            "你是一位短剧解说文案创作者。你只输出可供用户审核修改的解说正文，"
            "不要输出JSON、时间戳、编号、标题、解释或Markdown。"
        )

    def get_template(self) -> str:
        return """# 短剧解说正文创作任务

## 目标
为短剧《${drama_name}》创作一份可直接给用户审核修改的解说文案正文。此阶段不做画面匹配，不输出时间戳。

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

## 原始字幕
<subtitles>
${subtitle_content}
</subtitles>

## 输出语言
<narration_language>
${narration_language}
</narration_language>

## 用户选择的短剧类型
<drama_genre>
${drama_genre}
</drama_genre>

## 类型写作规则
必须按用户选择的短剧类型调整表达重点，不要自行改判类型：
- 霸总/甜宠：突出误会、身份差、暧昧拉扯、守护感和情绪反差。
- 逆袭/复仇：突出羞辱、反击、打脸、身份揭露和爽点升级。
- 家庭伦理：突出亲情撕扯、秘密、委屈、选择和道德冲突。
- 古装/权谋：突出身份、局势、算计、立场和反转。
- 悬疑/犯罪：突出线索、危机、动机和未揭开的疑问。
- 都市情感：突出关系裂痕、现实压力、误会和情绪拉扯。
- 年代/乡村：突出家庭处境、人情压力、生活困境和命运转折。
- 自定义类型：严格服从用户填写的类型方向。

## 开头钩子公式
开头必须使用“高能反转 + 情绪冲突 + 悬念钩子”：
1. 强身份或强处境：兵王、单亲妈妈、被赶出家门的女人、被全家看不起的人等。
2. 致命反差：刚立功就被迫退役、刚回家就发现钱被输光、刚结婚就遇到孩子/婆婆阻挠。
3. 后续悬念：真正的噩梦才开始、他要讨回的不是钱、这段关系真正难的不是相爱。

## 写作规则
1. 必须使用 ${narration_language}。
2. 严格基于剧情理解和字幕事实，不编造核心情节、身份、结局。
3. 先写完整故事线，再写金句；不要只堆爆点。
4. 每句话只表达一个信息点，适合后续按句匹配画面。
5. 句子尽量短，单句优先 15-35 字；信息复杂时拆成多句。
6. 每 2-3 句要有明确因果承接，让观众知道为什么从上一幕来到下一幕。
7. 总长度控制在 300-650 字；短素材取下限，长素材取上限。
8. 不要使用编号、项目符号、章节标题或括号说明。

## 输出要求
只输出解说正文。不要输出 JSON、时间戳、代码块或任何解释。"""