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

This commit is contained in:
2026-03-13 13:36:55 +08:00
parent 0d63d0e6d1
commit 0a12f204ba
6 changed files with 129 additions and 11 deletions

View File

@@ -180,6 +180,7 @@ async def generate_synopsis(
"你是一位专业的小说策划师,擅长根据创作参数生成引人入胜的故事简介。"
"请根据用户提供的类型、风格、主角、冲突等参数生成一段200-400字的中文故事简介。"
"简介需涵盖:世界观背景、主角基本情况、核心矛盾冲突、故事基调。"
"暴力程度和色情程度数值越高,简介中相关情节描写越多、越直接。"
"直接输出简介正文,不要加任何前缀标题或说明文字。"
)
parts = [f"类型:{data.genre}"]
@@ -193,6 +194,10 @@ async def generate_synopsis(
parts.append(f"冲突规模:{data.conflict_scale}")
parts.append(f"角色数量:约{data.num_characters}个主要角色")
parts.append(f"故事体量:约{data.num_chapters}")
if data.violence_level > 0:
parts.append(f"暴力程度:{data.violence_level}/10")
if data.eroticism_level > 0:
parts.append(f"色情程度:{data.eroticism_level}/10")
user_message = "\n".join(parts) + "\n\n请生成故事简介:"
try:
@@ -294,6 +299,7 @@ async def generate_synopsis_nsfw(
"你是一位专业的成人小说策划师,擅长根据创作参数生成引人入胜的故事简介。"
"请根据用户提供的类型、风格、主角、冲突等参数生成一段200-400字的中文故事简介。"
"简介需涵盖:世界观背景、主角基本情况、核心矛盾冲突、故事基调。"
"暴力程度和色情程度数值越高,简介中相关情节描写越多、越露骨直接。"
"直接输出简介正文,不要加任何前缀标题或说明文字。"
)
parts = [f"类型:{data.genre}"]
@@ -307,6 +313,10 @@ async def generate_synopsis_nsfw(
parts.append(f"冲突规模:{data.conflict_scale}")
parts.append(f"角色数量:约{data.num_characters}个主要角色")
parts.append(f"故事体量:约{data.num_chapters}")
if data.violence_level > 0:
parts.append(f"暴力程度:{data.violence_level}/10")
if data.eroticism_level > 0:
parts.append(f"色情程度:{data.eroticism_level}/10")
user_message = "\n".join(parts) + "\n\n请生成故事简介:"
try: