sion123
35488beef2
- 新增 gpt-image-generator.js 脚本，支持文生图、图生图/重绘、批量生成 - 更新 pipeline 和 phase-images 支持 GPT Image 模型 - 更新技能文档，添加 GPT Image 使用说明和 API 特点 - 新增配置文件中的 GPT Image API 参数
247 lines
9.1 KiB
JavaScript
247 lines
9.1 KiB
JavaScript
#!/usr/bin/env node
|
||
|
||
/**
|
||
* 统一视频生产流程编排器
|
||
*
|
||
* 串联: 生图 → 上传 → 生视频 → TTS → 成片
|
||
* manifest.json 是唯一数据源，支持断点续跑
|
||
*
|
||
* 模块结构:
|
||
* pipeline.js ← CLI 入口 + 编排逻辑（本文件）
|
||
* lib/pipeline-utils.js ← 共享工具函数 & 路径常量
|
||
* lib/phase-*.js ← 各阶段处理器
|
||
* lib/cmd-*.js ← 子命令实现
|
||
*
|
||
* 用法:
|
||
* node scripts/pipeline.js run --manifest ./output/batch/manifest.json --account military --resume
|
||
* node scripts/pipeline.js run --manifest ./output/batch/manifest.json --phase upload,videos
|
||
* node scripts/pipeline.js status --manifest ./output/batch/manifest.json
|
||
*/
|
||
|
||
const { loadManifest, saveManifest, loadAccountConfig, log } = require('./lib/pipeline-utils')
|
||
const { phaseImages } = require('./lib/phase-images')
|
||
const { phaseUpload } = require('./lib/phase-upload')
|
||
const { phaseVideos } = require('./lib/phase-videos')
|
||
const { phaseTts } = require('./lib/phase-tts')
|
||
const { phaseAssemble } = require('./lib/phase-assemble')
|
||
const { initManifest } = require('./lib/cmd-init')
|
||
const { validateManifest, validateAccount } = require('./lib/cmd-validate')
|
||
const { showStatus } = require('./lib/cmd-status')
|
||
const { confirmManifest } = require('./lib/cmd-confirm')
|
||
const { createAccount } = require('./lib/cmd-create-account')
|
||
|
||
// ============================================================================
|
||
// 阶段注册表
|
||
// ============================================================================
|
||
|
||
const ALL_PHASES = ['images', 'upload', 'videos', 'tts', 'assemble']
|
||
|
||
const PHASE_HANDLERS = {
|
||
images: phaseImages,
|
||
upload: phaseUpload,
|
||
videos: phaseVideos,
|
||
tts: phaseTts,
|
||
assemble: phaseAssemble,
|
||
}
|
||
|
||
// ============================================================================
|
||
// Pipeline 编排
|
||
// ============================================================================
|
||
|
||
async function runPipeline(manifestPath, options) {
|
||
const manifest = loadManifest(manifestPath)
|
||
|
||
// 加载账号配置
|
||
const accountId = manifest.account || options.account
|
||
let accountConfig = {}
|
||
if (accountId) {
|
||
accountConfig = loadAccountConfig(accountId)
|
||
log('pipeline', `账号: ${accountConfig.name || accountId}`)
|
||
}
|
||
|
||
const phases = options.phases || ALL_PHASES
|
||
|
||
if (!manifest.pipeline) {
|
||
manifest.pipeline = { phases: {} }
|
||
}
|
||
|
||
// --retry-failed: 重置失败 item 状态
|
||
if (options.retryFailed) {
|
||
applyRetryFailed(manifest, phases)
|
||
}
|
||
|
||
log('pipeline', `阶段: ${phases.join(' → ')}`)
|
||
|
||
for (const phase of phases) {
|
||
const handler = PHASE_HANDLERS[phase]
|
||
if (!handler) { log('pipeline', `未知阶段: ${phase}`); continue }
|
||
|
||
if (options.resume && manifest.pipeline.phases[phase] === 'done') {
|
||
log(phase, '已完成，跳过 (--resume)')
|
||
continue
|
||
}
|
||
|
||
manifest.pipeline.phases[phase] = 'running'
|
||
saveManifest(manifestPath, manifest)
|
||
|
||
try {
|
||
await handler(manifest, manifestPath, { ...options, accountConfig })
|
||
|
||
const failedCount = manifest.items.filter(it => it.status === 'failed').length
|
||
if (failedCount > 0) {
|
||
manifest.pipeline.phases[phase] = 'partial'
|
||
log(phase, `完成 (${failedCount} 个失败)`)
|
||
} else {
|
||
manifest.pipeline.phases[phase] = 'done'
|
||
log(phase, '完成')
|
||
}
|
||
} catch (err) {
|
||
manifest.pipeline.phases[phase] = 'failed'
|
||
log(phase, `阶段失败: ${err.message}`)
|
||
saveManifest(manifestPath, manifest)
|
||
throw err
|
||
}
|
||
saveManifest(manifestPath, manifest)
|
||
}
|
||
|
||
log('pipeline', '全部完成')
|
||
}
|
||
|
||
function applyRetryFailed(manifest, phases) {
|
||
let resetCount = 0
|
||
for (const item of manifest.items) {
|
||
if (item.status === 'failed' || item.status === 'partial') {
|
||
if (item.url && item.videoPrompt && !item.video) {
|
||
// 图片已上传但视频未生成 → 直接重试视频阶段
|
||
item.status = 'done'
|
||
item.error = ''
|
||
resetCount++
|
||
} else if (!item.url && item.imagePrompt) {
|
||
// 图片未上传 → 重试图片阶段
|
||
// 如果首帧已存在但 lastFrame 失败，只重置 lastFrame 相关
|
||
if (item.file && manifest.mode === 'framePair' && !item.lastFrame) {
|
||
item.status = 'done' // 保留首帧，只补 lastFrame
|
||
item.error = ''
|
||
resetCount++
|
||
} else {
|
||
item.status = 'pending'
|
||
item.error = ''
|
||
delete item.file // 清除旧文件引用，避免重复
|
||
resetCount++
|
||
}
|
||
}
|
||
}
|
||
}
|
||
if (phases.includes('videos')) {
|
||
if (manifest.items.some(it => it.status === 'done' && it.url && it.videoPrompt && !it.video)) {
|
||
manifest.pipeline.phases.videos = 'pending'
|
||
}
|
||
}
|
||
if (phases.includes('images')) {
|
||
if (manifest.items.some(it => (!it.status || it.status === 'pending') || (it.status === 'done' && manifest.mode === 'framePair' && !it.lastFrame))) {
|
||
manifest.pipeline.phases.images = 'pending'
|
||
}
|
||
}
|
||
if (phases.includes('upload')) manifest.pipeline.phases.upload = 'pending'
|
||
if (phases.includes('tts')) manifest.pipeline.phases.tts = 'pending'
|
||
if (resetCount > 0) log('pipeline', `重置 ${resetCount} 个失败 item (--retry-failed)`)
|
||
}
|
||
|
||
// ============================================================================
|
||
// CLI 参数解析
|
||
// ============================================================================
|
||
|
||
function parseArgs(argv) {
|
||
const args = {}
|
||
for (let i = 0; i < argv.length; i++) {
|
||
if (argv[i] === '--manifest' && argv[i + 1]) args.manifest = argv[++i]
|
||
else if (argv[i] === '--account' && argv[i + 1]) args.account = argv[++i]
|
||
else if (argv[i] === '--phase' && argv[i + 1]) args.phases = argv[++i].split(',')
|
||
else if (argv[i] === '--resume') args.resume = true
|
||
else if (argv[i] === '--retry-failed') args.retryFailed = true
|
||
else if (argv[i] === '--mode' && argv[i + 1]) args.mode = argv[++i]
|
||
else if (argv[i] === '--items' && argv[i + 1]) args.items = argv[++i]
|
||
else if (argv[i] === '--items-file' && argv[i + 1]) args.itemsFile = argv[++i]
|
||
else if (argv[i] === '--id' && argv[i + 1]) args.id = argv[++i]
|
||
else if (argv[i] === '--name' && argv[i + 1]) args.name = argv[++i]
|
||
else if (argv[i] === '--desc' && argv[i + 1]) args.desc = argv[++i]
|
||
else if (argv[i] === '--format' && argv[i + 1]) args.format = argv[++i]
|
||
else if (argv[i] === '--image-model' && argv[i + 1]) args.imageModel = argv[++i]
|
||
else if (argv[i] === '--video-model' && argv[i + 1]) args.videoModel = argv[++i]
|
||
else if (argv[i] === '--references' && argv[i + 1]) args.references = argv[++i]
|
||
else if (argv[i] === '--all') args.all = true
|
||
else if (!args.command) args.command = argv[i]
|
||
}
|
||
return args
|
||
}
|
||
|
||
// ============================================================================
|
||
// CLI 入口
|
||
// ============================================================================
|
||
|
||
async function main() {
|
||
const args = parseArgs(process.argv.slice(2))
|
||
const command = args.command
|
||
|
||
if (command === 'init') {
|
||
initManifest(args)
|
||
return
|
||
}
|
||
|
||
if (command === 'validate') {
|
||
if (!args.manifest) { console.error('用法: pipeline.js validate --manifest <path>'); process.exit(1) }
|
||
validateManifest(args.manifest)
|
||
return
|
||
}
|
||
|
||
if (command === 'confirm') {
|
||
confirmManifest(args)
|
||
return
|
||
}
|
||
|
||
if (command === 'status') {
|
||
if (!args.manifest) { console.error('用法: pipeline.js status --manifest <path>'); process.exit(1) }
|
||
showStatus(args.manifest)
|
||
return
|
||
}
|
||
|
||
if (command === 'run') {
|
||
if (!args.manifest) { console.error('用法: pipeline.js run --manifest <path> [--account id] [--phase p1,p2] [--resume] [--retry-failed]'); process.exit(1) }
|
||
await runPipeline(args.manifest, args)
|
||
return
|
||
}
|
||
|
||
if (command === 'create-account') {
|
||
await createAccount(args)
|
||
return
|
||
}
|
||
|
||
if (command === 'validate-account') {
|
||
if (!args.account) { console.error('用法: pipeline.js validate-account --account <id>'); process.exit(1) }
|
||
validateAccount(args.account)
|
||
return
|
||
}
|
||
|
||
console.log('用法:')
|
||
console.log(' pipeline.js create-account --id <id> --name <名称> [--desc ...] [--references file1,file2]')
|
||
console.log(' pipeline.js validate-account --account <id>')
|
||
console.log(' pipeline.js init --account <id> --mode <single|framePair> --items <JSON> [--items-file <path>] [--image-model gemini|gpt-image|mj] [--video-model veo3-fast|grok|kling] [--format 9:16]')
|
||
console.log(' pipeline.js validate --manifest <path>')
|
||
console.log(' pipeline.js confirm --manifest <path> --all')
|
||
console.log(' pipeline.js confirm --manifest <path> --items 1,3,5')
|
||
console.log(' pipeline.js run --manifest <path> [--account id] [--phase p1,p2] [--resume] [--retry-failed]')
|
||
console.log(' pipeline.js status --manifest <path>')
|
||
console.log('')
|
||
console.log('Manifest 路径约定: output/{account}_{date}_{NNN}/manifest.json（同天自增序号）')
|
||
console.log('阶段: images, upload, videos, tts, assemble')
|
||
}
|
||
|
||
if (require.main === module) {
|
||
main().catch(err => {
|
||
console.error(`\n错误: ${err.message}`)
|
||
process.exit(1)
|
||
})
|
||
}
|
||
|
||
module.exports = { runPipeline, showStatus, initManifest, validateManifest, createAccount, validateAccount }
|