宜博 / lemonboss

Browse Code »

Commit 17e6455eeb4d9d53fd707a706b8f01d9becf597b

Authored by yibo 2026-03-26 15:44:45 +0800

1 parent 5261d0d2

docs: add create-drama documentation and scripts

Inline Side-by-side

Showing 5 changed files with 1124 additions and 0 deletions

docs/create-drama/.env 0 → 100644

View file @17e6455

	1	+# === MySQL 数据库 ===
	2	+DB_HOST=rm-2zeel4sa31na88gd4qo.rwlb.rds.aliyuncs.com
	3	+DB_PORT=3306
	4	+DB_USER=lemon
	5	+DB_PASSWORD=Lemon@123456
	6	+DB_NAME=shadiao_drama
	7	+
	8	+# === 阿里云 OSS（图片上传） ===
	9	+OSS_REGION=oss-cn-beijing
	10	+OSS_ACCESS_KEY_ID=LTAI5tGtipK3GrbTwZm7JjTS
	11	+OSS_ACCESS_KEY_SECRET=r6axVSGVyIzL3Ty2T5pV1SVFeiir3t
	12	+OSS_BUCKET=shadiao-drama
	13	+OSS_ENDPOINT=https://oss-cn-beijing.aliyuncs.com
	14	+
	15	+# === 万界生图 API（Jimeng + Nano Banana Pro） ===
	16	+WANJIE_API_KEY=eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJleHAiOjE3NzA5NjIyNzgsImtleSI6IjVLNzVaOFROQzlGNEhNMzdQOVk3In0.2axgWHccUPjL3cQr2-1jAGLZ4hVAcTbn-LU78CMMex4
	17	+USER_ID=1
0	18	\ No newline at end of file
...	...

docs/create-drama/SKILL.md 0 → 100644

View file @17e6455

	1	+---
	2	+description: 从素材（新闻、故事、段子等）自动生成 classic 互动剧本（含角色图、场景图）并写入数据库
	3	+---
	4	+
	5	+# create-drama — 从素材自动生成完整 classic 互动剧本
	6	+
	7	+根据用户提供的内容素材（新闻、故事、段子、历史事件等），生成一个完整的 classic 类型互动剧本，包括：
	8	+- 剧本数据（角色、台词、表情、动作）
	9	+- 角色立绘图（AI 生图 → 去背景 → 上传 OSS）
	10	+- 背景图和封面图（AI 生图 → 上传 OSS）
	11	+- 写入数据库
	12	+
	13	+## 输入
	14	+
	15	+用户会在 `$ARGUMENTS` 中提供内容素材。素材可能是：
	16	+- 一段新闻报道
	17	+- 一个故事/段子
	18	+- 一个历史事件描述
	19	+- 一个创意想法
	20	+- 一个文件路径（需要先读取文件内容）
	21	+
	22	+## 执行步骤
	23	+
	24	+### 第 1 步：分析素材
	25	+
	26	+仔细阅读用户提供的素材内容。如果是文件路径，先用 Read 工具读取文件。
	27	+
	28	+### 第 2 步：生成完整剧本 JSON
	29	+
	30	+根据素材内容，你自己（Claude）直接生成剧本数据 JSON。不要调用任何外部 LLM API。
	31	+
	32	+生成的 JSON 必须严格遵循以下格式：
	33	+
	34	+```json
	35	+{
	36	+ "name": "剧本名称（简短，2-6个字）",
	37	+ "title": "显示标题（可以带 emoji，稍微长一点）",
	38	+ "subtitle": "— 副标题 —",
	39	+ "description": "剧本简介（一句话概括）",
	40	+ "systemPrompt": "用于后续 AI 续写的系统提示词（详见下方规则）",
	41	+ "isPublic": false,
	42	+ "userId": 1,
	43	+ "tags": "标签1,标签2,标签3",
	44	+ "characters": [
	45	+ {
	46	+ "name": "角色名",
	47	+ "gender": "male 或 female",
	48	+ "voiceId": "从音色列表选择",
	49	+ "imagePrompt": "角色立绘生图 prompt（详见下方规则）"
	50	+ }
	51	+ ],
	52	+ "script": [
	53	+ {
	54	+ "character": "角色名（必须在 characters 中定义过）",
	55	+ "dialogue": "台词内容",
	56	+ "avatar": "表情文件名",
	57	+ "action": "动画效果",
	58	+ "position": "left 或 right"
	59	+ }
	60	+ ],
	61	+ "backgroundPrompt": "背景图生图 prompt",
	62	+ "coverPrompt": "封面图生图 prompt"
	63	+}
	64	+```
	65	+
	66	+### 生成规则
	67	+
	68	+#### 角色 (characters)
	69	+
	70	+- 数量：2-5 个
	71	+- 每个角色必须有 name、gender、voiceId、imagePrompt
	72	+- voiceId 选择：参考 [reference.md](reference.md) 中的完整音色列表
	73	+ - 优先使用多情感音色（支持 happy/sad/angry 等情绪切换，效果更好）
	74	+ - 根据角色性格、年龄、性别匹配合适的音色
	75	+ - 英文角色使用英文多情感音色（Candice, Serena, Glen, Sylus, Corey, Nadia）
	76	+ - 搞笑角色可用 IP 仿音（唐僧、猪八戒等）或趣味口音
	77	+ - 常用推荐：
	78	+ - 男主角: `zh_male_yangguangqingnian_emo_v2_mars_bigtts`（阳光青年）或 `zh_male_junlangnanyou_emo_v2_mars_bigtts`（俊朗男友）
	79	+ - 女主角: `zh_female_roumeinvyou_emo_v2_mars_bigtts`（柔美女友）或 `zh_female_gaolengyujie_emo_v2_mars_bigtts`（高冷御姐）
	80	+ - 反派/严肃: `zh_male_lengkugege_emo_v2_mars_bigtts`（冷酷哥哥）或 `zh_male_aojiaobazong_emo_v2_mars_bigtts`（傲娇霸总）
	81	+ - 搞笑: `ICL_zh_male_youmodaye_tob`（幽默大爷）或 `zh_male_jingqiangkanye_emo_mars_bigtts`（京腔侃爷）
	82	+ - 老年: `zh_female_popo_mars_bigtts`（婆婆）或 `ICL_zh_male_youmodaye_tob`（幽默大爷）
	83	+ - 英文男: `en_male_sylus_emo_v2_mars_bigtts`（Sylus）或 `en_male_glen_emo_v2_mars_bigtts`（Glen）
	84	+ - 英文女: `en_female_candice_emo_v2_mars_bigtts`（Candice）或 `en_female_skye_emo_v2_mars_bigtts`（Serena）
	85	+
	86	+#### 角色立绘 imagePrompt
	87	+
	88	+每个角色必须生成 imagePrompt，遵循以下模板：
	89	+
	90	+```
	91	+生成一张[人物名字或身份]的全身站立画像。
	92	+核心风格：沙雕动漫风格（meme style），2D简笔画，线条清晰粗犷，色彩平涂，带有幽默滑稽感，略微的大头小身体比例。
	93	+脸部要求（最重要）：脸部必须是纯肉色的完全空白，绝对不要画眼睛、鼻子、嘴巴或眉毛等任何五官，保留一个干净完整的面部轮廓，方便后期添加表情包。
	94	+服装与动作：角色穿着[根据角色描述填写服装特征]，手里拿着[根据角色身份推断合适的道具]，[一个有趣的站姿或动作]。
	95	+背景要求：纯色干净背景（如浅蓝灰色），不要任何环境元素或杂物，边缘清晰，方便直接抠图。
	96	+```
	97	+
	98	+根据角色具体描述，在 [] 处填入合适内容。
	99	+
	100	+#### 背景图 backgroundPrompt
	101	+
	102	+```
	103	+横版宽屏场景背景图，卡通风格，适合互动剧背景。场景：[根据剧本内容描述场景]。不要人物，色彩鲜明，气氛感强。
	104	+```
	105	+
	106	+#### 封面图 coverPrompt
	107	+
	108	+```
	109	+剧目封面海报。标题：[剧本标题]。展示[角色名]等角色的剪影轮廓，色彩鲜明，有标题感，卡通风格。
	110	+```
	111	+
	112	+#### 台词 (script)
	113	+
	114	+- 数量：10-20 句（根据素材丰富度调整，尽量多一些，让剧情完整）
	115	+- avatar 可选值：`normal.png`, `happy.png`, `very_happy.png`, `angry.png`, `sad.png`, `cry.png`, `shock.png`, `smirk.png`, `speechless.png`
	116	+- action 可选值：
	117	+ - `none` — 默认，无动画
	118	+ - `shake` — 震动，适合愤怒/惊吓/强调
	119	+ - `jump` — 跳跃，适合惊喜/激动
	120	+ - `spin` — 旋转，适合混乱/搞笑
	121	+ - `zoom-in` — 放大，适合重要台词/特写
	122	+- position: 每个角色应该有固定站位（left/right），2个角色时一左一右，3个以上轮替
	123	+- 台词要生动有趣，符合互动剧的沙雕风格
	124	+- 对话要有戏剧冲突和反转
	125	+- 每句台词不要太长，控制在 30 字以内
	126	+
	127	+#### systemPrompt
	128	+
	129	+- 要详细描述世界观和角色设定
	130	+- 包含语言风格指导（用于后续 AI 续写对话）
	131	+- 至少 200 字
	132	+
	133	+#### tags
	134	+
	135	+- 1-3 个标签，逗号分隔
	136	+- 如：`搞笑,职场,反转`
	137	+
	138	+### 第 3 步：写入临时 JSON 文件
	139	+
	140	+将生成的 JSON 写入临时文件 `/tmp/drama_<timestamp>.json`
	141	+
	142	+### 第 4 步：生成图片并上传 OSS
	143	+
	144	+找到本 skill 的目录路径（即本文件 SKILL.md 所在目录），执行图片生成脚本：
	145	+
	146	+```bash
	147	+node <skill目录>/generate-images.js /tmp/drama_<timestamp>.json
	148	+```
	149	+
	150	+该脚本会：
	151	+1. 为每个角色生成立绘图（Jimeng → fallback Nano Banana Pro）
	152	+2. 自动去背景（rembg）
	153	+3. 为剧本生成背景图和封面图
	154	+4. 全部上传到阿里云 OSS
	155	+5. 将 URL 写回 JSON 文件（更新 characters[].bodyUrl, coverUrl, backgroundUrl）
	156	+
	157	+注意：该脚本需要 `ali-oss`, `sharp` npm 包以及 `rembg` Python 工具。如果 rembg 未安装，会跳过去背景步骤。
	158	+
	159	+### 第 5 步：写入数据库
	160	+
	161	+执行入库脚本：
	162	+
	163	+```bash
	164	+node <skill目录>/insert-drama.js /tmp/drama_<timestamp>.json
	165	+```
	166	+
	167	+脚本会在事务中创建：
	168	+- `dramas` 表记录（剧本元数据）
	169	+- `drama_characters` 表记录（角色配置 + 身体图 URL + 音色）
	170	+- `drama_emotions` 表记录（默认表情映射）
	171	+- `drama_scripts` 表记录（台词序列）
	172	+
	173	+### 第 6 步：清理并输出结果
	174	+
	175	+1. 删除临时 JSON 文件
	176	+2. 告知用户：
	177	+ - 剧本名称和 dramaKey
	178	+ - 角色列表及音色
	179	+ - 台词数量
	180	+ - 图片生成结果（成功/失败）
	181	+ - 简要剧情概述
	182	+
	183	+## 配置说明
	184	+
	185	+脚本会自动从以下位置加载配置（按优先级）：
	186	+1. `--env <path>` 参数指定的 .env 文件
	187	+2. 环境变量 `DRAMA_ENV_PATH` 指定的 .env 文件
	188	+3. skill 目录下的 `.env` 文件
	189	+4. 已有的环境变量
	190	+
	191	+所需配置项：
	192	+
	193	+```env
	194	+# 数据库
	195	+DB_HOST=127.0.0.1
	196	+DB_PORT=3306
	197	+DB_USER=root
	198	+DB_PASSWORD=xxx
	199	+DB_NAME=shadiao_drama
	200	+
	201	+# 阿里云 OSS（图片上传）
	202	+OSS_REGION=oss-cn-beijing
	203	+OSS_ACCESS_KEY_ID=xxx
	204	+OSS_ACCESS_KEY_SECRET=xxx
	205	+OSS_BUCKET=xxx
	206	+OSS_ENDPOINT=https://oss-cn-beijing.aliyuncs.com
	207	+
	208	+# 万界生图 API
	209	+WANJIE_API_KEY=xxx
	210	+```
	211	+
	212	+## 注意事项
	213	+
	214	+- 默认使用英文：所有内容（name、title、subtitle、description、systemPrompt、台词、tags、imagePrompt、backgroundPrompt、coverPrompt）默认用英文生成，除非用户明确要求中文
	215	+- 英文内容必须使用英文多情感音色（Candice, Serena, Glen, Sylus, Corey, Nadia）
	216	+- 风格偏沙雕搞笑，但要保留素材的核心信息
	217	+- 如果素材是严肃新闻，可以用吐槽/解说风格改编
	218	+- 角色名字要有趣，符合素材主题
	219	+- 如果用户明确要求中文，则使用中文音色
	220	+
	221	+## 复用到其他项目
	222	+
	223	+整个 `create-drama/` 目录是自包含的，复制到其他项目的 `.claude/skills/` 即可使用。
	224	+
	225	+前置条件：
	226	+1. Node.js 依赖：`npm i -g sequelize mysql2 ali-oss sharp`
	227	+2. Python 工具：`pip install rembg`（可选，用于角色图去背景）
	228	+3. 在 `create-drama/` 目录下修改 `.env` 配置
	229	+
	230	+目录结构：
	231	+```
	232	+.claude/skills/create-drama/
	233	+├── SKILL.md # Skill 定义（本文件）
	234	+├── reference.md # 完整音色参考文档
	235	+├── insert-drama.js # 数据库入库脚本
	236	+├── generate-images.js # 图片生成 + 去背景 + OSS 上传脚本
	237	+└── .env # 数据库 + OSS + 生图 API 配置
	238	+```
	239	+
	240	+## Additional Resources
	241	+
	242	+- 完整音色列表及情感参数：[reference.md](reference.md)
...	...

docs/create-drama/generate-images.js 0 → 100644

View file @17e6455

	1	+#!/usr/bin/env node
	2	+/**
	3	+ * generate-images.js — 为剧本生成角色图、背景图、封面图，并上传到 OSS
	4	+ *
	5	+ * 用法:
	6	+ * node generate-images.js <drama-json文件>
	7	+ *
	8	+ * 输入 JSON 中需要包含:
	9	+ * - characters[].imagePrompt (角色生图 prompt，由 SKILL.md 指导 Claude 生成)
	10	+ * - backgroundPrompt (背景图 prompt)
	11	+ * - coverPrompt (封面图 prompt)
	12	+ * - dramaKey (用于 OSS 路径)
	13	+ *
	14	+ * 输出: 在原 JSON 基础上补充 URL 字段，写回文件
	15	+ * - characters[].bodyUrl
	16	+ * - coverUrl
	17	+ * - backgroundUrl
	18	+ *
	19	+ * 依赖: ali-oss, sharp (可选), rembg (Python, 可选)
	20	+ */
	21	+
	22	+const path = require('path');
	23	+const fs = require('fs');
	24	+const https = require('https');
	25	+const http = require('http');
	26	+const { execSync } = require('child_process');
	27	+const os = require('os');
	28	+
	29	+// ---------- 加载 .env ----------
	30	+function loadEnv() {
	31	+ const envArgIdx = process.argv.indexOf('--env');
	32	+ const candidates = [
	33	+ envArgIdx !== -1 ? path.resolve(process.argv[envArgIdx + 1]) : null,
	34	+ process.env.DRAMA_ENV_PATH ? path.resolve(process.env.DRAMA_ENV_PATH) : null,
	35	+ path.resolve(__dirname, '.env'),
	36	+ ].filter(Boolean);
	37	+
	38	+ for (const envPath of candidates) {
	39	+ if (fs.existsSync(envPath)) {
	40	+ const envContent = fs.readFileSync(envPath, 'utf-8');
	41	+ for (const line of envContent.split('\n')) {
	42	+ const trimmed = line.trim();
	43	+ if (!trimmed \|\| trimmed.startsWith('#')) continue;
	44	+ const eqIdx = trimmed.indexOf('=');
	45	+ if (eqIdx === -1) continue;
	46	+ const key = trimmed.slice(0, eqIdx).trim();
	47	+ const val = trimmed.slice(eqIdx + 1).trim();
	48	+ if (!process.env[key]) process.env[key] = val;
	49	+ }
	50	+ console.log(`已加载配置: ${envPath}`);
	51	+ return;
	52	+ }
	53	+ }
	54	+ console.log('未找到 .env 文件，使用环境变量');
	55	+}
	56	+loadEnv();
	57	+
	58	+// ========== Wanjie 图片生成 API ==========
	59	+
	60	+const WANJIE_BASE = 'maas-openapi.wanjiedata.com';
	61	+
	62	+function wanjieRequest(action, body) {
	63	+ const apiKey = process.env.WANJIE_API_KEY;
	64	+ if (!apiKey) throw new Error('缺少 WANJIE_API_KEY 环境变量');
	65	+
	66	+ const jsonBody = JSON.stringify(body);
	67	+ return new Promise((resolve, reject) => {
	68	+ const req = https.request({
	69	+ hostname: WANJIE_BASE,
	70	+ path: `/api/jimeng/v1?Action=${action}&Version=2022-08-31`,
	71	+ method: 'POST',
	72	+ headers: {
	73	+ 'Authorization': apiKey,
	74	+ 'Content-Type': 'application/json',
	75	+ 'Content-Length': Buffer.byteLength(jsonBody),
	76	+ },
	77	+ }, (res) => {
	78	+ let data = '';
	79	+ res.on('data', chunk => data += chunk);
	80	+ res.on('end', () => {
	81	+ try { resolve(JSON.parse(data)); }
	82	+ catch (err) { reject(new Error(`解析响应失败: ${data.slice(0, 300)}`)); }
	83	+ });
	84	+ });
	85	+ req.on('error', reject);
	86	+ req.write(jsonBody);
	87	+ req.end();
	88	+ });
	89	+}
	90	+
	91	+/** Jimeng 文生图（异步模式） */
	92	+async function generateImageJimeng(prompt, width = 1024, height = 1024) {
	93	+ const res = await wanjieRequest('CVSync2AsyncSubmitTask', {
	94	+ req_key: 'jimeng_t2i_v40',
	95	+ prompt, width, height,
	96	+ });
	97	+
	98	+ if (res.code !== 10000 \|\| !res.data?.task_id) {
	99	+ throw new Error(`Jimeng 提交失败: ${JSON.stringify(res).slice(0, 300)}`);
	100	+ }
	101	+
	102	+ const taskId = res.data.task_id;
	103	+ console.log(` [Jimeng] 任务提交: task_id=${taskId}`);
	104	+
	105	+ // 轮询等待
	106	+ const start = Date.now();
	107	+ while (Date.now() - start < 120000) {
	108	+ const poll = await wanjieRequest('CVSync2AsyncGetResult', {
	109	+ req_key: 'jimeng_t2i_v40', task_id: taskId,
	110	+ });
	111	+
	112	+ if (poll.code === 10000 && poll.data?.image_urls?.length > 0) {
	113	+ const buffer = await downloadUrl(poll.data.image_urls[0]);
	114	+ return { buffer, mimeType: 'image/png' };
	115	+ }
	116	+ if (poll.code === 10000 && poll.data?.binary_data_base64?.length > 0) {
	117	+ return { buffer: Buffer.from(poll.data.binary_data_base64[0], 'base64'), mimeType: 'image/png' };
	118	+ }
	119	+ if (poll.code && poll.code !== 10000 && !poll.message?.includes('processing') && !poll.message?.includes('pending')) {
	120	+ throw new Error(`Jimeng 任务失败: ${JSON.stringify(poll).slice(0, 300)}`);
	121	+ }
	122	+
	123	+ await sleep(3000);
	124	+ }
	125	+ throw new Error(`Jimeng 超时: task_id=${taskId}`);
	126	+}
	127	+
	128	+/** Nano Banana Pro 文生图（同步模式，fallback） */
	129	+async function generateImageNano(prompt) {
	130	+ const apiKey = process.env.WANJIE_API_KEY;
	131	+ if (!apiKey) throw new Error('缺少 WANJIE_API_KEY');
	132	+
	133	+ const url = `https://${WANJIE_BASE}/api/v1beta/models/gemini-3-pro-image-preview:generateContent`;
	134	+ const body = JSON.stringify({ contents: [{ parts: [{ text: prompt }] }] });
	135	+
	136	+ for (let attempt = 1; attempt <= 3; attempt++) {
	137	+ try {
	138	+ const data = await httpsPost(url, body, { 'Authorization': `Bearer ${apiKey}` });
	139	+ const response = JSON.parse(data);
	140	+ const img = extractImageFromResponse(response);
	141	+ if (img) return img;
	142	+ throw new Error('无图片数据');
	143	+ } catch (err) {
	144	+ if (err.statusCode === 429 && attempt < 3) {
	145	+ console.warn(` [Nano] 429 限流，${3 * attempt}s 后重试...`);
	146	+ await sleep(3000 * attempt);
	147	+ continue;
	148	+ }
	149	+ throw err;
	150	+ }
	151	+ }
	152	+}
	153	+
	154	+function extractImageFromResponse(response) {
	155	+ const parts = response?.candidates?.[0]?.content?.parts;
	156	+ if (!parts) return null;
	157	+ for (const part of parts) {
	158	+ if (part.inlineData) {
	159	+ return { buffer: Buffer.from(part.inlineData.data, 'base64'), mimeType: part.inlineData.mimeType \|\| 'image/png' };
	160	+ }
	161	+ if (part.text && part.text.includes('data:image')) {
	162	+ const match = part.text.match(/data:image\/(png\|jpeg);base64,([A-Za-z0-9+/=]+)/);
	163	+ if (match) return { buffer: Buffer.from(match[2], 'base64'), mimeType: `image/${match[1]}` };
	164	+ }
	165	+ }
	166	+ return null;
	167	+}
	168	+
	169	+/** 生图：优先 Jimeng，fallback Nano */
	170	+async function generateImage(prompt) {
	171	+ try {
	172	+ return await generateImageJimeng(prompt);
	173	+ } catch (jimengErr) {
	174	+ console.warn(` [Jimeng] 失败: ${jimengErr.message.slice(0, 80)}，fallback Nano...`);
	175	+ return await generateImageNano(prompt);
	176	+ }
	177	+}
	178	+
	179	+// ========== 图片处理 ==========
	180	+
	181	+/** 去背景（rembg） */
	182	+function removeBg(inputBuffer) {
	183	+ const tmpDir = os.tmpdir();
	184	+ const ts = Date.now();
	185	+ const inputPath = path.join(tmpDir, `rembg_in_${ts}.png`);
	186	+ const outputPath = path.join(tmpDir, `rembg_out_${ts}.png`);
	187	+
	188	+ try {
	189	+ fs.writeFileSync(inputPath, inputBuffer);
	190	+ execSync(`rembg i "${inputPath}" "${outputPath}"`, { timeout: 60000 });
	191	+ const result = fs.readFileSync(outputPath);
	192	+ console.log(' [rembg] 去背景成功');
	193	+
	194	+ // 尝试 sharp 裁剪
	195	+ try {
	196	+ const sharp = require('sharp');
	197	+ // sharp 操作是 async，但我们在 sync context 用 execSync 调 sharp
	198	+ // 改用 async
	199	+ return trimImage(result);
	200	+ } catch {
	201	+ return result;
	202	+ }
	203	+ } catch (err) {
	204	+ console.warn(` [rembg] 失败(${err.message.slice(0, 60)})，返回原图`);
	205	+ return inputBuffer;
	206	+ } finally {
	207	+ try { fs.unlinkSync(inputPath); } catch { }
	208	+ try { fs.unlinkSync(outputPath); } catch { }
	209	+ }
	210	+}
	211	+
	212	+async function trimImage(buffer) {
	213	+ try {
	214	+ const sharp = require('sharp');
	215	+ const trimmed = await sharp(buffer).trim().png().toBuffer();
	216	+ const meta = await sharp(trimmed).metadata();
	217	+ const padX = Math.round(meta.width * 0.05);
	218	+ const padY = Math.round(meta.height * 0.05);
	219	+ return await sharp(trimmed)
	220	+ .extend({ top: padY, bottom: padY, left: padX, right: padX, background: { r: 0, g: 0, b: 0, alpha: 0 } })
	221	+ .png().toBuffer();
	222	+ } catch {
	223	+ return buffer;
	224	+ }
	225	+}
	226	+
	227	+// ========== OSS 上传 ==========
	228	+
	229	+function getOSSClient() {
	230	+ const OSS = require('ali-oss');
	231	+ return new OSS({
	232	+ region: process.env.OSS_REGION,
	233	+ accessKeyId: process.env.OSS_ACCESS_KEY_ID,
	234	+ accessKeySecret: process.env.OSS_ACCESS_KEY_SECRET,
	235	+ bucket: process.env.OSS_BUCKET,
	236	+ });
	237	+}
	238	+
	239	+function getPublicUrl(key) {
	240	+ const endpoint = process.env.OSS_ENDPOINT \|\| `https://${process.env.OSS_REGION}.aliyuncs.com`;
	241	+ const bucket = process.env.OSS_BUCKET;
	242	+ return `https://${bucket}.${endpoint.replace('https://', '')}/${key}`;
	243	+}
	244	+
	245	+async function uploadToOSS(ossKey, buffer, contentType = 'image/png') {
	246	+ const client = getOSSClient();
	247	+ const result = await client.put(ossKey, buffer, { headers: { 'Content-Type': contentType } });
	248	+ return getPublicUrl(result.name);
	249	+}
	250	+
	251	+// ========== 工具函数 ==========
	252	+
	253	+function sleep(ms) { return new Promise(r => setTimeout(r, ms)); }
	254	+
	255	+function httpsPost(url, body, extraHeaders = {}) {
	256	+ return new Promise((resolve, reject) => {
	257	+ const urlObj = new URL(url);
	258	+ const req = https.request({
	259	+ hostname: urlObj.hostname,
	260	+ path: urlObj.pathname + urlObj.search,
	261	+ method: 'POST',
	262	+ headers: { 'Content-Type': 'application/json', 'Content-Length': Buffer.byteLength(body), ...extraHeaders },
	263	+ }, (res) => {
	264	+ let data = '';
	265	+ res.on('data', chunk => data += chunk);
	266	+ res.on('end', () => {
	267	+ if (res.statusCode === 429) { const e = new Error('429'); e.statusCode = 429; reject(e); return; }
	268	+ resolve(data);
	269	+ });
	270	+ });
	271	+ req.on('error', reject);
	272	+ req.write(body);
	273	+ req.end();
	274	+ });
	275	+}
	276	+
	277	+function downloadUrl(url) {
	278	+ return new Promise((resolve, reject) => {
	279	+ const protocol = url.startsWith('https') ? https : http;
	280	+ protocol.get(url, (res) => {
	281	+ if (res.statusCode >= 300 && res.statusCode < 400 && res.headers.location) {
	282	+ return downloadUrl(res.headers.location).then(resolve).catch(reject);
	283	+ }
	284	+ const chunks = [];
	285	+ res.on('data', chunk => chunks.push(chunk));
	286	+ res.on('end', () => resolve(Buffer.concat(chunks)));
	287	+ res.on('error', reject);
	288	+ }).on('error', reject);
	289	+ });
	290	+}
	291	+
	292	+// ========== 主逻辑 ==========
	293	+
	294	+async function main() {
	295	+ const filePath = process.argv.find(a => a.endsWith('.json'));
	296	+ if (!filePath) {
	297	+ console.error('用法: node generate-images.js <drama-json文件>');
	298	+ process.exit(1);
	299	+ }
	300	+
	301	+ const absPath = path.resolve(filePath);
	302	+ const data = JSON.parse(fs.readFileSync(absPath, 'utf-8'));
	303	+ const dramaKey = data.dramaKey \|\| `drama_${Date.now().toString(36)}`;
	304	+ data.dramaKey = dramaKey;
	305	+
	306	+ // 检查 OSS 配置
	307	+ const ossOk = process.env.OSS_REGION && process.env.OSS_ACCESS_KEY_ID && process.env.OSS_ACCESS_KEY_SECRET && process.env.OSS_BUCKET;
	308	+ if (!ossOk) {
	309	+ console.error('OSS 配置不完整，请检查 .env 中的 OSS_REGION, OSS_ACCESS_KEY_ID, OSS_ACCESS_KEY_SECRET, OSS_BUCKET');
	310	+ process.exit(1);
	311	+ }
	312	+
	313	+ // 检查生图 API 配置
	314	+ if (!process.env.WANJIE_API_KEY) {
	315	+ console.error('缺少 WANJIE_API_KEY');
	316	+ process.exit(1);
	317	+ }
	318	+
	319	+ const characters = data.characters \|\| [];
	320	+ const total = characters.length + 2; // chars + background + cover
	321	+ let completed = 0;
	322	+ const tick = (label) => { completed++; console.log(`[${completed}/${total}] ${label} 完成`); };
	323	+
	324	+ // 并行处理所有图片
	325	+ const tasks = [];
	326	+
	327	+ // 角色图
	328	+ for (const char of characters) {
	329	+ tasks.push((async () => {
	330	+ const prompt = char.imagePrompt \|\| `全身立绘，纯色背景，沙雕动漫风格。角色：${char.name}。脸部空白无五官。`;
	331	+ console.log(`\n生成角色图: ${char.name}`);
	332	+ console.log(` prompt: ${prompt.slice(0, 80)}...`);
	333	+
	334	+ const { buffer } = await generateImage(prompt);
	335	+
	336	+ // 去背景
	337	+ console.log(` 正在去背景...`);
	338	+ const processed = await removeBg(buffer);
	339	+
	340	+ // 上传 OSS
	341	+ const safeName = Buffer.from(char.name).toString('hex').slice(0, 16);
	342	+ const ossKey = `dramas/${dramaKey}/characters/${safeName}.png`;
	343	+ const url = await uploadToOSS(ossKey, processed);
	344	+ char.bodyUrl = url;
	345	+ console.log(` 上传完成: ${url}`);
	346	+ tick(char.name);
	347	+ })());
	348	+ }
	349	+
	350	+ // 背景图
	351	+ tasks.push((async () => {
	352	+ const prompt = data.backgroundPrompt \|\| `横版场景背景图，宽屏比例，场景：${data.description \|\| data.name}，卡通风格，无人物。`;
	353	+ console.log(`\n生成背景图`);
	354	+ console.log(` prompt: ${prompt.slice(0, 80)}...`);
	355	+
	356	+ const { buffer } = await generateImage(prompt);
	357	+ const ossKey = `dramas/${dramaKey}/background.png`;
	358	+ const url = await uploadToOSS(ossKey, buffer);
	359	+ data.backgroundUrl = url;
	360	+ console.log(` 上传完成: ${url}`);
	361	+ tick('背景图');
	362	+ })());
	363	+
	364	+ // 封面图
	365	+ tasks.push((async () => {
	366	+ const prompt = data.coverPrompt \|\| `剧目封面海报，标题：${data.title \|\| data.name}，色彩鲜明。`;
	367	+ console.log(`\n生成封面图`);
	368	+ console.log(` prompt: ${prompt.slice(0, 80)}...`);
	369	+
	370	+ const { buffer } = await generateImage(prompt);
	371	+ const ossKey = `dramas/${dramaKey}/cover.png`;
	372	+ const url = await uploadToOSS(ossKey, buffer);
	373	+ data.coverUrl = url;
	374	+ console.log(` 上传完成: ${url}`);
	375	+ tick('封面图');
	376	+ })());
	377	+
	378	+ await Promise.allSettled(tasks);
	379	+
	380	+ // 写回文件
	381	+ fs.writeFileSync(absPath, JSON.stringify(data, null, 2), 'utf-8');
	382	+ console.log(`\n图片生成完成，已更新 JSON 文件: ${absPath}`);
	383	+ console.log(` 角色图: ${characters.filter(c => c.bodyUrl).length}/${characters.length}`);
	384	+ console.log(` 背景图: ${data.backgroundUrl ? '✓' : '✗'}`);
	385	+ console.log(` 封面图: ${data.coverUrl ? '✓' : '✗'}`);
	386	+}
	387	+
	388	+main().catch(err => {
	389	+ console.error('执行失败:', err);
	390	+ process.exit(1);
	391	+});
...	...

docs/create-drama/insert-drama.js 0 → 100644

View file @17e6455

	1	+#!/usr/bin/env node
	2	+/**
	3	+ * insert-drama.js — 独立脚本：将 classic 类型剧本数据插入数据库
	4	+ *
	5	+ * 用法:
	6	+ * node insert-drama.js <json文件路径>
	7	+ * echo '{"name":...}' \| node insert-drama.js --stdin
	8	+ *
	9	+ * JSON 格式:
	10	+ * {
	11	+ * "name": "剧本名称",
	12	+ * "title": "显示标题（可带 emoji）",
	13	+ * "subtitle": "副标题",
	14	+ * "description": "剧本简介",
	15	+ * "systemPrompt": "用于 AI 续写的系统提示词",
	16	+ * "isPublic": true,
	17	+ * "userId": 1,
	18	+ * "characters": [
	19	+ * { "name": "角色名", "gender": "male", "voiceId": "zh_male_lengkugege_emo_v2_mars_bigtts" }
	20	+ * ],
	21	+ * "script": [
	22	+ * { "character": "角色名", "dialogue": "台词", "avatar": "happy.png", "action": "none", "position": "left" }
	23	+ * ]
	24	+ * }
	25	+ */
	26	+
	27	+const path = require('path');
	28	+const fs = require('fs');
	29	+
	30	+// 加载 .env 文件（优先级：--env 参数 > 环境变量 DRAMA_ENV_PATH > 脚本同目录 .env）
	31	+function loadEnv() {
	32	+ const envArgIdx = process.argv.indexOf('--env');
	33	+ const candidates = [
	34	+ envArgIdx !== -1 ? path.resolve(process.argv[envArgIdx + 1]) : null,
	35	+ process.env.DRAMA_ENV_PATH ? path.resolve(process.env.DRAMA_ENV_PATH) : null,
	36	+ path.resolve(__dirname, '.env'),
	37	+ ].filter(Boolean);
	38	+
	39	+ for (const envPath of candidates) {
	40	+ if (fs.existsSync(envPath)) {
	41	+ const envContent = fs.readFileSync(envPath, 'utf-8');
	42	+ for (const line of envContent.split('\n')) {
	43	+ const trimmed = line.trim();
	44	+ if (!trimmed \|\| trimmed.startsWith('#')) continue;
	45	+ const eqIdx = trimmed.indexOf('=');
	46	+ if (eqIdx === -1) continue;
	47	+ const key = trimmed.slice(0, eqIdx).trim();
	48	+ const val = trimmed.slice(eqIdx + 1).trim();
	49	+ if (!process.env[key]) process.env[key] = val;
	50	+ }
	51	+ console.log(`已加载配置: ${envPath}`);
	52	+ return;
	53	+ }
	54	+ }
	55	+ console.log('未找到 .env 文件，使用环境变量或默认值');
	56	+}
	57	+loadEnv();
	58	+
	59	+const { Sequelize, DataTypes } = require('sequelize');
	60	+
	61	+// ---------- DB 连接 ----------
	62	+const sequelize = new Sequelize(
	63	+ process.env.DB_NAME \|\| 'shadiao_drama',
	64	+ process.env.DB_USER \|\| 'root',
	65	+ process.env.DB_PASSWORD \|\| '',
	66	+ {
	67	+ host: process.env.DB_HOST \|\| '127.0.0.1',
	68	+ port: parseInt(process.env.DB_PORT \|\| '3306', 10),
	69	+ dialect: 'mysql',
	70	+ logging: false,
	71	+ timezone: '+08:00',
	72	+ define: { charset: 'utf8mb4', collate: 'utf8mb4_unicode_ci' },
	73	+ }
	74	+);
	75	+
	76	+// ---------- Model 定义（内联，避免依赖项目代码） ----------
	77	+const Drama = sequelize.define('Drama', {
	78	+ id: { type: DataTypes.INTEGER, autoIncrement: true, primaryKey: true },
	79	+ dramaKey: { type: DataTypes.STRING(50), allowNull: false, unique: true, field: 'drama_key' },
	80	+ name: { type: DataTypes.STRING(100), allowNull: false },
	81	+ title: { type: DataTypes.STRING(100), allowNull: true },
	82	+ subtitle: { type: DataTypes.STRING(200), allowNull: true },
	83	+ description: { type: DataTypes.TEXT, allowNull: true },
	84	+ coverUrl: { type: DataTypes.STRING(500), allowNull: true, field: 'cover_url' },
	85	+ backgroundUrl: { type: DataTypes.STRING(500), allowNull: true, field: 'background_url' },
	86	+ systemPrompt: { type: DataTypes.TEXT, allowNull: true, field: 'system_prompt' },
	87	+ status: { type: DataTypes.TINYINT, defaultValue: 1 },
	88	+ userId: { type: DataTypes.INTEGER, allowNull: true, field: 'user_id' },
	89	+ isPublic: { type: DataTypes.BOOLEAN, defaultValue: true, field: 'is_public' },
	90	+ type: { type: DataTypes.STRING(30), defaultValue: 'classic' },
	91	+ emojiPack: { type: DataTypes.STRING(50), defaultValue: 'default', field: 'emoji_pack' },
	92	+ vrmConfig: { type: DataTypes.TEXT, allowNull: true, field: 'vrm_config' },
	93	+ tags: { type: DataTypes.STRING(255), allowNull: true },
	94	+ score: { type: DataTypes.DECIMAL(10, 2), defaultValue: 0 },
	95	+}, { tableName: 'dramas', timestamps: true, underscored: true });
	96	+
	97	+const DramaCharacter = sequelize.define('DramaCharacter', {
	98	+ id: { type: DataTypes.INTEGER, autoIncrement: true, primaryKey: true },
	99	+ dramaId: { type: DataTypes.INTEGER, allowNull: false, field: 'drama_id' },
	100	+ characterName: { type: DataTypes.STRING(50), allowNull: false, field: 'character_name' },
	101	+ bodyUrl: { type: DataTypes.STRING(500), allowNull: true, field: 'body_url' },
	102	+ faceOffsetTop: { type: DataTypes.STRING(20), allowNull: true, field: 'face_offset_top' },
	103	+ faceOffsetLeft: { type: DataTypes.STRING(20), allowNull: true, field: 'face_offset_left' },
	104	+ faceOffsetWidth: { type: DataTypes.STRING(20), allowNull: true, field: 'face_offset_width' },
	105	+ voiceId: { type: DataTypes.STRING(100), allowNull: true, field: 'voice_id' },
	106	+ isDefault: { type: DataTypes.TINYINT, defaultValue: 0, field: 'is_default' },
	107	+}, { tableName: 'drama_characters', timestamps: false, underscored: true });
	108	+
	109	+const DramaEmotion = sequelize.define('DramaEmotion', {
	110	+ id: { type: DataTypes.INTEGER, autoIncrement: true, primaryKey: true },
	111	+ dramaId: { type: DataTypes.INTEGER, allowNull: false, field: 'drama_id' },
	112	+ avatarKey: { type: DataTypes.STRING(50), allowNull: false, field: 'avatar_key' },
	113	+ faceUrl: { type: DataTypes.STRING(500), allowNull: true, field: 'face_url' },
	114	+ emotionName: { type: DataTypes.STRING(30), allowNull: true, field: 'emotion_name' },
	115	+ emotionScale: { type: DataTypes.INTEGER, defaultValue: 3, field: 'emotion_scale' },
	116	+ isDefault: { type: DataTypes.TINYINT, defaultValue: 0, field: 'is_default' },
	117	+ characterName: { type: DataTypes.STRING(50), allowNull: true, field: 'character_name' },
	118	+}, { tableName: 'drama_emotions', timestamps: false, underscored: true });
	119	+
	120	+const DramaScript = sequelize.define('DramaScript', {
	121	+ id: { type: DataTypes.INTEGER, autoIncrement: true, primaryKey: true },
	122	+ dramaId: { type: DataTypes.INTEGER, allowNull: false, field: 'drama_id' },
	123	+ seq: { type: DataTypes.INTEGER, allowNull: false },
	124	+ characterName: { type: DataTypes.STRING(50), allowNull: false, field: 'character_name' },
	125	+ avatar: { type: DataTypes.STRING(50), allowNull: true },
	126	+ dialogue: { type: DataTypes.TEXT, allowNull: false },
	127	+ action: { type: DataTypes.STRING(30), allowNull: true },
	128	+ position: { type: DataTypes.STRING(10), allowNull: true },
	129	+ sfx: { type: DataTypes.STRING(100), allowNull: true },
	130	+ audio: { type: DataTypes.STRING(500), allowNull: true },
	131	+}, { tableName: 'drama_scripts', timestamps: false, underscored: true });
	132	+
	133	+// ---------- 默认表情配置 ----------
	134	+const DEFAULT_EMOJI_BASE = 'https://shadiao-drama.oss-cn-beijing.aliyuncs.com/dramas/zhenhuan_chuan/emojis';
	135	+const DEFAULT_EMOJIS = [
	136	+ { key: 'normal.png', name: '正常', emotion: 'neutral', scale: 3 },
	137	+ { key: 'happy.png', name: '开心', emotion: 'happy', scale: 4 },
	138	+ { key: 'very_happy.png', name: '非常开心', emotion: 'happy', scale: 5 },
	139	+ { key: 'angry.png', name: '生气', emotion: 'angry', scale: 4 },
	140	+ { key: 'sad.png', name: '伤心', emotion: 'sad', scale: 4 },
	141	+ { key: 'cry.png', name: '哭', emotion: 'sad', scale: 5 },
	142	+ { key: 'shock.png', name: '惊讶', emotion: 'surprised', scale: 5 },
	143	+ { key: 'smirk.png', name: '撇嘴', emotion: 'happy', scale: 3 },
	144	+ { key: 'speechless.png', name: '无语', emotion: 'neutral', scale: 2 },
	145	+];
	146	+
	147	+// ---------- 音色情绪兼容映射 ----------
	148	+const VOICE_EMOTIONS = {
	149	+ 'zh_male_lengkugege_emo_v2_mars_bigtts': ['angry','coldness','fear','happy','hate','neutral','sad','depressed'],
	150	+ 'zh_female_tianxinxiaomei_emo_v2_mars_bigtts': ['sad','fear','hate','neutral'],
	151	+ 'zh_female_gaolengyujie_emo_v2_mars_bigtts': ['happy','sad','angry','surprised','fear','hate','excited','coldness','neutral'],
	152	+ 'zh_male_aojiaobazong_emo_v2_mars_bigtts': ['neutral','happy','angry','hate'],
	153	+ 'zh_male_guangzhoudege_emo_mars_bigtts': ['angry','fear','neutral'],
	154	+ 'zh_male_jingqiangkanye_emo_mars_bigtts': ['happy','angry','surprised','hate','neutral'],
	155	+ 'zh_female_linjuayi_emo_v2_mars_bigtts': ['neutral','angry','coldness','depressed','surprised'],
	156	+ 'zh_male_yourougongzi_emo_v2_mars_bigtts': ['happy','angry','fear','hate','excited','neutral','depressed'],
	157	+ 'zh_male_ruyayichen_emo_v2_mars_bigtts': ['happy','sad','angry','fear','excited','coldness','neutral'],
	158	+ 'zh_male_junlangnanyou_emo_v2_mars_bigtts': ['happy','sad','angry','surprised','fear','neutral'],
	159	+ 'zh_male_beijingxiaoye_emo_v2_mars_bigtts': ['angry','surprised','fear','excited','coldness','neutral'],
	160	+ 'zh_female_roumeinvyou_emo_v2_mars_bigtts': ['happy','sad','angry','surprised','fear','hate','excited','coldness','neutral'],
	161	+ 'zh_male_yangguangqingnian_emo_v2_mars_bigtts': ['happy','sad','angry','fear','excited','coldness','neutral'],
	162	+ 'zh_female_meilinvyou_emo_v2_mars_bigtts': ['sad','fear','neutral'],
	163	+ 'zh_female_shuangkuaisisi_emo_v2_mars_bigtts': ['happy','sad','angry','surprised','excited','coldness','neutral'],
	164	+ 'zh_male_shenyeboke_emo_v2_mars_bigtts': ['surprised','sad','neutral','hate','happy','fear','excited','depressed','coldness','angry'],
	165	+ 'zh_male_zhoujielun_emo_v2_mars_bigtts': ['neutral'],
	166	+};
	167	+
	168	+function validateEmotion(speaker, emotion) {
	169	+ if (!emotion \|\| emotion === 'neutral') return emotion;
	170	+ const supported = VOICE_EMOTIONS[speaker];
	171	+ if (!supported) return null;
	172	+ return supported.includes(emotion) ? emotion : 'neutral';
	173	+}
	174	+
	175	+// ---------- 构建默认表情记录 ----------
	176	+function buildEmotionRecords(dramaId, characters, defaultVoice) {
	177	+ const records = [];
	178	+ const charVoiceMap = {};
	179	+ if (characters) {
	180	+ for (const char of characters) {
	181	+ if (char.voiceId) charVoiceMap[char.name] = char.voiceId;
	182	+ }
	183	+ }
	184	+
	185	+ // 共享默认表情
	186	+ for (const emo of DEFAULT_EMOJIS) {
	187	+ const validEmotion = validateEmotion(defaultVoice, emo.emotion) \|\| 'neutral';
	188	+ records.push({
	189	+ dramaId,
	190	+ avatarKey: emo.key,
	191	+ faceUrl: `${DEFAULT_EMOJI_BASE}/${encodeURIComponent(emo.name)}.png`,
	192	+ emotionName: validEmotion,
	193	+ emotionScale: emo.scale,
	194	+ isDefault: 0,
	195	+ characterName: null,
	196	+ });
	197	+ }
	198	+ records.push({
	199	+ dramaId,
	200	+ avatarKey: '_default',
	201	+ faceUrl: `${DEFAULT_EMOJI_BASE}/${encodeURIComponent('正常')}.png`,
	202	+ emotionName: 'neutral',
	203	+ emotionScale: 3,
	204	+ isDefault: 1,
	205	+ characterName: null,
	206	+ });
	207	+
	208	+ return records;
	209	+}
	210	+
	211	+// ---------- 主逻辑 ----------
	212	+async function main() {
	213	+ let jsonStr;
	214	+
	215	+ if (process.argv.includes('--stdin')) {
	216	+ // 从 stdin 读取
	217	+ jsonStr = '';
	218	+ for await (const chunk of process.stdin) {
	219	+ jsonStr += chunk;
	220	+ }
	221	+ } else if (process.argv[2]) {
	222	+ // 从文件读取
	223	+ const filePath = path.resolve(process.argv[2]);
	224	+ if (!fs.existsSync(filePath)) {
	225	+ console.error(`文件不存在: ${filePath}`);
	226	+ process.exit(1);
	227	+ }
	228	+ jsonStr = fs.readFileSync(filePath, 'utf-8');
	229	+ } else {
	230	+ console.error('用法: node insert-drama.js <json文件> 或 node insert-drama.js --stdin');
	231	+ process.exit(1);
	232	+ }
	233	+
	234	+ let data;
	235	+ try {
	236	+ data = JSON.parse(jsonStr);
	237	+ } catch (e) {
	238	+ console.error('JSON 解析失败:', e.message);
	239	+ process.exit(1);
	240	+ }
	241	+
	242	+ const { name, title, subtitle, description, systemPrompt, isPublic, userId, characters, script, tags } = data;
	243	+
	244	+ if (!name \|\| !script \|\| !Array.isArray(script) \|\| script.length === 0) {
	245	+ console.error('缺少必要字段: name, script');
	246	+ process.exit(1);
	247	+ }
	248	+ if (!characters \|\| !Array.isArray(characters) \|\| characters.length === 0) {
	249	+ console.error('缺少必要字段: characters');
	250	+ process.exit(1);
	251	+ }
	252	+
	253	+ // 连接数据库
	254	+ await sequelize.authenticate();
	255	+ console.log('数据库连接成功');
	256	+
	257	+ const t = await sequelize.transaction();
	258	+
	259	+ try {
	260	+ const dramaKey = `user_${userId \|\| 1}_${Date.now().toString(36)}`;
	261	+
	262	+ // 1. 创建 Drama 记录
	263	+ const drama = await Drama.create({
	264	+ dramaKey,
	265	+ name,
	266	+ title: title \|\| name,
	267	+ subtitle: subtitle \|\| '',
	268	+ description: description \|\| '',
	269	+ coverUrl: data.coverUrl \|\| '',
	270	+ backgroundUrl: data.backgroundUrl \|\| '',
	271	+ systemPrompt: systemPrompt \|\| '',
	272	+ status: 1,
	273	+ userId: userId \|\| null,
	274	+ isPublic: isPublic !== false,
	275	+ type: 'classic',
	276	+ emojiPack: 'default',
	277	+ tags: tags \|\| null,
	278	+ }, { transaction: t });
	279	+
	280	+ // 2. 创建角色记录
	281	+ const charRecords = characters.map(char => ({
	282	+ dramaId: drama.id,
	283	+ characterName: char.name,
	284	+ bodyUrl: char.bodyUrl \|\| '',
	285	+ faceOffsetTop: char.faceOffset?.top \|\| '8%',
	286	+ faceOffsetLeft: char.faceOffset?.left \|\| '35%',
	287	+ faceOffsetWidth: char.faceOffset?.width \|\| '30%',
	288	+ voiceId: char.voiceId \|\| '',
	289	+ isDefault: char.isDefault ? 1 : 0,
	290	+ }));
	291	+ // 添加 _default 角色
	292	+ const firstChar = characters[0];
	293	+ charRecords.push({
	294	+ dramaId: drama.id,
	295	+ characterName: '_default',
	296	+ bodyUrl: firstChar.bodyUrl \|\| '',
	297	+ faceOffsetTop: firstChar.faceOffset?.top \|\| '8%',
	298	+ faceOffsetLeft: firstChar.faceOffset?.left \|\| '35%',
	299	+ faceOffsetWidth: firstChar.faceOffset?.width \|\| '30%',
	300	+ voiceId: firstChar.voiceId \|\| '',
	301	+ isDefault: 1,
	302	+ });
	303	+ await DramaCharacter.bulkCreate(charRecords, { transaction: t });
	304	+
	305	+ // 3. 创建表情记录
	306	+ const defaultVoice = characters[0]?.voiceId \|\| '';
	307	+ const emotionRecords = buildEmotionRecords(drama.id, characters, defaultVoice);
	308	+ await DramaEmotion.bulkCreate(emotionRecords, { transaction: t });
	309	+
	310	+ // 4. 创建台词记录
	311	+ const scriptRecords = script.map((line, i) => ({
	312	+ dramaId: drama.id,
	313	+ seq: i,
	314	+ characterName: line.character,
	315	+ avatar: line.avatar \|\| 'normal.png',
	316	+ dialogue: line.dialogue,
	317	+ action: line.action \|\| 'none',
	318	+ position: line.position \|\| 'left',
	319	+ sfx: line.sfx \|\| null,
	320	+ audio: null,
	321	+ }));
	322	+ await DramaScript.bulkCreate(scriptRecords, { transaction: t });
	323	+
	324	+ await t.commit();
	325	+
	326	+ console.log(`\n剧本创建成功！`);
	327	+ console.log(` ID: ${drama.id}`);
	328	+ console.log(` Key: ${dramaKey}`);
	329	+ console.log(` 名称: ${name}`);
	330	+ console.log(` 角色数: ${characters.length}`);
	331	+ console.log(` 台词数: ${script.length}`);
	332	+
	333	+ // 输出 JSON 结果供调用方解析
	334	+ console.log(`\n__RESULT__${JSON.stringify({ id: drama.id, dramaKey, name })}__END__`);
	335	+ } catch (err) {
	336	+ await t.rollback();
	337	+ console.error('插入失败:', err.message);
	338	+ process.exit(1);
	339	+ } finally {
	340	+ await sequelize.close();
	341	+ }
	342	+}
	343	+
	344	+main().catch(err => {
	345	+ console.error('执行失败:', err);
	346	+ process.exit(1);
	347	+});
...	...

docs/create-drama/reference.md 0 → 100644

View file @17e6455

	1	+# 音色参考文档
	2	+
	3	+## 多情感音色（推荐优先使用）
	4	+
	5	+### 中文多情感
	6	+
	7	+\| 音色名称 \| voice_type \| 支持的情感 \|
	8	+\|---------\|-----------\|-----------\|
	9	+\| 冷酷哥哥 \| `zh_male_lengkugege_emo_v2_mars_bigtts` \| angry, coldness, fear, happy, hate, neutral, sad, depressed \|
	10	+\| 甜心小美 \| `zh_female_tianxinxiaomei_emo_v2_mars_bigtts` \| sad, fear, hate, neutral \|
	11	+\| 高冷御姐 \| `zh_female_gaolengyujie_emo_v2_mars_bigtts` \| happy, sad, angry, surprised, fear, hate, excited, coldness, neutral \|
	12	+\| 傲娇霸总 \| `zh_male_aojiaobazong_emo_v2_mars_bigtts` \| neutral, happy, angry, hate \|
	13	+\| 广州德哥 \| `zh_male_guangzhoudege_emo_mars_bigtts` \| angry, fear, neutral \|
	14	+\| 京腔侃爷 \| `zh_male_jingqiangkanye_emo_mars_bigtts` \| happy, angry, surprised, hate, neutral \|
	15	+\| 邻居阿姨 \| `zh_female_linjuayi_emo_v2_mars_bigtts` \| neutral, angry, coldness, depressed, surprised \|
	16	+\| 优柔公子 \| `zh_male_yourougongzi_emo_v2_mars_bigtts` \| happy, angry, fear, hate, excited, neutral, depressed \|
	17	+\| 儒雅男友 \| `zh_male_ruyayichen_emo_v2_mars_bigtts` \| happy, sad, angry, fear, excited, coldness, neutral \|
	18	+\| 俊朗男友 \| `zh_male_junlangnanyou_emo_v2_mars_bigtts` \| happy, sad, angry, surprised, fear, neutral \|
	19	+\| 北京小爷 \| `zh_male_beijingxiaoye_emo_v2_mars_bigtts` \| angry, surprised, fear, excited, coldness, neutral \|
	20	+\| 柔美女友 \| `zh_female_roumeinvyou_emo_v2_mars_bigtts` \| happy, sad, angry, surprised, fear, hate, excited, coldness, neutral \|
	21	+\| 阳光青年 \| `zh_male_yangguangqingnian_emo_v2_mars_bigtts` \| happy, sad, angry, fear, excited, coldness, neutral \|
	22	+\| 魅力女友 \| `zh_female_meilinvyou_emo_v2_mars_bigtts` \| sad, fear, neutral \|
	23	+\| 爽快思思 \| `zh_female_shuangkuaisisi_emo_v2_mars_bigtts` \| happy, sad, angry, surprised, excited, coldness, neutral \|
	24	+\| 深夜播客 \| `zh_male_shenyeboke_emo_v2_mars_bigtts` \| surprised, sad, neutral, hate, happy, fear, excited, depressed, coldness, angry \|
	25	+
	26	+### 英文多情感
	27	+
	28	+\| 音色名称 \| voice_type \| 语种 \| 支持的情感 \|
	29	+\|---------\|-----------\|------\|-----------\|
	30	+\| Candice \| `en_female_candice_emo_v2_mars_bigtts` \| 美式英语 \| affectionate, angry, ASMR, chat, excited, happy, neutral, warm \|
	31	+\| Serena \| `en_female_skye_emo_v2_mars_bigtts` \| 美式英语 \| affectionate, angry, ASMR, chat, excited, happy, neutral, sad, warm \|
	32	+\| Glen \| `en_male_glen_emo_v2_mars_bigtts` \| 美式英语 \| affectionate, angry, ASMR, chat, excited, happy, neutral, sad, warm \|
	33	+\| Sylus \| `en_male_sylus_emo_v2_mars_bigtts` \| 美式英语 \| affectionate, angry, ASMR, authoritative, chat, excited, happy, neutral, sad, warm \|
	34	+\| Corey \| `en_male_corey_emo_v2_mars_bigtts` \| 英式英语 \| angry, ASMR, authoritative, chat, affectionate, excited, happy, neutral, sad, warm \|
	35	+\| Nadia \| `en_female_nadia_tips_emo_v2_mars_bigtts` \| 英式英语 \| affectionate, angry, ASMR, chat, excited, happy, neutral, sad, warm \|
	36	+
	37	+## 通用中文音色（无多情感，但音质好）
	38	+
	39	+\| 音色名称 \| voice_type \| 说明 \|
	40	+\|---------\|-----------\|------\|
	41	+\| 湾湾小何 \| `zh_female_wanwanxiaohe_moon_bigtts` \| 台湾口音女声 \|
	42	+\| 幽默大爷 \| `ICL_zh_male_youmodaye_tob` \| 搞笑老年男声 \|
	43	+\| 温柔女神 \| `ICL_zh_female_wenrounvshen_239eff5e8ffa_tob` \| 温柔女声 \|
	44	+\| 机灵小伙 \| `ICL_zh_male_shenmi_v1_tob` \| 活泼男声 \|
	45	+\| 元气甜妹 \| `ICL_zh_female_wuxi_tob` \| 甜美女声 \|
	46	+\| 渊博小叔 \| `zh_male_yuanboxiaoshu_moon_bigtts` \| 成熟男声 \|
	47	+\| 邻家女孩 \| `zh_female_linjianvhai_moon_bigtts` \| 清新女声 \|
	48	+\| 解说小明 \| `zh_male_jieshuoxiaoming_moon_bigtts` \| 解说风格 \|
	49	+\| 奶气萌娃 \| `zh_male_naiqimengwa_mars_bigtts` \| 儿童声 \|
	50	+\| 婆婆 \| `zh_female_popo_mars_bigtts` \| 老年女声 \|
	51	+
	52	+## 角色扮演音色（适合特定角色类型）
	53	+
	54	+\| 音色名称 \| voice_type \| 风格 \|
	55	+\|---------\|-----------\|------\|
	56	+\| 霸道总裁 \| `ICL_zh_male_badaozongcai_v1_tob` \| 霸气男声 \|
	57	+\| 邪魅御姐 \| `ICL_zh_female_xiangliangya_v1_tob` \| 魅惑女声 \|
	58	+\| 油腻大叔 \| `ICL_zh_male_you_tob` \| 油腻中年男 \|
	59	+\| 病弱公子 \| `ICL_zh_male_bingruogongzi_tob` \| 虚弱男声 \|
	60	+\| 纯真少女 \| `ICL_zh_female_chunzhenshaonv_e588402fb8ad_tob` \| 天真女声 \|
	61	+\| 孤傲公子 \| `ICL_zh_male_guaogongzi_v1_tob` \| 高冷男声 \|
	62	+\| 撒娇男友 \| `ICL_zh_male_sajiaonanyou_tob` \| 撒娇男声 \|
	63	+\| 成熟姐姐 \| `ICL_zh_female_chengshujiejie_tob` \| 成熟女声 \|
	64	+\| 傲娇女友 \| `ICL_zh_female_aojiaonvyou_tob` \| 傲娇女声 \|
	65	+\| 贴心女友 \| `ICL_zh_female_tiexinnvyou_tob` \| 温柔女声 \|
	66	+\| 热血少年 \| `ICL_zh_male_rexueshaonian_tob` \| 热血男声 \|
	67	+\| 冷峻上司 \| `ICL_zh_male_lengjunshangsi_tob` \| 严肃男声 \|
	68	+
	69	+## IP 仿音（趣味）
	70	+
	71	+\| 音色名称 \| voice_type \|
	72	+\|---------\|-----------\|
	73	+\| 唐僧 \| `zh_male_tangseng_mars_bigtts` \|
	74	+\| 猪八戒 \| `zh_male_zhubajie_mars_bigtts` \|
	75	+\| 猴哥 \| `zh_male_sunwukong_mars_bigtts` \|
	76	+\| 佩奇猪 \| `zh_female_peiqi_mars_bigtts` \|
	77	+\| 熊二 \| `zh_male_xionger_mars_bigtts` \|
	78	+\| 鲁班七号 \| `zh_male_lubanqihao_mars_bigtts` \|
	79	+
	80	+## 趣味口音
	81	+
	82	+\| 音色名称 \| voice_type \| 口音 \|
	83	+\|---------\|-----------\|------\|
	84	+\| 双节棍小哥 \| `zh_male_zhoujielun_emo_v2_mars_bigtts` \| 台湾口音 \|
	85	+\| 豫州子轩 \| `zh_male_yuzhouzixuan_moon_bigtts` \| 河南口音 \|
	86	+\| 呆萌川妹 \| `zh_female_daimengchuanmei_moon_bigtts` \| 四川口音 \|
	87	+\| 广西远舟 \| `zh_male_guangxiyuanzhou_moon_bigtts` \| 广西口音 \|
	88	+\| 妹坨洁儿 \| `zh_female_meituojieer_moon_bigtts` \| 长沙口音 \|
	89	+
	90	+## 通用英文音色
	91	+
	92	+### 美式英语
	93	+
	94	+\| 音色名称 \| voice_type \|
	95	+\|---------\|-----------\|
	96	+\| Lauren \| `en_female_lauren_moon_bigtts` \|
	97	+\| Bruce \| `en_male_bruce_moon_bigtts` \|
	98	+\| Michael \| `en_male_michael_moon_bigtts` \|
	99	+\| Nara \| `en_female_nara_moon_bigtts` \|
	100	+\| Adam \| `en_male_adam_mars_bigtts` \|
	101	+\| Amanda \| `en_female_amanda_mars_bigtts` \|
	102	+\| Jackson \| `en_male_jackson_mars_bigtts` \|
	103	+\| Lucas \| `zh_male_M100_conversation_wvae_bigtts` \|
	104	+\| Luna \| `en_female_sarah_new_conversation_wvae_bigtts` \|
	105	+
	106	+### 英式英语
	107	+
	108	+\| 音色名称 \| voice_type \|
	109	+\|---------\|-----------\|
	110	+\| Emily \| `en_female_emily_mars_bigtts` \|
	111	+\| Smith \| `en_male_smith_mars_bigtts` \|
	112	+\| Anna \| `en_female_anna_mars_bigtts` \|
	113	+\| Dave \| `en_male_dave_moon_bigtts` \|
	114	+
	115	+## 情感参数速查
	116	+
	117	+### 中文情感
	118	+```
	119	+happy, sad, angry, surprised, fear, hate, excited, coldness, neutral,
	120	+depressed, lovey-dovey, shy, comfort, tension, tender, storytelling,
	121	+radio, magnetic, advertising, vocal-fry, ASMR, news, entertainment, dialect
	122	+```
	123	+
	124	+### 英文情感
	125	+```
	126	+neutral, happy, angry, sad, excited, chat, ASMR, warm, affectionate, authoritative
	127	+```
...	...