bdim/Canto
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

feat: update emotion handling and adjust alpha levels in TTS and LLM services

Browse Source
This commit is contained in:
bdim404
2026-04-07 14:17:29 +08:00
parent a8d6195cdb
commit a144540cbe
3 changed files with 74 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

48
qwen3-tts-backend/core/llm_service.py
View File

@@ -321,26 +321,17 @@ class LLMService:
@staticmethod
def _emotion_limits(violence_level: int, eroticism_level: int) -> tuple[str, str]:
v = violence_level / 10
e = eroticism_level / 10
female_happy = round(0.20 + 0.45 * e, 2)
angry = round(0.15 + 0.65 * v, 2)
sad = round(0.10 + 0.40 * v, 2)
fear = round(0.10 + 0.60 * v, 2)
hate = round(0.35 + 0.25 * max(v, e), 2)
low = round(0.35 + 0.45 * e, 2)
surprise= round(0.10 + 0.35 * max(v, e), 2)
limits = (
f"愤怒={angry}、悲伤={sad}、恐惧={fear}、厌恶={hate}、低沉={low}、惊讶={surprise}、"
f"开心：男性角色上限=0.20，女性角色上限={female_happy}"
)
guidance_parts = []
if violence_level >= 4:
guidance_parts.append(f"暴力程度{violence_level}/10，台词中的愤怒、恐惧、悲伤情绪必须强烈外露，不得克制")
if eroticism_level >= 4:
guidance_parts.append(f"色情程度{eroticism_level}/10，女性台词中的开心、低沉、挑逗情绪应充分表达")
if violence_level >= 7:
guidance_parts.append(f"暴力程度{violence_level}/10，激烈场景的愤怒/恐惧/悲伤强度应用7-10级，不得克制")
elif violence_level >= 4:
guidance_parts.append(f"暴力程度{violence_level}/10，台词中的愤怒/恐惧/悲伤情绪可用4-7级")
if eroticism_level >= 7:
guidance_parts.append(f"色情程度{eroticism_level}/10，女性台词中的开心/低沉情绪应用7-10级充分表达")
elif eroticism_level >= 4:
guidance_parts.append(f"色情程度{eroticism_level}/10，女性台词中的开心/低沉情绪可用4-7级")
guidance = "；".join(guidance_parts)
return limits, guidance
return "", guidance
async def generate_chapter_script(
self,
@@ -383,11 +374,9 @@ class LLMService:
" 【角色名】\"对话内容\"（情感词:强度）\n\n"
"情感标注规则：\n"
"- 情感词可选：开心、愤怒、悲伤、恐惧、厌恶、低沉、惊讶\n"
"- 单一情感：（情感词:强度），如（开心:0.5）、（悲伤:0.3）\n"
"- 混合情感：（情感1:比重+情感2:比重），如（开心:0.6+悲伤:0.2）、（愤怒:0.3+恐惧:0.4）\n"
"- 混合情感时每个情感的比重独立设定，反映各自对情绪的贡献\n"
f"- 各情感比重上限（严格不超过）：{limits_str}\n"
"- 鼓励使用低值（0.05–0.10）表达微弱、内敛或一闪而过的情绪，无需非强即无\n"
"- 每行只允许标注一个情感词，格式：（情感词:强度级别），强度为1–10的整数，10最强\n"
"- 示例：（开心:6）、（悲伤:3）、（愤怒:8）\n"
"- 鼓励使用低值（1–3）表达微弱、内敛或一闪而过的情绪，无需非强即无\n"
"- 确实没有任何情绪色彩时可省略整个括号\n"
+ char_personality_str
+ narrator_rule
@@ -468,18 +457,15 @@ class LLMService:
"所有非对话的叙述文字归属于旁白角色。\n"
"同时根据语境为每个片段判断是否有明显情绪，有则在 emo_text 中标注，无则留空。\n"
"可选情绪词：开心、愤怒、悲伤、恐惧、厌恶、低沉、惊讶。\n"
"emo_text 格式规则：\n"
" 单一情感：直接填情感词，用 emo_alpha 设置强度，如 emo_text=\"开心\", emo_alpha=0.3\n"
" 混合情感：用 情感词:比重 格式拼接，emo_alpha 设为 1.0，如 emo_text=\"开心:0.6+悲伤:0.2\", emo_alpha=1.0\n"
"各情感比重上限（严格不超过）：开心=0.20、愤怒=0.15、悲伤=0.1、恐惧=0.1、厌恶=0.35、低沉=0.35、惊讶=0.10。\n"
"鼓励用低值（0.05–0.10）表达微弱或内敛的情绪，不要非强即无；完全无情绪色彩时 emo_text 置空。\n"
"emo_text 只允许单一情感词；emo_alpha 为1–10的整数表示强度（10最强）；完全无情绪色彩时 emo_text 置空，emo_alpha 为 0。\n"
"鼓励用低值（1–3）表达微弱或内敛的情绪，不要非强即无。\n"
+ personality_str
+ "同一角色的连续台词，情绪应尽量保持一致或仅有微弱变化，避免相邻片段间情绪跳跃。\n"
"只输出JSON数组，不要有其他文字，格式如下：\n"
'[{"character": "旁白", "text": "叙述文字", "emo_text": "", "emo_alpha": 0}, '
'{"character": "角色名", "text": "淡淡的问候", "emo_text": "开心", "emo_alpha": 0.08}, '
'{"character": "角色名", "text": "激动的欢呼", "emo_text": "开心", "emo_alpha": 0.18}, '
'{"character": "角色名", "text": "含泪的笑", "emo_text": "开心:0.12+悲伤:0.08", "emo_alpha": 1.0}]'
'{"character": "角色名", "text": "淡淡的问候", "emo_text": "开心", "emo_alpha": 3}, '
'{"character": "角色名", "text": "激动的欢呼", "emo_text": "开心", "emo_alpha": 8}, '
'{"character": "角色名", "text": "愤怒的质问", "emo_text": "愤怒", "emo_alpha": 7}]'
)
user_message = f"请解析以下章节文本：\n\n{chapter_text}"
result = await self.stream_chat_json(system_prompt, user_message, on_token, max_tokens=16384, usage_callback=usage_callback)