From c0e0b8ec6db9ad5be484b38717463c6d6048bdf0 Mon Sep 17 00:00:00 2001 From: linyq Date: Mon, 12 May 2025 11:20:31 +0800 Subject: [PATCH] =?UTF-8?q?feat(SDE):=20=E4=BC=98=E5=8C=96=E5=89=A7?= =?UTF-8?q?=E6=83=85=E5=88=86=E6=9E=90=E6=8F=90=E7=A4=BA=E8=AF=8D=E5=92=8C?= =?UTF-8?q?=E8=A7=A3=E8=AF=B4=E6=96=87=E6=A1=88=E7=94=9F=E6=88=90?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 在 generate_short_summary.py 中添加 temperature 参数，用于控制生成文案的随机性- 修改 prompt.py 中的段落数要求，改为与字幕长度成正比 - 在 short_drama_explanation.py 中添加日志输出，记录使用的模型和温度 --- app/services/SDE/prompt.py | 2 +- app/services/SDE/short_drama_explanation.py | 2 +- webui/tools/generate_short_summary.py | 3 ++- 3 files changed, 4 insertions(+), 3 deletions(-) diff --git a/app/services/SDE/prompt.py b/app/services/SDE/prompt.py index eaf5752..78385cc 100644 --- a/app/services/SDE/prompt.py +++ b/app/services/SDE/prompt.py @@ -17,7 +17,7 @@ subtitle_plot_analysis_v1 = """ 1. **整体剧情分析**：简要概括整个短剧的核心剧情脉络、主要冲突和结局（如果有的话）。 2. **分段剧情解析与时间戳定位**： * 将整个短剧划分为若干个关键的剧情段落（例如：开端、发展、转折、高潮、结局，或根据具体情节自然划分）。 - * 段落数不得少于 20 个 + * 段落数应该与字幕长度成正比。 * 对于每一个剧情段落： * **概括该段落的主要内容**：用简洁的语言描述这段剧情发生了什么。 * **标注对应的时间戳范围**：明确指出该剧情段落对应的开始字幕时间戳和结束字幕时间戳。请直接从字幕中提取时间信息。 diff --git a/app/services/SDE/short_drama_explanation.py b/app/services/SDE/short_drama_explanation.py index e7d1c08..56a460d 100644 --- a/app/services/SDE/short_drama_explanation.py +++ b/app/services/SDE/short_drama_explanation.py @@ -351,7 +351,7 @@ def analyze_subtitle( base_url=base_url, custom_prompt=custom_prompt ) - + logger.debug(f"使用模型: {analyzer.model} 开始分析, 温度: {analyzer.temperature}") # 分析字幕 if subtitle_content: result = analyzer.analyze_subtitle(subtitle_content) diff --git a/webui/tools/generate_short_summary.py b/webui/tools/generate_short_summary.py index 6e5eb0b..eb2a6f4 100644 --- a/webui/tools/generate_short_summary.py +++ b/webui/tools/generate_short_summary.py @@ -60,7 +60,8 @@ def generate_script_short_sunmmary(params, subtitle_path, video_theme, temperatu api_key=text_api_key, model=text_model, base_url=text_base_url, - save_result=True + save_result=True, + temperature=temperature ) """ 3. 根据剧情生成解说文案