208 lines
7.7 KiB
Python
208 lines
7.7 KiB
Python
"""
|
|
小说生成器配置文件
|
|
包含各种高级配置选项和预设模板
|
|
"""
|
|
|
|
from dataclasses import dataclass
|
|
from typing import Dict, List, Any, Optional
|
|
|
|
@dataclass
|
|
class AdvancedNovelConfig:
|
|
"""高级小说生成配置"""
|
|
# 基础配置
|
|
genre: str
|
|
target_total_chars: int
|
|
model: str = "qwen-turbo"
|
|
temperature: float = 0.8
|
|
|
|
# 质量控制
|
|
enable_logic_check: bool = True
|
|
max_logic_iterations: int = 3
|
|
enable_emotion_enhancement: bool = True
|
|
enable_dialogue_optimization: bool = True
|
|
|
|
# 写作风格
|
|
writing_style: str = "balanced" # balanced, literary, commercial, dramatic
|
|
narrative_perspective: str = "third" # first, third, mixed
|
|
tone: str = "neutral" # light, neutral, serious, humorous
|
|
|
|
# 内容控制
|
|
max_characters: int = 8 # 最大角色数量
|
|
min_dialogue_ratio: float = 0.3 # 对话占比最小值
|
|
max_description_ratio: float = 0.4 # 描写占比最大值
|
|
|
|
# 技术参数
|
|
segment_size: int = 2000 # 分段处理大小
|
|
parallel_workers: int = 3 # 并行处理线程数
|
|
retry_attempts: int = 3 # 重试次数
|
|
|
|
|
|
# 预设配置模板
|
|
PRESET_CONFIGS = {
|
|
"精品长篇": AdvancedNovelConfig(
|
|
genre="都市逆袭",
|
|
target_total_chars=25000,
|
|
model="qwen-max",
|
|
temperature=0.9,
|
|
enable_logic_check=True,
|
|
max_logic_iterations=5,
|
|
writing_style="literary",
|
|
enable_emotion_enhancement=True,
|
|
enable_dialogue_optimization=True,
|
|
max_characters=12,
|
|
parallel_workers=4
|
|
)
|
|
}
|
|
|
|
# 写作风格配置
|
|
WRITING_STYLES = {
|
|
"balanced": {
|
|
"description": "平衡风格，适合大众阅读",
|
|
"dialogue_ratio": 0.35,
|
|
"description_ratio": 0.30,
|
|
"action_ratio": 0.35,
|
|
"complexity": "medium"
|
|
},
|
|
|
|
"literary": {
|
|
"description": "文学风格，注重深度和内涵",
|
|
"dialogue_ratio": 0.25,
|
|
"description_ratio": 0.45,
|
|
"action_ratio": 0.30,
|
|
"complexity": "high"
|
|
},
|
|
|
|
"commercial": {
|
|
"description": "商业风格，节奏快，易读",
|
|
"dialogue_ratio": 0.45,
|
|
"description_ratio": 0.25,
|
|
"action_ratio": 0.30,
|
|
"complexity": "low"
|
|
},
|
|
|
|
"dramatic": {
|
|
"description": "戏剧风格，冲突强烈",
|
|
"dialogue_ratio": 0.50,
|
|
"description_ratio": 0.20,
|
|
"action_ratio": 0.30,
|
|
"complexity": "medium"
|
|
}
|
|
}
|
|
|
|
# 语调配置
|
|
TONE_CONFIGS = {
|
|
"light": {
|
|
"description": "轻松愉快的语调",
|
|
"keywords": ["温馨", "轻松", "幽默", "治愈"],
|
|
"avoid_keywords": ["沉重", "压抑", "悲伤"]
|
|
},
|
|
|
|
"neutral": {
|
|
"description": "中性平衡的语调",
|
|
"keywords": ["自然", "真实", "平衡"],
|
|
"avoid_keywords": ["过于夸张", "过于沉重"]
|
|
},
|
|
|
|
"serious": {
|
|
"description": "严肃深刻的语调",
|
|
"keywords": ["深刻", "思考", "内省", "成长"],
|
|
"avoid_keywords": ["轻浮", "肤浅"]
|
|
},
|
|
|
|
"humorous": {
|
|
"description": "幽默风趣的语调",
|
|
"keywords": ["幽默", "风趣", "机智", "轻松"],
|
|
"avoid_keywords": ["严肃", "沉重", "悲伤"]
|
|
}
|
|
}
|
|
|
|
# 题材扩展配置
|
|
EXTENDED_GENRES = {
|
|
"都市逆袭": {
|
|
"core": "底层主角通过智慧和机遇逐步崛起，每一步都有合理动机和代价",
|
|
"typical_characters": ["奋斗主角", "贵人导师", "竞争对手", "支持朋友", "家人"],
|
|
"common_conflicts": ["阶层差距", "能力质疑", "机遇选择", "道德考验"],
|
|
"emotional_themes": ["自我证明", "成长蜕变", "责任担当", "情感成熟"]
|
|
},
|
|
|
|
"甜宠先婚后爱": {
|
|
"core": "两个有血有肉的人从陌生到相知相爱的心理变化过程",
|
|
"typical_characters": ["独立女主", "温柔男主", "闺蜜", "家人", "工作伙伴"],
|
|
"common_conflicts": ["误解矛盾", "价值观差异", "外界阻力", "内心恐惧"],
|
|
"emotional_themes": ["信任建立", "情感觉醒", "相互理解", "共同成长"]
|
|
},
|
|
|
|
"悬疑推理": {
|
|
"core": "通过逻辑推理和线索收集，逐步揭开真相的智力较量过程",
|
|
"typical_characters": ["推理主角", "神秘反派", "关键证人", "警方助手", "受害者"],
|
|
"common_conflicts": ["线索迷雾", "逻辑陷阱", "时间压力", "真相冲击"],
|
|
"emotional_themes": ["真相追求", "正义坚持", "智慧较量", "人性探索"]
|
|
},
|
|
|
|
"末世生存": {
|
|
"core": "在绝望环境中求生，展现人性光辉与黑暗的生存史诗",
|
|
"typical_characters": ["生存主角", "幸存团队", "资源争夺者", "希望守护者", "堕落者"],
|
|
"common_conflicts": ["资源争夺", "环境威胁", "人性考验", "道德选择"],
|
|
"emotional_themes": ["生存意志", "人性光辉", "团队协作", "希望重燃"]
|
|
},
|
|
|
|
"玄幻升级": {
|
|
"core": "在奇幻世界中通过修炼和历练不断突破自我的成长之路",
|
|
"typical_characters": ["修炼主角", "师父导师", "同门师兄", "强敌对手", "红颜知己"],
|
|
"common_conflicts": ["境界瓶颈", "强敌挑战", "门派争斗", "天道考验"],
|
|
"emotional_themes": ["修炼感悟", "师徒情深", "友情考验", "爱情升华"]
|
|
},
|
|
|
|
"校园青春": {
|
|
"core": "青春期的成长、友情、初恋和自我发现的故事",
|
|
"typical_characters": ["青涩主角", "初恋对象", "好友团体", "老师", "家长"],
|
|
"common_conflicts": ["学业压力", "情感困惑", "友情考验", "家庭期待"],
|
|
"emotional_themes": ["青春迷茫", "初恋美好", "友情珍贵", "成长痛苦"]
|
|
},
|
|
|
|
"职场励志": {
|
|
"core": "职场新人或转型者在工作中的成长和突破",
|
|
"typical_characters": ["职场新人", "严厉上司", "竞争同事", "导师前辈", "客户"],
|
|
"common_conflicts": ["工作挑战", "人际关系", "职业选择", "价值冲突"],
|
|
"emotional_themes": ["专业成长", "自我价值", "团队合作", "领导力"]
|
|
},
|
|
|
|
"家庭温情": {
|
|
"core": "家庭成员之间的情感纠葛和最终和解",
|
|
"typical_characters": ["家庭成员", "长辈", "晚辈", "亲戚", "邻居"],
|
|
"common_conflicts": ["代沟问题", "价值观差异", "生活压力", "情感隔阂"],
|
|
"emotional_themes": ["亲情珍贵", "理解包容", "传承责任", "家庭和谐"]
|
|
}
|
|
}
|
|
|
|
def get_preset_config(preset_name: str) -> Optional[AdvancedNovelConfig]:
|
|
"""获取预设配置"""
|
|
return PRESET_CONFIGS.get(preset_name)
|
|
|
|
def get_writing_style_info(style: str) -> Dict[str, Any]:
|
|
"""获取写作风格信息"""
|
|
return WRITING_STYLES.get(style, WRITING_STYLES["balanced"])
|
|
|
|
def get_tone_info(tone: str) -> Dict[str, Any]:
|
|
"""获取语调信息"""
|
|
return TONE_CONFIGS.get(tone, TONE_CONFIGS["neutral"])
|
|
|
|
def get_genre_info(genre: str) -> Dict[str, Any]:
|
|
"""获取题材信息"""
|
|
return EXTENDED_GENRES.get(genre, EXTENDED_GENRES["都市逆袭"])
|
|
|
|
def list_available_presets() -> List[str]:
|
|
"""列出可用的预设配置"""
|
|
return list(PRESET_CONFIGS.keys())
|
|
|
|
def list_available_genres() -> List[str]:
|
|
"""列出可用的题材"""
|
|
return list(EXTENDED_GENRES.keys())
|
|
|
|
def list_available_styles() -> List[str]:
|
|
"""列出可用的写作风格"""
|
|
return list(WRITING_STYLES.keys())
|
|
|
|
def list_available_tones() -> List[str]:
|
|
"""列出可用的语调"""
|
|
return list(TONE_CONFIGS.keys()) |