@waterwx/dsh-novel-forge 1.5.0 → 1.6.0

package/lib/index.js

@@ -2116,10 +2116,13 @@ async function extractRoles(ctx, config, project) {
 	return roles;
 }
 /** ✨ AI 从全书提炼场景库：正文/编年录 → 高频重要场景的结构化视觉锚点。 */
-async function extractScenes(ctx, config, project) {
+async function extractScenes(ctx, config, project, styleId, filterId) {
+	const styleWords = styleKeywords(styleId, filterId);
 	const system = [
 		"你是一位网文漫剧场景导演。根据本书的大纲、编年录与已写章节摘录，提炼「镜头场景」——漫剧分镜/生图时每个镜头要知道\"在哪、什么时间光态、拍什么情节、人物什么状态\"。",
 		"按五幕结构覆盖全书核心场景（短篇 10 章内可少于五幕，按实际剧情组织）：第一幕后场/核心主场、第二幕对外卖场/营业区、第三幕禁忌区域/地下深仓、第四幕外部人间世界（短暂出逃）、第五幕回归与抉择。",
+		"【当前视觉风格】（moment/palette/moods/zh/en 必须按此风格措辞，不能写中性描述）：" + styleWords,
+		"重要：每个场景的 zh 提示词段首必须原样嵌入【当前视觉风格】词块原文；en 提示词末尾追加风格标签。不得省略或改写。",
 		"每个场景必须是「镜头场景」而非仅场地：给出该场景的关键情节镜头（人物动作+情绪+镜头推进）。",
 		"数量控制：最多输出 8 个场景；每个场景必须能对应到正文实际出现过的地点与情节，不得凭空虚构。",
 		"每个场景输出：",
@@ -2177,10 +2180,11 @@ async function extractScenes(ctx, config, project) {
 			elements: strArr(entry.elements).map((t) => t.slice(0, 60)).slice(0, 6),
 			palette: strArr(entry.palette).map((t) => t.slice(0, 40)).slice(0, 4),
 			moods: strArr(entry.moods).map((t) => t.slice(0, 20)).slice(0, 4),
-			zh: str(entry.zh).slice(0, 800),
-			en: str(entry.en).slice(0, 600),
+			zh: ensureStyleEmbedded(str(entry.zh).slice(0, 800), styleWords, "zh"),
+			en: ensureStyleEmbedded(str(entry.en).slice(0, 600), styleWords, "en"),
 			tags: strArr(entry.tags).map((t) => t.slice(0, 24)).slice(0, 6),
-			source: str(entry.source).slice(0, 120)
+			source: str(entry.source).slice(0, 120),
+			styleId
 		});
 	}
 	return scenes;
@@ -2354,6 +2358,104 @@ function importBookTextFromText(raw, outputDir, bookName) {
 function importBookText(filePath, outputDir) {
 	return importBookTextFromText(decodeTextSmart(readFileSync(filePath)), outputDir, basename(filePath, extname(filePath)).slice(0, 40) || "导入小说");
 }
+/** 常见职业/身份尾缀：角色名如「周野律师」正文可能只写「周野」或「周野的律师」。 */
+const ROLE_NAME_SUFFIXES = [
+	"律师",
+	"辩护律师",
+	"医生",
+	"老师",
+	"教授",
+	"先生",
+	"女士",
+	"小姐",
+	"警官",
+	"警察",
+	"局长",
+	"总经理",
+	"经理",
+	"老板",
+	"师父",
+	"师傅",
+	"道长",
+	"老祖",
+	"长老",
+	"掌门",
+	"少主",
+	"公主",
+	"王子",
+	"王妃",
+	"皇后",
+	"皇帝",
+	"王爷",
+	"公子",
+	"姑娘",
+	"夫人",
+	"太太",
+	"大人",
+	"将军",
+	"丞相",
+	"尚书",
+	"员外"
+];
+/**
+* 从角色名/身份拆出正文可能使用的检索词：
+* - specific：具体称谓（含职业尾缀或 3 字以上身份片段），如「周野的律师」「辩护律师」「律师」——优先用，避免误抓到同名主干（周野）的段落；
+* - stems：名字主干（如「周野」），最后兜底。
+*/
+function roleFallbackTokens(name, identity) {
+	const specific = /* @__PURE__ */ new Set();
+	const stems = /* @__PURE__ */ new Set();
+	const addName = (s) => {
+		const t = s.trim();
+		if (t.length < 2) return;
+		specific.add(t);
+		for (const q of [
+			"辩护",
+			"助理",
+			"高级",
+			"首席",
+			"御用",
+			"御前",
+			"专职"
+		]) if (t.includes(q)) specific.add(t.replace(q, ""));
+		for (const suf of ROLE_NAME_SUFFIXES) if (t.endsWith(suf) && t.length > suf.length + 1) {
+			const stem = t.slice(0, -suf.length);
+			specific.add(stem + "的" + suf);
+			specific.add(suf);
+			if (stem.length >= 2) stems.add(stem);
+		}
+	};
+	addName(name);
+	for (const part of (identity ?? "").split(/[的，,。\s]+/)) {
+		const p = part.trim();
+		if (p.length < 2) continue;
+		if (p.length >= 3 || ROLE_NAME_SUFFIXES.some((s) => p.endsWith(s))) {
+			specific.add(p);
+			for (const suf of ROLE_NAME_SUFFIXES) if (p.endsWith(suf) && p.length > suf.length + 1) {
+				specific.add(suf);
+				stems.add(p.slice(0, -suf.length));
+			}
+		} else stems.add(p);
+	}
+	return {
+		specific: [...specific].filter((t) => t.length >= 2),
+		stems: [...stems].filter((t) => t.length >= 2)
+	};
+}
+/** 保证风格词块已内嵌（LLM 偶发漏嵌时兜底）：zh 段首前缀，en 末尾追加。 */
+function ensureStyleEmbedded(text, styleWords, lang) {
+	const t = text.trim();
+	if (styleWords === "" || t === "") return t;
+	if (t.includes(styleWords)) return t;
+	return lang === "zh" ? styleWords + "，" + t : t + "，" + styleWords;
+}
+/** 对 zh/en 一对提示词统一补风格词块。 */
+function withStyle(pair, styleWords) {
+	return {
+		zh: ensureStyleEmbedded(pair.zh, styleWords, "zh"),
+		en: ensureStyleEmbedded(pair.en, styleWords, "en")
+	};
+}
 /**
 * 为单个角色提炼「动漫形象描述词」：扫描该角色出场的已写章节正文，
 * 截取含外貌描写的段落，交给 LLM 提炼中文描述 + 英文绘图标签。
@@ -2361,6 +2463,7 @@ function importBookText(filePath, outputDir) {
 async function extractRoleVisual(ctx, config, project, outputDir, roleName, styleId, filterId) {
 	const role = (project.roles ?? []).find((r) => r.name === roleName);
 	if (role === void 0) throw new Error(`角色「${roleName}」不在角色库中`);
+	role.promptStyleId = styleId;
 	const appearanceHints = /(发|眉|眼|眸|脸|肤|唇|身材|身高|衣|袍|裙|衫|靴|腰带|气质|模样|长相|容貌|披|束|扎|戴|佩|挂|绣|青|白|黑|红|蓝|紫|灰|银|金|少年|青年|少女|汉子|老者|中年|纤细|挺拔|瘦削|壮实|清秀|俊朗|英气|阴鸷|慈眉)/;
 	const roleText = `${role.name} ${role.identity ?? ""}`;
 	const colorWords = [
@@ -2427,33 +2530,48 @@ async function extractRoleVisual(ctx, config, project, outputDir, roleName, styl
 		...roleWords.filter((w) => roleText.includes(w)),
 		...locationWords.filter((w) => roleText.includes(w))
 	])).filter((t) => t.length >= 2);
-	const matchesRole = (para) => para.includes(roleName) || searchTokens.some((tok) => para.includes(tok));
+	const matchesRole = (para, tokens) => para.includes(roleName) || tokens.some((tok) => para.includes(tok));
 	const excerpts = [];
-	const written = project.chapters.filter((c) => c.status !== "pending" && c.status !== "generating" && c.file !== void 0).slice(-60);
-	for (const chapter of written) {
-		const body = readChapterFile(outputDir, chapter);
-		if (body === void 0) continue;
-		const paras = body.split(/\n{2,}/).map((p) => p.trim()).filter((p) => p.length > 0);
-		let perChapter = 0;
-		for (const para of paras) {
-			if (perChapter >= 2 || excerpts.length >= 12) break;
-			if (!matchesRole(para)) continue;
-			if (appearanceHints.test(para) || excerpts.length < 4) {
-				excerpts.push({
-					no: chapter.no,
-					text: para.slice(0, 220)
-				});
-				perChapter++;
+	const written = project.chapters.filter((c) => c.status !== "pending" && c.status !== "generating" && c.file !== void 0);
+	/** 扫描一批章节，收集该角色出场且可能含外貌描写的段落（每章最多 2 段，共 12 段）。 */
+	const scanChapters = (list, tokens) => {
+		for (const chapter of list) {
+			if (excerpts.length >= 12) break;
+			const body = readChapterFile(outputDir, chapter);
+			if (body === void 0) continue;
+			const paras = body.split(/\n{2,}/).map((p) => p.trim()).filter((p) => p.length > 0);
+			let perChapter = 0;
+			for (const para of paras) {
+				if (perChapter >= 2 || excerpts.length >= 12) break;
+				if (!matchesRole(para, tokens)) continue;
+				if (appearanceHints.test(para) || excerpts.length < 4) {
+					excerpts.push({
+						no: chapter.no,
+						text: para.slice(0, 220)
+					});
+					perChapter++;
+				}
 			}
 		}
-		if (excerpts.length >= 12) break;
+	};
+	scanChapters(written.slice(-60), searchTokens);
+	if (excerpts.length === 0) scanChapters(written, searchTokens);
+	if (excerpts.length === 0) {
+		const fb = roleFallbackTokens(role.name, role.identity);
+		if (fb.specific.length > 0) scanChapters(written, fb.specific);
+	}
+	if (excerpts.length === 0) {
+		const fb = roleFallbackTokens(role.name, role.identity);
+		if (fb.stems.length > 0) scanChapters(written, fb.stems);
 	}
-	if (excerpts.length === 0) throw new Error(`正文中未找到「${roleName}」的出场描写（仅搜索最近 60 章），请确认角色名与正文一致`);
+	if (excerpts.length === 0) throw new Error(`正文中未找到「${roleName}」的出场描写（已扫描全书及角色名拆分词），请确认角色名与正文一致，或该角色尚未在正文出场`);
 	const rules = project.visualRules ?? [];
+	const styleWords = styleKeywords(styleId, filterId);
 	const raw = parseJsonObject(await complete(ctx, config, {
 		system: [
 			"你是一位动漫角色设定师与 AI 绘图提示词工程师。根据网文正文中该角色的实际外貌描写，输出「形象锚点」与「四类生图提示词」——一次完成，用于 AI 绘图（NovelAI / Stable Diffusion / Midjourney / 豆包等）生成一致的角色立绘。",
-			"【当前视觉风格】（portrait 立绘的「风格」字段必须原样使用，sheet/expressions/details 同样内嵌）：" + styleKeywords(styleId, filterId),
+			"【当前视觉风格】（portrait 立绘的「风格」字段必须原样使用，sheet/expressions/details 同样内嵌）：" + styleWords,
+			"重要：每段 zh 提示词的段首必须原样嵌入【当前视觉风格】词块原文；en 提示词末尾追加风格标签。不得省略或改写。",
 			"硬性要求（依据优先）：",
 			"1. 发色/发型/瞳色/服装/气质/标志物必须来自提供的正文段落，不得凭空发明。",
 			"2. 正文未明确写到的项目（如瞳色没写、身高没写），用「未定」标注或直接不写，不要编造数值。",
@@ -2489,8 +2607,8 @@ async function extractRoleVisual(ctx, config, project, outputDir, roleName, styl
 		maxTokens: Math.max(config.maxTokens, 12e3),
 		reasoning: config.analysisReasoning ?? "low"
 	}));
-	const zh = typeof raw.zh === "string" ? raw.zh.trim().slice(0, 500) : "";
-	const en = typeof raw.en === "string" ? raw.en.trim().slice(0, 1500) : "";
+	let zh = typeof raw.zh === "string" ? raw.zh.trim().slice(0, 500) : "";
+	let en = typeof raw.en === "string" ? raw.en.trim().slice(0, 1500) : "";
 	const tags = Array.isArray(raw.tags) ? raw.tags.filter((t) => typeof t === "string" && t.trim() !== "").map((t) => t.trim().slice(0, 20)).slice(0, 12) : [];
 	const source = typeof raw.source === "string" ? raw.source.trim().slice(0, 300) : "";
 	const expressions = Array.isArray(raw.expressions) ? raw.expressions.filter((e) => typeof e === "string" && e.trim() !== "").map((e) => e.trim().slice(0, 12)).slice(0, 12) : [];
@@ -2520,6 +2638,17 @@ async function extractRoleVisual(ctx, config, project, outputDir, roleName, styl
 		promptKit = void 0;
 	}
 	if (zh === "" || en === "") throw new Error("形象描述提炼失败：LLM 未返回有效 JSON");
+	zh = ensureStyleEmbedded(zh, styleWords, "zh");
+	en = ensureStyleEmbedded(en, styleWords, "en");
+	if (promptKit !== void 0) promptKit = {
+		portrait: withStyle(promptKit.portrait, styleWords),
+		sheet: withStyle(promptKit.sheet, styleWords),
+		expressions: promptKit.expressions.map((e) => ({
+			...e,
+			...withStyle(e, styleWords)
+		})),
+		details: withStyle(promptKit.details, styleWords)
+	};
 	return {
 		zh,
 		en,
@@ -2537,11 +2666,14 @@ async function generateRolePromptKit(ctx, config, project, roleName, styleId, fi
 	const role = (project.roles ?? []).find((r) => r.name === roleName);
 	if (role === void 0) throw new Error(`角色「${roleName}」不在角色库中`);
 	if (role.imagePrompt === void 0) throw new Error(`角色「${roleName}」还没有形象锚点，请先生成锚点`);
+	role.promptStyleId = styleId;
 	const rules = project.visualRules ?? [];
+	const styleWords = styleKeywords(styleId, filterId);
 	const raw = parseJsonObject(await complete(ctx, config, {
 		system: [
 			"你是一位 AI 绘图提示词工程师。基于给定角色的形象锚点、表情清单与本书视觉规则，输出四类生图提示词（每类 zh+en 各一段）。",
-			"【当前视觉风格】（每类提示词必须内嵌）：" + styleKeywords(styleId, filterId),
+			"【当前视觉风格】（每类提示词必须内嵌）：" + styleWords,
+			"重要：每段 zh 提示词的段首必须原样嵌入【当前视觉风格】词块原文；en 提示词末尾追加风格标签。不得省略或改写。",
 			"四类：",
 			"1. portrait 立绘：正面站立全身人像，3D动漫超精细建模，纯白纯色背景；按字段流组织：身份（男子，角色名，外表年龄）→ 发型发色 → 胡茬 → 眼眸 → 面部 → 气质 → 上身服装分件（含磨损）→ 下身 → 鞋 → 标志物（含细节）→ 收尾（角色设计稿，细节完整展示，无多余杂物，全身完整无裁切）。",
 			"2. sheet 四视图：同一角色的 正面/左侧面/右侧面/背面 四视图设定表（character sheet），纯白背景，四个视角分别描述。",
@@ -2575,14 +2707,16 @@ async function generateRolePromptKit(ctx, config, project, roleName, styleId, fi
 	};
 	const expressionsRaw = Array.isArray(raw.expressions) ? raw.expressions.filter((e) => typeof e === "object" && e !== null) : [];
 	const kit = {
-		portrait: pair(raw.portrait),
-		sheet: pair(raw.sheet),
+		portrait: withStyle(pair(raw.portrait), styleWords),
+		sheet: withStyle(pair(raw.sheet), styleWords),
 		expressions: expressionsRaw.map((e) => ({
 			name: str(e.name).slice(0, 12) || "表情",
-			zh: str(e.zh).slice(0, 800),
-			en: str(e.en).slice(0, 1200)
+			...withStyle({
+				zh: str(e.zh).slice(0, 800),
+				en: str(e.en).slice(0, 1200)
+			}, styleWords)
 		})).slice(0, 12),
-		details: pair(raw.details)
+		details: withStyle(pair(raw.details), styleWords)
 	};
 	if (kit.portrait.zh === "" || kit.portrait.en === "") throw new Error("提示词精修失败：LLM 未返回有效 JSON");
 	return kit;
@@ -3243,15 +3377,22 @@ async function summarizeChapter(ctx, config, project, outputDir, chapterNo) {
 	saveProject(outputDir, project);
 	return chapter.summary;
 }
-/** 把分镜产出写入项目持久化（按章 upsert）。 */
+/** 把分镜产出写入项目持久化（按章 upsert）。重新生成上游时级联清掉下游旧产物：
+* 新骨架 → 清分镜表+提示词；新分镜表 → 清提示词；新提示词不动上游。 */
 function saveChapterStoryboard(project, outputDir, entry) {
 	if (project.storyboards === void 0) project.storyboards = [];
 	const idx = project.storyboards.findIndex((e) => e.chapterNo === entry.chapterNo);
-	if (idx === -1) project.storyboards.push(entry);
-	else project.storyboards[idx] = {
-		...project.storyboards[idx],
+	const next = {
+		...(idx === -1 ? void 0 : project.storyboards[idx]) ?? {},
 		...entry
 	};
+	if (entry.skeleton !== void 0) {
+		next.table = void 0;
+		next.prompts = void 0;
+	}
+	if (entry.table !== void 0) next.prompts = void 0;
+	if (idx === -1) project.storyboards.push(next);
+	else project.storyboards[idx] = next;
 	project.updatedAt = (/* @__PURE__ */ new Date()).toISOString();
 	saveProject(outputDir, project);
 }
@@ -3266,7 +3407,8 @@ async function generateStoryboardTable(ctx, config, project, outputDir, chapterN
 	if (body === void 0) throw new Error(`章节 ${chapterNo} 的正文文件不存在`);
 	if ((skeleton.beats ?? []).length === 0) throw new Error("骨架为空，请先生成剧情骨架");
 	const roles = (project.roles ?? []).filter((r) => body.includes(r.name)).slice(0, 8).map((r) => `${r.name}（${r.roleLabel === "protagonist" ? "主角" : r.roleLabel === "antagonist" ? "反派" : r.roleLabel === "female_lead" ? "女主" : "配角"}）：${r.identity ?? ""}`);
-	const scenes = (project.scenes ?? []).filter((s) => body.includes(s.name)).slice(0, 6).map((s) => `${s.name}：${s.summary ?? ""}`);
+	const usedScenes = (project.scenes ?? []).filter((s) => body.includes(s.name)).slice(0, 6);
+	const scenes = usedScenes.map((s) => `${s.name}：${s.summary ?? ""}`);
 	const rules = (project.visualRules ?? []).map((r) => "- " + r);
 	const baseStyle = styleId !== void 0 ? findStyle(styleId) : void 0;
 	const filterStyle = filterId !== void 0 ? findStyle(filterId) : void 0;
@@ -3332,7 +3474,8 @@ async function generateStoryboardTable(ctx, config, project, outputDir, chapterN
 	if (missing.length > 0) console.warn(`[dsh-novel-forge] storyboard: beat ${missing.join(",")} 无镜头覆盖`);
 	const table = {
 		chapterNo,
-		shots
+		shots,
+		usedScenes: usedScenes.map((s) => s.name)
 	};
 	saveChapterStoryboard(project, outputDir, {
 		chapterNo,
@@ -3364,6 +3507,7 @@ async function generateStoryboardPrompts(ctx, config, project, outputDir, chapte
 			"任务：把分镜表的每个镜头写成一段可直接粘贴进视频生成工具的提示词。",
 			"提示词结构（按顺序）：风格词块 → 画面主体（人物动作+表情+服装标志物）→ 机位运镜（含时长感）→ 光效氛围。单段中文 60-140 字，逗号分隔，流畅自然。",
 			"风格词块（必须原样放在每段开头）：" + stylePrefix,
+			"重要：每段提示词开头必须原样嵌入风格词块原文，不得省略、不得改写。",
 			"硬性要求：",
 			"1. 只写画面与镜头，禁止写音频/台词配音说明（视频模型无音频）。",
 			"2. 服装/发色/标志物必须沿用镜头表与视觉规则，禁止自行更换。",
@@ -3385,7 +3529,7 @@ async function generateStoryboardPrompts(ctx, config, project, outputDir, chapte
 	const shotIds = new Set(table.shots.map((s) => s.id));
 	const prompts = Array.isArray(raw.prompts) ? raw.prompts.filter((v) => typeof v === "object" && v !== null).map((entry) => ({
 		shotId: shotIds.has(entry.shotId) ? entry.shotId : "",
-		text: typeof entry.text === "string" ? entry.text.trim().slice(0, 400) : ""
+		text: ensureStyleEmbedded(typeof entry.text === "string" ? entry.text.trim().slice(0, 400) : "", stylePrefix, "zh")
 	})).filter((x) => x.shotId !== "" && x.text !== "") : [];
 	if (prompts.length === 0) throw new Error("模型未输出有效提示词，请重试");
 	saveChapterStoryboard(project, outputDir, {
@@ -7542,7 +7686,7 @@ function makeRoutes(deps) {
 			if (project.scenes === void 0) project.scenes = [];
 			const op = body?.op;
 			if (op === "extract") try {
-				const candidates = await extractScenes(ctx, config, project);
+				const candidates = await extractScenes(ctx, config, project, body?.styleId, body?.filterId);
 				writeJson(res, 200, {
 					scenes: project.scenes,
 					candidates