113 lines
5.9 KiB
JavaScript
113 lines
5.9 KiB
JavaScript
/* build-create-prompts.mjs — 构造 6 个视频的 AI 创作任务 prompt（与前端 AI创作 按钮完全一致：人设卡+视频选题块+创作指令），并提交 /api/run */
|
|
const { DatabaseSync } = require('node:sqlite');
|
|
const path = require('path');
|
|
|
|
const DB = path.join(__dirname, 'mcn-plugin.db');
|
|
const API = 'http://localhost:8900/api/run';
|
|
const ACCOUNT_NAME = '俊希';
|
|
const ACCOUNT_ID = 3;
|
|
const VIDEO_IDS = [309, 313, 315, 316, 318, 320];
|
|
|
|
const db = new DatabaseSync(DB, { readOnly: true });
|
|
|
|
// —— 与前端一致的函数（data-pages.js）——
|
|
const stripTitleQuotes = (s) => String(s || '').trim().replace(/^["“”'‘’「『]+|["“”'‘’」』]+$/g, '');
|
|
const cleanTitle = (t) => {
|
|
const s = String(t || '').trim();
|
|
const idx = s.indexOf('#');
|
|
if (idx <= 0) return stripTitleQuotes(s);
|
|
return stripTitleQuotes(s.slice(0, idx)) + s.slice(idx);
|
|
};
|
|
const personaForCreate = (text) => {
|
|
const s = String(text || '').trim();
|
|
const idx = s.indexOf('## 十一、');
|
|
return idx >= 0 ? s.slice(0, idx).trim() : s;
|
|
};
|
|
const extractVideoTopic = (content) => {
|
|
const s = String(content || '');
|
|
const i = s.indexOf('# 选题');
|
|
if (i < 0) return '';
|
|
const rest = s.slice(i + '# 选题'.length);
|
|
const j = rest.search(/\n# /);
|
|
return (j >= 0 ? rest.slice(0, j) : rest).trim();
|
|
};
|
|
|
|
// —— 读俊希 persona ——
|
|
const p = db.prepare('SELECT content_json FROM account_persona WHERE account_id=? ORDER BY id DESC LIMIT 1').get(ACCOUNT_ID);
|
|
const personaText = personaForCreate(p ? p.content_json : '');
|
|
console.log('persona len:', personaText.length);
|
|
|
|
const tasks = [];
|
|
for (const vid of VIDEO_IDS) {
|
|
const v = db.prepare('SELECT id, aweme_id, video_title, video_url FROM account_videos WHERE id=?').get(vid);
|
|
if (!v) { console.error('video not found', vid); continue; }
|
|
const s = db.prepare('SELECT content_json FROM account_video_source WHERE aweme_id=?').get(v.aweme_id);
|
|
const content = String(s ? s.content_json : '');
|
|
const title = cleanTitle(v.video_title);
|
|
|
|
// 前端 buildTopicBlock 逻辑
|
|
const tc = extractVideoTopic(content);
|
|
let block, mode, createTail;
|
|
if (tc) {
|
|
block = `\n【视频选题】\n标题：《${title}》\n# 选题\n${tc}\n`;
|
|
mode = 'topic';
|
|
createTail = '请结合账号设定与视频选题，创作一个全新的故事脚本（新故事、新情节，不要复刻原视频），输出完整故事脚本。';
|
|
} else {
|
|
const body = content.trim();
|
|
if (!body) {
|
|
block = `\n【视频选题】\n标题：《${title}》\n`;
|
|
mode = 'none';
|
|
createTail = '请结合账号设定与视频选题，创作一个全新的故事脚本（新故事、新情节，不要复刻原视频），输出完整故事脚本。';
|
|
} else {
|
|
block = `\n【视频解析内容（供提炼选题）】\n标题：《${title}》\n${body}\n`;
|
|
mode = 'content';
|
|
createTail = '请先通读以上【视频解析内容】，提炼出该视频的选题（赛道/受众/视频主题/开场钩子/中间留人/结尾落点等），再结合账号设定与提炼出的选题，创作一个全新的故事脚本（新故事、新情节，不要复刻原视频），输出完整故事脚本。';
|
|
}
|
|
}
|
|
const personaBlock = personaText ? `\n【账号设定（人设卡）】\n${personaText}\n` : '';
|
|
const noPersonaNote = personaText ? '' : `\n该账号暂无账号设定，请先为达人「${ACCOUNT_NAME}」提取账号设定（生成人设卡），再基于账号设定与选题创作。`;
|
|
|
|
const prompt = `请调用 Skill 工具加载「短视频脚本创作」技能（技能根目录 = 用户技能目录 ~/.workbuddy/skills/短视频脚本创作/，先读取其 SKILL.md 及 references 规范；若 Skill 工具无法加载，请直接读取该目录下的 SKILL.md 文件）。遵循 S1-S11 完整创作流程，最终输出 S9 标准脚本格式（按场次+对白组织，画面标注【景别｜环境｜光线｜场景｜场景元素】），禁止输出分镜头表格。
|
|
创作时必须逐条读取并严格执行 SKILL.md 的「开场钩子铁律」与【账号设定】的「内容规则」（如果…则…）——二者为硬约束，不得自由发挥、不得跳过，具体标准一律以 SKILL.md 与内容规则原文为准。
|
|
请为达人「${ACCOUNT_NAME}」创作一条全新的短视频故事脚本。${personaBlock}${noPersonaNote}${block}${createTail}
|
|
|
|
【任务完成后必须执行】
|
|
将完整脚本保存到工作台数据库 mcn-plugin.db 的 rewrite_log 表（数据库路径：D:\\AgentSkill\\mcn-short-video\\project\\短视频脚本创作\\V1.0\\mcn-work-shop\\mcn-plugin.db），字段：
|
|
- video_id=${v.id}（整数）
|
|
- aweme_id=${v.aweme_id}
|
|
- account_id=${ACCOUNT_ID}
|
|
- title=脚本标题
|
|
- script_text=完整脚本全文
|
|
- status='done'
|
|
（用 node:sqlite 或 sqlite3 写入；若该表不存在则创建后再写）
|
|
完成后回复：脚本标题 + 是否已保存到数据库。`;
|
|
|
|
tasks.push({ videoId: v.id, mode, title, prompt });
|
|
console.log(`TASK v${v.id} | mode=${mode} | promptLen=${prompt.length} | ${title.slice(0, 30)}`);
|
|
}
|
|
db.close();
|
|
|
|
// —— 提交任务（逐个串行，间隔 3s）——
|
|
const sleep = (ms) => new Promise((r) => setTimeout(r, ms));
|
|
(async () => {
|
|
const results = [];
|
|
for (const t of tasks) {
|
|
try {
|
|
const resp = await fetch(API, {
|
|
method: 'POST',
|
|
headers: { 'Content-Type': 'application/json' },
|
|
body: JSON.stringify({ prompt: t.prompt, name: '创作指令', skills: ['短视频脚本创作'] }),
|
|
});
|
|
const d = await resp.json();
|
|
results.push({ videoId: t.videoId, id: d.id, ok: d.ok, error: d.error });
|
|
console.log(`SUBMIT v${t.videoId} →`, d.ok ? ('id=' + d.id) : ('ERR ' + d.error));
|
|
} catch (e) {
|
|
results.push({ videoId: t.videoId, error: e.message });
|
|
console.log(`SUBMIT v${t.videoId} → EXCEPTION ${e.message}`);
|
|
}
|
|
await sleep(3000);
|
|
}
|
|
console.log('\n=== RESULTS ===');
|
|
console.log(JSON.stringify(results, null, 1));
|
|
})();
|