feat: add violence and eroticism level parameters to synopsis and script generation requests

qwen3-tts-backend/core/llm_service.py

@@ -96,6 +96,9 @@ class LLMService:
                 raw = "\n".join(inner).strip()
         if not raw:
             raise ValueError("LLM returned empty JSON after stripping markdown")
+        if not raw.startswith(("{", "[")):
+            logger.error(f"LLM refused or returned non-JSON. Raw (first 500): {raw[:500]}")
+            raise ValueError(f"LLM拒绝响应：{raw[:200]}")
         try:
             return json.loads(raw)
         except json.JSONDecodeError:
@@ -245,6 +248,8 @@ class LLMService:
         style: str,
         num_characters: int,
         usage_callback: Optional[Callable[[int, int], None]] = None,
+        violence_level: int = 0,
+        eroticism_level: int = 0,
     ) -> list[Dict]:
         genre_label = f"{genre}{'/' + subgenre if subgenre else ''}"
         system_prompt = (
@@ -272,6 +277,10 @@ class LLMService:
         if style:
             parts.append(f"风格：{style}")
         parts.append(f"故事简介：{premise}")
+        if violence_level > 0:
+            parts.append(f"暴力程度：{violence_level}/10")
+        if eroticism_level > 0:
+            parts.append(f"色情程度：{eroticism_level}/10")
         parts.append(f"请为这个故事创作 {num_characters} 个主要角色，再加上旁白narrator，共 {num_characters + 1} 个角色。")
         user_message = "\n".join(parts)
         result = await self.stream_chat_json(system_prompt, user_message, max_tokens=4096, usage_callback=usage_callback)
@@ -286,6 +295,8 @@ class LLMService:
         num_chapters: int,
         characters: list[Dict],
         usage_callback: Optional[Callable[[int, int], None]] = None,
+        violence_level: int = 0,
+        eroticism_level: int = 0,
     ) -> list[Dict]:
         system_prompt = (
             "你是一个专业的故事创作助手。请根据给定的故事信息和角色列表，创作章节大纲。\n"
@@ -295,12 +306,15 @@ class LLMService:
         )
         genre_label = f"{genre}{'/' + subgenre if subgenre else ''}"
         char_names = [c.get("name", "") for c in characters if c.get("name") not in ("narrator", "旁白")]
+        violence_note = f"暴力程度：{violence_level}/10\n" if violence_level > 0 else ""
+        eroticism_note = f"色情程度：{eroticism_level}/10\n" if eroticism_level > 0 else ""
         user_message = (
             f"故事类型：{genre_label}\n"
             + (f"风格：{style}\n" if style else "")
             + f"故事简介：{premise}\n"
             f"主要角色：{', '.join(char_names)}\n"
-            f"请创作 {num_chapters} 章的大纲。"
+            + violence_note + eroticism_note
+            + f"请创作 {num_chapters} 章的大纲。"
         )
         result = await self.stream_chat_json(system_prompt, user_message, max_tokens=4096, usage_callback=usage_callback)
         return result.get("chapters", [])
@@ -315,6 +329,8 @@ class LLMService:
         characters: list[Dict],
         on_token=None,
         usage_callback: Optional[Callable[[int, int], None]] = None,
+        violence_level: int = 0,
+        eroticism_level: int = 0,
     ) -> str:
         char_names = [c.get("name", "") for c in characters if c.get("name") not in ("narrator", "旁白")]
         names_str = "、".join(char_names)
@@ -337,10 +353,13 @@ class LLMService:
             "- 每行为一个独立片段，不要有空行\n"
             "- 直接输出脚本内容，不要有其他说明文字"
         )
+        violence_note = f"暴力程度：{violence_level}/10\n" if violence_level > 0 else ""
+        eroticism_note = f"色情程度：{eroticism_level}/10\n" if eroticism_level > 0 else ""
         user_message = (
             f"故事类型：{genre}\n"
-            f"故事简介：{premise}\n\n"
-            f"第 {chapter_index + 1} 章：{chapter_title}\n"
+            f"故事简介：{premise}\n"
+            + violence_note + eroticism_note
+            + f"\n第 {chapter_index + 1} 章：{chapter_title}\n"
             f"章节内容：{chapter_summary}\n\n"
             "请创作这一章的完整对话脚本，包含旁白叙述和角色对话，内容充实，段落自然流畅。"
         )
@@ -358,6 +377,8 @@ class LLMService:
         additional_chapters: int,
         characters: list[Dict],
         usage_callback: Optional[Callable[[int, int], None]] = None,
+        violence_level: int = 0,
+        eroticism_level: int = 0,
     ) -> list[Dict]:
         system_prompt = (
             "你是一个专业的故事创作助手。请根据已有章节大纲，续写新的章节大纲。\n"
@@ -373,12 +394,15 @@ class LLMService:
             f"第{ch.get('index', i) + 1}章「{ch.get('title', '')}」：{ch.get('summary', '')}"
             for i, ch in enumerate(existing_chapters)
         )
+        violence_note = f"暴力程度：{violence_level}/10\n" if violence_level > 0 else ""
+        eroticism_note = f"色情程度：{eroticism_level}/10\n" if eroticism_level > 0 else ""
         user_message = (
             f"故事类型：{genre_label}\n"
             + (f"风格：{style}\n" if style else "")
             + f"故事简介：{premise}\n"
-            f"主要角色：{', '.join(char_names)}\n\n"
-            f"已有章节大纲（共{len(existing_chapters)}章）：\n{existing_summary}\n\n"
+            f"主要角色：{', '.join(char_names)}\n"
+            + violence_note + eroticism_note
+            + f"\n已有章节大纲（共{len(existing_chapters)}章）：\n{existing_summary}\n\n"
             f"请从第{start_index}章（索引{start_index}）开始，续写{additional_chapters}章大纲，剧情要承接上文。"
         )
         result = await self.stream_chat_json(system_prompt, user_message, max_tokens=4096, usage_callback=usage_callback)