init
This commit is contained in:
puke
puke
4
reelforge/prompts/__init__.py
Normal file
4
reelforge/prompts/__init__.py
Normal file
@@ -0,0 +1,4 @@
|
||||
"""
|
||||
Prompt templates for content generation
|
||||
"""
|
||||
|
||||
174
reelforge/prompts/image_prompt_template.py
Normal file
174
reelforge/prompts/image_prompt_template.py
Normal file
@@ -0,0 +1,174 @@
|
||||
"""
|
||||
Image prompt generation template
|
||||
"""
|
||||
|
||||
import json
|
||||
from typing import List, Optional
|
||||
|
||||
|
||||
# ==================== PRESET IMAGE STYLES ====================
|
||||
# Predefined visual styles for different use cases
|
||||
|
||||
IMAGE_STYLE_PRESETS = {
|
||||
"stick_figure": {
|
||||
"name": "火柴人简笔画",
|
||||
"description": "stick figure style sketch, black and white lines, pure white background, minimalist hand-drawn feel",
|
||||
"use_case": "通用场景，简单直观"
|
||||
},
|
||||
|
||||
"book": {
|
||||
"name": "书籍阅读场景",
|
||||
"description": "warm book-related scenes, reading atmosphere, books and bookshelves, soft lighting, cozy learning environment, professional photography style",
|
||||
"use_case": "书单号视频（推荐）"
|
||||
},
|
||||
|
||||
"minimal": {
|
||||
"name": "极简抽象",
|
||||
"description": "minimalist abstract art, geometric shapes, clean composition, modern design, soft pastel colors",
|
||||
"use_case": "现代感、艺术感"
|
||||
},
|
||||
|
||||
"concept": {
|
||||
"name": "概念化视觉",
|
||||
"description": "conceptual visual metaphors, symbolic elements, thought-provoking imagery, artistic interpretation",
|
||||
"use_case": "深度内容、哲学思考"
|
||||
},
|
||||
}
|
||||
|
||||
# Default preset
|
||||
DEFAULT_IMAGE_STYLE = "stick_figure"
|
||||
|
||||
|
||||
IMAGE_PROMPT_GENERATION_PROMPT = """# 角色定位
|
||||
你是一个专业的视觉创意设计师，擅长为视频脚本创作富有表现力和象征性的图像提示词，将抽象概念转化为具象的视觉画面。
|
||||
|
||||
# 核心任务
|
||||
基于已有的视频脚本，为每个分镜的"旁白内容"创作对应的**英文**图像提示词，确保视觉画面与叙述内容完美配合，增强观众的理解和记忆。
|
||||
|
||||
# 输入内容
|
||||
{narrations_json}
|
||||
|
||||
# 输出要求
|
||||
|
||||
## 图像提示词规范
|
||||
- 语言：**必须使用英文**（用于 AI 图像生成模型）
|
||||
- 画面风格：{style_description}
|
||||
- 描述结构：scene + character action + emotion + symbolic elements
|
||||
- 描述长度：确保描述清晰完整且富有创意（建议 50-100 个英文单词）
|
||||
|
||||
## 视觉创意要求
|
||||
- 每个图像都要准确反映对应旁白的具体内容和情感
|
||||
- 使用象征手法将抽象概念视觉化（如用路径代表人生选择，用锁链代表束缚等）
|
||||
- 画面要表现出丰富的情感和动作，增强视觉冲击力
|
||||
- 通过构图和元素安排突出主题，避免过于直白的表现方式
|
||||
|
||||
## 关键英文词汇参考
|
||||
- 象征元素：symbolic elements
|
||||
- 表情：expression / facial expression
|
||||
- 动作：action / gesture / movement
|
||||
- 场景：scene / setting
|
||||
- 氛围：atmosphere / mood
|
||||
|
||||
## 视觉与文案配合原则
|
||||
- 图像要服务于文案，成为文案内容的视觉延伸
|
||||
- 避免与文案内容无关或矛盾的视觉元素
|
||||
- 选择最能增强文案说服力的视觉表现方式
|
||||
- 确保观众能通过图像快速理解文案的核心观点
|
||||
|
||||
## 创意指导
|
||||
1. **现象描述类文案**：用直观的场景表现社会现象
|
||||
2. **原因分析类文案**：用因果关系的视觉比喻表现内在逻辑
|
||||
3. **影响论证类文案**：用后果场景或对比手法表现影响程度
|
||||
4. **深入探讨类文案**：用抽象概念的具象化表现深刻思考
|
||||
5. **结论启发类文案**：用开放式场景或指引性元素表现启发性
|
||||
|
||||
# 输出格式
|
||||
严格按照以下JSON格式输出，**图像提示词必须是英文**：
|
||||
|
||||
```json
|
||||
{{
|
||||
"image_prompts": [
|
||||
"[detailed English image prompt following the style requirements]",
|
||||
"[detailed English image prompt following the style requirements]"
|
||||
]
|
||||
}}
|
||||
```
|
||||
|
||||
# 重要提醒
|
||||
1. 只输出JSON格式内容，不要添加任何解释说明
|
||||
2. 确保JSON格式严格正确，可以被程序直接解析
|
||||
3. 输入是 {{"narrations": [旁白数组]}} 格式，输出是 {{"image_prompts": [图像提示词数组]}} 格式
|
||||
4. **图像提示词必须使用英文**（for AI image generation models）
|
||||
5. 图像提示词必须准确反映对应旁白的具体内容和情感
|
||||
6. 每个图像都要有创意性和视觉冲击力，避免千篇一律
|
||||
7. 严格遵守上述指定的画面风格要求（{style_description}）
|
||||
8. 确保视觉画面能增强文案的说服力和观众的理解度
|
||||
|
||||
现在，请为上述旁白创作对应的**英文**图像提示词。只输出JSON，不要其他内容。
|
||||
"""
|
||||
|
||||
|
||||
def build_image_prompt_prompt(
|
||||
narrations: List[str],
|
||||
min_words: int,
|
||||
max_words: int,
|
||||
image_style_preset: Optional[str] = None,
|
||||
image_style_description: Optional[str] = None
|
||||
) -> str:
|
||||
"""
|
||||
Build image prompt generation prompt
|
||||
|
||||
Args:
|
||||
narrations: List of narrations
|
||||
min_words: Minimum word count
|
||||
max_words: Maximum word count
|
||||
image_style_preset: Preset style name (e.g., "book", "stick_figure", "minimal", "concept")
|
||||
Available presets: see IMAGE_STYLE_PRESETS
|
||||
image_style_description: Custom style description (overrides preset if provided)
|
||||
Example: "warm book scenes, soft lighting, professional photography"
|
||||
|
||||
Returns:
|
||||
Formatted prompt
|
||||
|
||||
Examples:
|
||||
# Use preset style
|
||||
>>> build_image_prompt_prompt(narrations, 50, 100, image_style_preset="book")
|
||||
|
||||
# Use custom style
|
||||
>>> build_image_prompt_prompt(
|
||||
... narrations, 50, 100,
|
||||
... image_style_description="cyberpunk style, neon colors, futuristic"
|
||||
... )
|
||||
|
||||
# Use default style (stick_figure)
|
||||
>>> build_image_prompt_prompt(narrations, 50, 100)
|
||||
"""
|
||||
# Determine style description
|
||||
if image_style_description:
|
||||
# Custom description takes priority
|
||||
style_desc = image_style_description
|
||||
elif image_style_preset:
|
||||
# Use preset
|
||||
if image_style_preset not in IMAGE_STYLE_PRESETS:
|
||||
raise ValueError(
|
||||
f"Unknown preset '{image_style_preset}'. "
|
||||
f"Available presets: {list(IMAGE_STYLE_PRESETS.keys())}"
|
||||
)
|
||||
style_desc = IMAGE_STYLE_PRESETS[image_style_preset]["description"]
|
||||
else:
|
||||
# Use default preset
|
||||
style_desc = IMAGE_STYLE_PRESETS[DEFAULT_IMAGE_STYLE]["description"]
|
||||
|
||||
narrations_json = json.dumps(
|
||||
{"narrations": narrations},
|
||||
ensure_ascii=False,
|
||||
indent=2
|
||||
)
|
||||
|
||||
return IMAGE_PROMPT_GENERATION_PROMPT.format(
|
||||
narrations_json=narrations_json,
|
||||
min_words=min_words,
|
||||
max_words=max_words,
|
||||
style_description=style_desc
|
||||
)
|
||||
|
||||
323
reelforge/prompts/narration_template.py
Normal file
323
reelforge/prompts/narration_template.py
Normal file
@@ -0,0 +1,323 @@
|
||||
"""
|
||||
Narration generation prompt template
|
||||
|
||||
Supports three content sources:
|
||||
1. Book: Generate book review narrations from book information
|
||||
2. Topic: Generate narrations from a topic/theme
|
||||
3. Content: Extract/refine narrations from user-provided content
|
||||
"""
|
||||
|
||||
from typing import Optional
|
||||
from reelforge.models.storyboard import BookInfo
|
||||
|
||||
|
||||
# ==================== BOOK NARRATION PROMPT ====================
|
||||
# For generating book review style narrations
|
||||
|
||||
BOOK_NARRATION_PROMPT = """# 角色定位
|
||||
你是一位专业的书籍解读专家，擅长像"樊登读书"那样，用深入浅出的方式讲解书籍核心内容，帮助观众快速理解一本书的精华。
|
||||
|
||||
# 核心任务
|
||||
用户会输入一本书的名称，你需要为这本书创作 {n_storyboard} 个书籍解读分镜，每个分镜包含"旁白（用于TTS生成视频讲解音频）"，像在跟朋友推荐书籍一样，自然、有价值、引发共鸣
|
||||
|
||||
# 输出要求
|
||||
|
||||
## 旁白规范（书籍解读风格）
|
||||
- 用途定位：用于TTS生成书单号短视频音频，像樊登读书那样讲解书籍精华
|
||||
- 字数限制：严格控制在{min_words}~{max_words}个字（最低不少于{min_words}字）
|
||||
- 结尾格式：结尾不要使用标点符号
|
||||
- 内容要求：提炼书籍的核心观点，用通俗易懂的语言讲解，每个分镜传递一个有价值的洞察
|
||||
- 风格要求：像跟朋友聊天一样，通俗、真诚、有启发性，避免学术化和生硬的表达
|
||||
- 开场建议：第一个分镜可以用提问、场景、痛点等方式引发共鸣，吸引观众注意
|
||||
- 核心内容：中间分镜提炼书中的关键观点，用生活化的例子帮助理解，像樊登那样深入浅出
|
||||
- 结尾建议：最后一个分镜给出行动建议或启发，让观众有收获感
|
||||
- 衔接建议：用"你有没有发现"、"其实"、"更重要的是"、"这本书告诉我们"等连接词，保持连贯
|
||||
- 情绪与语气：温和、真诚、有热情，像一个读过书的朋友在分享收获
|
||||
- 禁止项：不出现网址、表情符号、数字编号、不说空话套话、不过度煽情、不使用"这本书说"等生硬表达
|
||||
- 字数检查：生成后必须自我验证不少于{min_words}个字，如不足则补充具体观点或生活化例子
|
||||
- 内容结构：遵循"引发共鸣 → 提炼观点 → 深入讲解 → 给出启发"的叙述逻辑，确保每个分镜都有价值
|
||||
|
||||
## 分镜连贯性要求
|
||||
- {n_storyboard} 个分镜应围绕这本书的核心内容展开，形成完整的书籍解读
|
||||
- 遵循"吸引注意 → 提炼观点 → 深入讲解 → 给出启发"的叙述逻辑
|
||||
- 每个分镜像同一个人在连贯分享读书心得，语气一致、自然流畅
|
||||
- 通过书籍的核心观点自然过渡，形成完整的解读脉络
|
||||
- 确保内容有价值、有启发，让观众觉得"这个视频值得看"
|
||||
|
||||
# 输出格式
|
||||
严格按照以下JSON格式输出，不要添加任何额外的文字说明：
|
||||
|
||||
```json
|
||||
{{
|
||||
"narrations": [
|
||||
"第一段{min_words}~{max_words}字，用提问或场景引发共鸣，吸引观众",
|
||||
"第二段{min_words}~{max_words}字，提炼书中核心观点，深入浅出讲解",
|
||||
"第三段{min_words}~{max_words}字，给出行动建议或启发，让观众有收获"
|
||||
]
|
||||
}}
|
||||
```
|
||||
|
||||
# 示例输出
|
||||
假设用户输入书名："{topic}"，输出示例：
|
||||
|
||||
```json
|
||||
{{
|
||||
"narrations": [
|
||||
"你有没有这样的经历，明明知道该做什么，但就是做不到，这本书告诉我们，问题的关键在于习惯",
|
||||
"作者提出了一个简单但有力的观点，改变不需要靠意志力，而是要设计一个好的系统",
|
||||
"书中有个很有意思的例子，如果你想养成阅读习惯，不要逼自己每天读一小时，而是先从每天读一页开始",
|
||||
"更重要的是，习惯的复利效应非常惊人，每天进步百分之一，一年后你会进步三十七倍",
|
||||
"所以与其追求完美的计划，不如从一个小到不可能失败的习惯开始，然后坚持下去"
|
||||
]
|
||||
}}
|
||||
```
|
||||
|
||||
# 重要提醒
|
||||
1. 只输出JSON格式内容，不要添加任何解释说明
|
||||
2. 确保JSON格式严格正确，可以被程序直接解析
|
||||
3. 旁白必须严格控制在{min_words}~{max_words}字之间，用通俗易懂的语言，像樊登那样讲解
|
||||
4. {n_storyboard} 个分镜要围绕这本书的核心观点展开，形成完整的书籍解读
|
||||
5. 每个分镜都要有价值，提炼书中的洞察，避免空洞的介绍
|
||||
6. 输出格式为 {{"narrations": [旁白数组]}} 的JSON对象
|
||||
|
||||
现在，请为书籍《{book_name}》创作 {n_storyboard} 个分镜的解读旁白。只输出JSON，不要其他内容。
|
||||
"""
|
||||
|
||||
|
||||
# ==================== TOPIC NARRATION PROMPT ====================
|
||||
# For generating narrations from a topic/theme
|
||||
|
||||
TOPIC_NARRATION_PROMPT = """# 角色定位
|
||||
你是一位专业的内容创作专家，擅长将话题扩展成引人入胜的短视频脚本，用深入浅出的方式讲解观点，帮助观众理解复杂概念。
|
||||
|
||||
# 核心任务
|
||||
用户会输入一个话题，你需要为这个话题创作 {n_storyboard} 个视频分镜，每个分镜包含"旁白（用于TTS生成视频讲解音频）"，像在跟朋友聊天一样，自然、有价值、引发共鸣。
|
||||
|
||||
# 输入话题
|
||||
{topic}
|
||||
|
||||
# 输出要求
|
||||
|
||||
## 旁白规范
|
||||
- 用途定位：用于TTS生成短视频音频，通俗易懂地讲解话题
|
||||
- 字数限制：严格控制在{min_words}~{max_words}个字（最低不少于{min_words}字）
|
||||
- 结尾格式：结尾不要使用标点符号
|
||||
- 内容要求：围绕话题展开，每个分镜传递一个有价值的观点或洞察
|
||||
- 风格要求：像跟朋友聊天一样，通俗、真诚、有启发性，避免学术化和生硬的表达
|
||||
- 开场建议：第一个分镜可以用提问、场景、痛点等方式引发共鸣，吸引观众注意
|
||||
- 核心内容：中间分镜展开核心观点，用生活化的例子帮助理解
|
||||
- 结尾建议：最后一个分镜给出行动建议或启发，让观众有收获感
|
||||
- 衔接建议：用"你有没有发现"、"其实"、"更重要的是"等连接词，保持连贯
|
||||
- 情绪与语气：温和、真诚、有热情，像一个有见解的朋友在分享思考
|
||||
- 禁止项：不出现网址、表情符号、数字编号、不说空话套话、不过度煽情
|
||||
- 字数检查：生成后必须自我验证不少于{min_words}个字，如不足则补充具体观点或例子
|
||||
- 内容结构：遵循"引发共鸣 → 提出观点 → 深入讲解 → 给出启发"的叙述逻辑
|
||||
|
||||
## 分镜连贯性要求
|
||||
- {n_storyboard} 个分镜应围绕话题展开，形成完整的观点表达
|
||||
- 遵循"吸引注意 → 提出观点 → 深入讲解 → 给出启发"的叙述逻辑
|
||||
- 每个分镜像同一个人在连贯分享观点，语气一致、自然流畅
|
||||
- 通过观点的递进自然过渡，形成完整的论述脉络
|
||||
- 确保内容有价值、有启发，让观众觉得"这个视频值得看"
|
||||
|
||||
# 输出格式
|
||||
严格按照以下JSON格式输出，不要添加任何额外的文字说明：
|
||||
|
||||
```json
|
||||
{{
|
||||
"narrations": [
|
||||
"第一段{min_words}~{max_words}字，用提问或场景引发共鸣",
|
||||
"第二段{min_words}~{max_words}字，展开核心观点",
|
||||
"第三段{min_words}~{max_words}字，给出启发或建议"
|
||||
]
|
||||
}}
|
||||
```
|
||||
|
||||
# 重要提醒
|
||||
1. 只输出JSON格式内容，不要添加任何解释说明
|
||||
2. 确保JSON格式严格正确，可以被程序直接解析
|
||||
3. 旁白必须严格控制在{min_words}~{max_words}字之间，用通俗易懂的语言
|
||||
4. {n_storyboard} 个分镜要围绕话题展开，形成完整的观点表达
|
||||
5. 每个分镜都要有价值，提供洞察，避免空洞的陈述
|
||||
6. 输出格式为 {{"narrations": [旁白数组]}} 的JSON对象
|
||||
|
||||
现在，请为话题创作 {n_storyboard} 个分镜的旁白。只输出JSON，不要其他内容。
|
||||
"""
|
||||
|
||||
|
||||
# ==================== CONTENT NARRATION PROMPT ====================
|
||||
# For extracting/refining narrations from user-provided content
|
||||
|
||||
CONTENT_NARRATION_PROMPT = """# 角色定位
|
||||
你是一位专业的内容提炼专家，擅长从用户提供的内容中提取核心要点，并转化成适合短视频的脚本。
|
||||
|
||||
# 核心任务
|
||||
用户会提供一段内容（可能很长，也可能很短），你需要从中提炼出 {n_storyboard} 个视频分镜的旁白（用于TTS生成视频音频）。
|
||||
|
||||
# 用户提供的内容
|
||||
{content}
|
||||
|
||||
# 输出要求
|
||||
|
||||
## 旁白规范
|
||||
- 用途定位：用于TTS生成短视频音频
|
||||
- 字数限制：严格控制在{min_words}~{max_words}个字（最低不少于{min_words}字）
|
||||
- 结尾格式：结尾不要使用标点符号
|
||||
- 提炼策略：
|
||||
* 如果用户内容较长：提取{n_storyboard}个核心要点，去除冗余信息
|
||||
* 如果用户内容较短：在保留核心观点的基础上适当扩展，增加例子或解释
|
||||
* 如果用户内容刚好：优化表达，使其更适合口播
|
||||
- 风格要求：保持用户内容的核心观点，但用更口语化、适合TTS的方式表达
|
||||
- 开场建议：第一个分镜可以用提问或场景引入，吸引观众注意
|
||||
- 核心内容：中间分镜展开用户内容的核心要点
|
||||
- 结尾建议：最后一个分镜给出总结或启发
|
||||
- 情绪与语气：温和、真诚、自然，像在跟朋友分享观点
|
||||
- 禁止项：不出现网址、表情符号、数字编号、不说空话套话
|
||||
- 字数检查：生成后必须自我验证每段不少于{min_words}个字
|
||||
|
||||
## 分镜连贯性要求
|
||||
- {n_storyboard} 个分镜应基于用户内容的核心观点展开，形成完整表达
|
||||
- 保持逻辑连贯，自然过渡
|
||||
- 每个分镜像同一个人在讲述，语气一致
|
||||
- 确保提炼的内容忠于用户原意，但更适合短视频呈现
|
||||
|
||||
# 输出格式
|
||||
严格按照以下JSON格式输出，不要添加任何额外的文字说明：
|
||||
|
||||
```json
|
||||
{{
|
||||
"narrations": [
|
||||
"第一段{min_words}~{max_words}字的旁白",
|
||||
"第二段{min_words}~{max_words}字的旁白",
|
||||
"第三段{min_words}~{max_words}字的旁白"
|
||||
]
|
||||
}}
|
||||
```
|
||||
|
||||
# 重要提醒
|
||||
1. 只输出JSON格式内容，不要添加任何解释说明
|
||||
2. 确保JSON格式严格正确，可以被程序直接解析
|
||||
3. 旁白必须严格控制在{min_words}~{max_words}字之间
|
||||
4. 必须输出恰好 {n_storyboard} 个分镜的旁白
|
||||
5. 内容要忠于用户原意，但优化为更适合口播的表达
|
||||
6. 输出格式为 {{"narrations": [旁白数组]}} 的JSON对象
|
||||
|
||||
现在，请从上述内容中提炼出 {n_storyboard} 个分镜的旁白。只输出JSON，不要其他内容。
|
||||
"""
|
||||
|
||||
|
||||
# ==================== PROMPT BUILDER FUNCTIONS ====================
|
||||
|
||||
def build_book_narration_prompt(
|
||||
book_info: BookInfo,
|
||||
n_storyboard: int,
|
||||
min_words: int,
|
||||
max_words: int
|
||||
) -> str:
|
||||
"""
|
||||
Build book review narration prompt
|
||||
|
||||
Args:
|
||||
book_info: Book information
|
||||
n_storyboard: Number of storyboard frames
|
||||
min_words: Minimum word count
|
||||
max_words: Maximum word count
|
||||
|
||||
Returns:
|
||||
Formatted prompt
|
||||
"""
|
||||
# Build book description for prompt
|
||||
book_name = book_info.title
|
||||
if book_info.author:
|
||||
book_name = f"{book_info.title} - {book_info.author}"
|
||||
|
||||
return BOOK_NARRATION_PROMPT.format(
|
||||
book_name=book_name,
|
||||
n_storyboard=n_storyboard,
|
||||
min_words=min_words,
|
||||
max_words=max_words
|
||||
)
|
||||
|
||||
|
||||
def build_topic_narration_prompt(
|
||||
topic: str,
|
||||
n_storyboard: int,
|
||||
min_words: int,
|
||||
max_words: int
|
||||
) -> str:
|
||||
"""
|
||||
Build topic narration prompt
|
||||
|
||||
Args:
|
||||
topic: Topic or theme
|
||||
n_storyboard: Number of storyboard frames
|
||||
min_words: Minimum word count
|
||||
max_words: Maximum word count
|
||||
|
||||
Returns:
|
||||
Formatted prompt
|
||||
"""
|
||||
return TOPIC_NARRATION_PROMPT.format(
|
||||
topic=topic,
|
||||
n_storyboard=n_storyboard,
|
||||
min_words=min_words,
|
||||
max_words=max_words
|
||||
)
|
||||
|
||||
|
||||
def build_content_narration_prompt(
|
||||
content: str,
|
||||
n_storyboard: int,
|
||||
min_words: int,
|
||||
max_words: int
|
||||
) -> str:
|
||||
"""
|
||||
Build content refinement narration prompt
|
||||
|
||||
Args:
|
||||
content: User-provided content
|
||||
n_storyboard: Number of storyboard frames
|
||||
min_words: Minimum word count
|
||||
max_words: Maximum word count
|
||||
|
||||
Returns:
|
||||
Formatted prompt
|
||||
"""
|
||||
return CONTENT_NARRATION_PROMPT.format(
|
||||
content=content,
|
||||
n_storyboard=n_storyboard,
|
||||
min_words=min_words,
|
||||
max_words=max_words
|
||||
)
|
||||
|
||||
|
||||
def build_narration_prompt(
|
||||
topic: str,
|
||||
n_storyboard: int,
|
||||
min_words: int,
|
||||
max_words: int
|
||||
) -> str:
|
||||
"""
|
||||
Build narration generation prompt (legacy function for backward compatibility)
|
||||
|
||||
Args:
|
||||
topic: Topic (book name or discussion topic)
|
||||
n_storyboard: Number of storyboard frames
|
||||
min_words: Minimum word count
|
||||
max_words: Maximum word count
|
||||
|
||||
Returns:
|
||||
Formatted prompt
|
||||
|
||||
Note:
|
||||
This function is kept for backward compatibility.
|
||||
Use build_book_narration_prompt, build_topic_narration_prompt,
|
||||
or build_content_narration_prompt instead.
|
||||
"""
|
||||
return build_topic_narration_prompt(
|
||||
topic=topic,
|
||||
n_storyboard=n_storyboard,
|
||||
min_words=min_words,
|
||||
max_words=max_words
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user