agens-studio 0.1.0

package/cli.js

@@ -0,0 +1,978 @@
+#!/usr/bin/env node
+
+/**
+ * Agens 创作工作台 CLI v1.0
+ * =========================================
+ * 复用现有 service 层（agensClient / mimoClient / mimoVision / assetStore / promptMemory）
+ * 提供纯终端交互：菜单选择 → 参数输入 → AI 生成 → 下载结果 → 自动打开
+ *
+ * 启动：node cli.js  或  npm run cli
+ */
+
+import readline from 'node:readline';
+import fs from 'node:fs/promises';
+import path from 'node:path';
+import { randomUUID } from 'node:crypto';
+import { exec } from 'node:child_process';
+import { fileURLToPath } from 'node:url';
+import { config, USER_CONFIG_FILE, saveUserConfig, readUserConfig } from './config.js';
+import { generateImage, generateVideo, getTaskStatus } from './src/services/agensClient.js';
+import { optimizePrompt, optimizeNegativePrompt, dimensionsToText } from './src/services/mimoClient.js';
+import { describeImage, descriptionToText } from './src/services/mimoVision.js';
+import * as assetStore from './src/services/assetStore.js';
+import * as promptMemory from './src/services/promptMemory.js';
+
+// ═══════════════════ ANSI 色彩 ═══════════════════
+
+const C = {
+  reset: '\x1b[0m', bold: '\x1b[1m',
+  red: '\x1b[31m', green: '\x1b[32m', yellow: '\x1b[33m',
+  blue: '\x1b[34m', cyan: '\x1b[36m', white: '\x1b[37m', gray: '\x1b[90m',
+};
+
+// ═══════════════════ 终端 I/O 工具 ═══════════════════
+
+const rl = readline.createInterface({ input: process.stdin, output: process.stdout });
+let _rlClosed = false;
+rl.on('close', () => { _rlClosed = true; });
+
+/**
+ * 交互式提问，支持默认值
+ * @param {string} q 提示文字
+ * @param {string} [def] 默认值（回车时使用）
+ * @returns {Promise<string>}
+ */
+function ask(q, def) {
+  if (_rlClosed) {
+    process.exit(0);
+  }
+  const suffix = def != null && def !== '' ? ` (${def})` : '';
+  return new Promise((resolve) => {
+    rl.question(`${C.cyan}${q}${suffix}：${C.reset}`, (a) => {
+      resolve((a || '').trim() || (def != null ? def : ''));
+    });
+  });
+}
+
+/** 按回车继续 */
+async function pause() {
+  return new Promise((resolve) => {
+    rl.question(`\n${C.gray}按回车继续...${C.reset}`, () => resolve());
+  });
+}
+
+/** 确认（默认 Y） */
+async function confirm(q) {
+  const a = await ask(q + ' (Y/n)', 'Y');
+  return a.toLowerCase() !== 'n';
+}
+
+/**
+ * 编号选择菜单
+ * @param {string} title 标题
+ * @param {string[]} opts 选项文字
+ * @returns {Promise<number>} 选中的索引（0-based），取消返回 -1
+ */
+async function choose(title, opts) {
+  console.log(`\n${C.yellow}${title}${C.reset}`);
+  opts.forEach((o, i) => console.log(`  ${C.cyan}[${i + 1}]${C.reset} ${o}`));
+  console.log(`  ${C.gray}[0] 取消${C.reset}`);
+  const c = await ask('选择');
+  const n = parseInt(c, 10);
+  if (isNaN(n) || n < 0 || n > opts.length) return -1;
+  return n - 1; // 0 = 取消 → -1
+}
+
+// ═══════════════════ 通用工具函数 ═══════════════════
+
+function sleep(ms) { return new Promise((r) => setTimeout(r, ms)); }
+
+function formatSize(bytes) {
+  if (bytes < 1024) return bytes + ' B';
+  if (bytes < 1048576) return (bytes / 1024).toFixed(1) + ' KB';
+  return (bytes / 1048576).toFixed(1) + ' MB';
+}
+
+function fmtDate(ts) {
+  return new Date(ts).toLocaleString('zh-CN', { timeZone: 'Asia/Shanghai' });
+}
+
+/** 去掉用户拖拽文件路径时自带的引号 */
+function cleanPath(input) {
+  let s = (input || '').trim();
+  if ((s.startsWith('"') && s.endsWith('"')) || (s.startsWith("'") && s.endsWith("'"))) {
+    s = s.slice(1, -1).trim();
+  }
+  return s;
+}
+
+/** 读取本地图片 → base64 data URI（用于生图 API） */
+async function imgToDataUri(input) {
+  const s = cleanPath(input);
+  if (!s) return '';
+  if (s.startsWith('data:')) return s;
+  try {
+    const abs = path.isAbsolute(s) ? s : path.resolve(s);
+    const buf = await fs.readFile(abs);
+    const ext = path.extname(abs).slice(1).toLowerCase();
+    const mime = 'image/' + (ext === 'jpg' ? 'jpeg' : ext || 'png');
+    return 'data:' + mime + ';base64,' + buf.toString('base64');
+  } catch {
+    return s; // 可能是 URL
+  }
+}
+
+/** 读取本地图片 → raw base64（用于视频 API） */
+async function imgToB64(input) {
+  const s = cleanPath(input);
+  if (!s) return '';
+  if (s.startsWith('data:')) {
+    const i = s.indexOf(',');
+    return i >= 0 ? s.slice(i + 1) : s;
+  }
+  try {
+    const abs = path.isAbsolute(s) ? s : path.resolve(s);
+    const buf = await fs.readFile(abs);
+    return buf.toString('base64');
+  } catch {
+    return s;
+  }
+}
+
+/** 读取本地图片 → base64 data URI（用于 vision 识图 API） */
+async function imgToVisionUri(input) {
+  const s = cleanPath(input);
+  if (!s) throw new Error('图片路径为空');
+  if (s.startsWith('data:')) return s;
+  const abs = path.isAbsolute(s) ? s : path.resolve(s);
+  const buf = await fs.readFile(abs);
+  const ext = path.extname(abs).slice(1).toLowerCase();
+  const mime = 'image/' + (ext === 'jpg' ? 'jpeg' : ext || 'png');
+  return 'data:' + mime + ';base64,' + buf.toString('base64');
+}
+
+// ═══════════════════ 公共流程 ═══════════════════
+
+/**
+ * 提示词优化交互流程
+ * @param {string} prompt 原始提示词
+ * @param {string} mode 生成模式（image/img2img/text2video 等）
+ * @param {string} [imageDesc] 图生图时的识图结果
+ * @returns {Promise<string>} 最终使用的提示词
+ */
+async function promptOptimizeFlow(prompt, mode, imageDesc) {
+  if (!prompt) return prompt;
+  const wantOpt = await confirm('是否用 AI 优化提示词');
+  if (!wantOpt) return prompt;
+
+  console.log(`${C.yellow}[...] 正在优化提示词（MiMo ${config.mimo.model}）...${C.reset}`);
+  try {
+    const result = await optimizePrompt({
+      text: prompt, mode, imageDescription: imageDesc,
+    });
+    const optText = result.dimensions
+      ? dimensionsToText(result.dimensions)
+      : result.optimized;
+
+    console.log(`\n${C.green}═══ 优化结果 ═══${C.reset}`);
+    if (result.dimensions) {
+      for (const [k, v] of Object.entries(result.dimensions)) {
+        console.log(`  ${C.cyan}[${k}]${C.reset} ${v}`);
+      }
+      console.log('');
+    }
+    console.log(`  ${optText}`);
+
+    const use = await confirm('使用优化后的提示词');
+    return use ? optText : prompt;
+  } catch (e) {
+    console.log(`${C.red}优化失败：${e.message}${C.reset}`);
+    return prompt;
+  }
+}
+
+/**
+ * 反向提示词交互
+ * @param {string} positive 正向提示词（供参考）
+ * @returns {Promise<string>}
+ */
+async function negPromptFlow(positive) {
+  const wantNeg = await confirm('是否添加反向提示词（排除不想要的元素）');
+  if (!wantNeg) return '';
+  const negInput = await ask('描述不想出现的内容', '模糊, 低质量, 变形, 多余手指');
+  if (!negInput) return '';
+  console.log(`${C.yellow}[...] 优化反向提示词...${C.reset}`);
+  try {
+    const result = await optimizeNegativePrompt({ text: negInput, mode: 'image', positive });
+    console.log(`${C.green}反向提示词：${C.reset}${result.optimized || result}`);
+    return result.optimized || result;
+  } catch (e) {
+    console.log(`${C.red}反向优化失败：${e.message}${C.reset}`);
+    return negInput;
+  }
+}
+
+/**
+ * 视频轮询（显示进度条）
+ * @param {string} videoId
+ * @returns {Promise<{status, videoUrl, error}>}
+ */
+async function pollVideo(videoId) {
+  const maxAttempts = config.agens.pollMaxAttempts || 120;
+  const interval = config.agens.pollInterval || 5000;
+
+  for (let i = 0; i < maxAttempts; i++) {
+    await sleep(interval);
+    let info;
+    try {
+      info = await getTaskStatus(videoId);
+    } catch {
+      process.stdout.write(`${C.gray}?${C.reset}`);
+      continue;
+    }
+
+    const pct = info.progress || Math.min(95, 5 + i * 3);
+    const bar = makeBar(pct, 30);
+    process.stdout.write(`\r  ${bar} ${String(Math.round(pct)).padStart(3)}%  `);
+
+    if (info.status === 'success') {
+      process.stdout.write('\r' + ' '.repeat(50) + '\r');
+      return { status: 'success', videoUrl: info.result?.videoUrl };
+    }
+    if (info.status === 'failed') {
+      process.stdout.write('\r' + ' '.repeat(50) + '\r');
+      return { status: 'failed', error: info.result?.error || '生成失败' };
+    }
+  }
+  return { status: 'failed', error: '轮询超时（约 10 分钟）' };
+}
+
+function makeBar(pct, w) {
+  const filled = Math.round(w * pct / 100);
+  return `${C.green}${'█'.repeat(filled)}${C.gray}${'░'.repeat(w - filled)}${C.reset}`;
+}
+
+/**
+ * 下载远程文件到本地 assets
+ * - 支持 data: URI 直接写入（无需网络请求）
+ * - 远程 URL 带 60s 超时 + 调试日志
+ * @returns {Promise<{filePath, asset}>}
+ */
+async function downloadResult(url, kind, mime, meta) {
+  // data URI：直接走本地保存，不下载
+  if (url.startsWith('data:')) {
+    console.log(`${C.yellow}[...] 正在保存（base64 数据）...${C.reset}`);
+    const b64 = url.split(',')[1];
+    if (!b64) throw new Error('data URI 格式异常');
+    const buf = Buffer.from(b64, 'base64');
+    const ext = mime === 'video/mp4' ? '.mp4' : '.png';
+    const subdir = kind === 'video' ? 'videos' : 'images';
+    // randomUUID already imported at top
+    const id = randomUUID();
+    const filename = `${id}${ext}`;
+    const absPath = path.join(config.dirs[subdir], filename);
+    await fs.mkdir(config.dirs[subdir], { recursive: true });
+    await fs.writeFile(absPath, buf);
+    // 登记到 assetStore
+    const asset = await assetStore.registerLocal({ absPath, kind, meta });
+    return { filePath: absPath, asset };
+  }
+
+  // 远程 URL
+  const preview = url.length > 80 ? url.slice(0, 80) + '...' : url;
+  console.log(`${C.yellow}[...] 正在保存到本地...${C.reset}`);
+  console.log(`  ${C.gray}URL: ${preview}${C.reset}`);
+
+  try {
+    const asset = await assetStore.downloadAndRegister({ url, kind, mime, meta });
+    const absPath = path.join(config.dirs.root, asset.path);
+    return { filePath: absPath, asset };
+  } catch (dlErr) {
+    // 下载失败：尝试直接 fetch 带超时
+    console.log(`  ${C.red}assetStore 下载失败：${dlErr.message}${C.reset}`);
+    console.log(`${C.yellow}[...] 重试下载（60s 超时）...${C.reset}`);
+
+    const controller = new AbortController();
+    const timer = setTimeout(() => controller.abort(), 60000);
+    try {
+      const res = await fetch(url, { signal: controller.signal });
+      clearTimeout(timer);
+      if (!res.ok) throw new Error(`HTTP ${res.status}`);
+      const buf = Buffer.from(await res.arrayBuffer());
+      const ext = mime === 'video/mp4' ? '.mp4' : '.png';
+      const subdir = kind === 'video' ? 'videos' : 'images';
+      // randomUUID already imported at top
+      const id = randomUUID();
+      const filename = `${id}${ext}`;
+      const absPath = path.join(config.dirs[subdir], filename);
+      await fs.mkdir(config.dirs[subdir], { recursive: true });
+      await fs.writeFile(absPath, buf);
+      const asset = await assetStore.registerLocal({ absPath, kind, meta });
+      console.log(`  ${C.green}重试下载成功！${C.reset}`);
+      return { filePath: absPath, asset };
+    } catch (e2) {
+      clearTimeout(timer);
+      const reason = e2.name === 'AbortError' ? '超时（60s）' : e2.message;
+      throw new Error(`下载失败：${reason}\n原始 URL: ${url.slice(0, 200)}`);
+    }
+  }
+}
+
+/** Windows 下用 start 命令打开文件 */
+function openFile(filePath) {
+  exec(`start "" "${filePath}"`, (err) => {
+    if (err) console.log(`${C.gray}（自动打开失败，请手动查看文件）${C.reset}`);
+  });
+}
+
+/** 生成完成后展示结果 + 打开 */
+async function showResult(filePath, asset, w, h) {
+  console.log(`\n${C.green}${C.bold}═══ 生成完成！═══${C.reset}`);
+  console.log(`  文件：${filePath}`);
+  console.log(`  大小：${formatSize(asset.size)}`);
+  if (w && h) console.log(`  尺寸：${w} x ${h}`);
+  console.log('');
+  if (await confirm('打开文件')) {
+    openFile(filePath);
+  }
+}
+
+// ═══════════════════ 首次运行 Setup Wizard ═══════════════════
+
+/**
+ * 首次运行引导：配置 API Key
+ * 保存到 ~/.agens-cli/config.json，后续启动自动加载
+ */
+async function setupWizard(isFirstRun) {
+  if (isFirstRun) {
+    console.log('');
+    console.log(`${C.yellow}╔══════════════════════════════════════════╗${C.reset}`);
+    console.log(`${C.yellow}║    欢迎使用 Agens 创作工作台 CLI！       ║${C.reset}`);
+    console.log(`${C.yellow}║    首次运行需要配置 API Key               ║${C.reset}`);
+    console.log(`${C.yellow}╚══════════════════════════════════════════╝${C.reset}`);
+    console.log('');
+    console.log(`${C.gray}配置文件将保存到：${USER_CONFIG_FILE}${C.reset}`);
+    console.log(`${C.gray}你可以随时用菜单 [9] 重新配置${C.reset}`);
+    console.log('');
+  } else {
+    console.log(`\n${C.bold}─────── 配置管理 ───────${C.reset}`);
+    const existing = readUserConfig();
+    if (existing) {
+      const mask = (s) => s ? s.slice(0, 8) + '···' + s.slice(-4) : '(未设置)';
+      console.log(`  当前 Agnes Key：${mask(existing.agens_api_key)}`);
+      console.log(`  当前 MiMo Key：${mask(existing.mimo_api_key)}`);
+      console.log('');
+      if (!(await confirm('重新配置'))) return;
+    }
+  }
+
+  const cfg = readUserConfig() || {};
+
+  // Agnes API Key
+  console.log(`${C.cyan}── Agnes AI API Key ──${C.reset}`);
+  console.log(`${C.gray}用于图片生成和视频生成${C.reset}`);
+  console.log(`${C.gray}获取方式：https://agnes-ai.com 注册后在控制台获取${C.reset}`);
+  const agnesKey = await ask('Agnes API Key', cfg.agens_api_key || '');
+  if (agnesKey) cfg.agens_api_key = agnesKey;
+
+  // MiMo API Key
+  console.log(`\n${C.cyan}── MiMo 大模型 API Key ──${C.reset}`);
+  console.log(`${C.gray}用于 AI 提示词优化和识图（可选，不填则跳过优化功能）${C.reset}`);
+  console.log(`${C.gray}获取方式：https://xiaomimimo.com Token Plan 获取${C.reset}`);
+  const mimoKey = await ask('MiMo API Key (tp- 开头)', cfg.mimo_api_key || '');
+  if (mimoKey) cfg.mimo_api_key = mimoKey;
+
+  // 高级设置（可选）
+  console.log('');
+  const wantAdvanced = await confirm('是否配置高级选项（自定义 Base URL 等）');
+  if (wantAdvanced) {
+    const agnesBaseUrl = await ask('Agnes Base URL', cfg.agens_image_base_url || 'https://apihub.agnes-ai.com/v1');
+    if (agnesBaseUrl !== 'https://apihub.agnes-ai.com/v1') cfg.agens_image_base_url = agnesBaseUrl;
+
+    const mimoBaseUrl = await ask('MiMo Base URL', cfg.mimo_base_url || 'https://token-plan-cn.xiaomimimo.com/v1');
+    if (mimoBaseUrl !== 'https://token-plan-cn.xiaomimimo.com/v1') cfg.mimo_base_url = mimoBaseUrl;
+  }
+
+  // 保存
+  saveUserConfig(cfg);
+  console.log(`\n${C.green}${C.bold}配置已保存到 ${USER_CONFIG_FILE}${C.reset}`);
+
+  if (isFirstRun) {
+    console.log(`\n${C.yellow}请重新启动 CLI 以加载新配置：${C.reset}`);
+    console.log(`  ${C.cyan}node cli.js${C.reset}`);
+    process.exit(0);
+  }
+
+  console.log(`${C.gray}提示：部分配置需要重启 CLI 才能生效${C.reset}`);
+  await pause();
+}
+
+// ═══════════════════ 初始化 ═══════════════════
+
+async function init() {
+  await assetStore.init();
+  await promptMemory.init();
+  // 确保目录就绪
+  for (const dir of [config.dirs.images, config.dirs.videos, config.dirs.data, config.dirs.uploads]) {
+    await fs.mkdir(dir, { recursive: true });
+  }
+}
+
+// ═══════════════════ 主菜单 ═══════════════════
+
+async function mainMenu() {
+  while (true) {
+    console.log('');
+    console.log(`${C.yellow}==================================================${C.reset}`);
+    console.log(`${C.yellow}${C.bold}  Agens 创作工作台 CLI v1.0${C.reset}`);
+    console.log(`${C.yellow}  图片 · 视频 · AI 提示词优化 · 素材库${C.reset}`);
+    console.log(`${C.yellow}==================================================${C.reset}`);
+    console.log(`  API：${config.agens.image.apiKey ? C.green + '已配置' + C.reset : C.red + '未配置' + C.reset}`);
+    console.log(`  模型：${config.agens.image.model} / ${config.agens.video.model}`);
+    console.log('');
+    console.log(`  ${C.cyan}[1]${C.reset} 文生图`);
+    console.log(`  ${C.cyan}[2]${C.reset} 图生图`);
+    console.log(`  ${C.cyan}[3]${C.reset} 文生视频`);
+    console.log(`  ${C.cyan}[4]${C.reset} 图生视频`);
+    console.log(`  ${C.cyan}[5]${C.reset} 多图生视频`);
+    console.log(`  ${C.cyan}[6]${C.reset} 关键帧动画`);
+    console.log(`  ${C.cyan}[7]${C.reset} 素材库`);
+    console.log(`  ${C.cyan}[8]${C.reset} 查看配置`);
+    console.log(`  ${C.cyan}[9]${C.reset} 配置 API Key`);
+    console.log(`  ${C.gray}[0]${C.reset} 退出`);
+    console.log('');
+
+    const c = await ask('请选择功能');
+
+    switch (c) {
+      case '1': await modeText2Image(); break;
+      case '2': await modeImage2Image(); break;
+      case '3': await modeText2Video(); break;
+      case '4': await modeImage2Video(); break;
+      case '5': await modeMulti2Video(); break;
+      case '6': await modeKeyframe(); break;
+      case '7': await menuAssets(); break;
+      case '8': await menuConfig(); break;
+      case '9': await setupWizard(false); break;
+      case '0':
+        console.log(`\n${C.green}再见！${C.reset}\n`);
+        process.exit(0);
+        break;
+      default:
+        console.log(`${C.red}无效选择，请输入 0-9 的数字${C.reset}`);
+    }
+  }
+}
+
+// ═══════════════════ [1] 文生图 ═══════════════════
+
+async function modeText2Image() {
+  console.log(`\n${C.bold}─────── 文生图 ───────${C.reset}`);
+
+  const rawPrompt = await ask('描述你想生成的画面');
+  if (!rawPrompt) { console.log(`${C.yellow}已取消${C.reset}`); return; }
+
+  const finalPrompt = await promptOptimizeFlow(rawPrompt, 'image');
+
+  const ratio = await ask('比例 (1:1 / 3:4 / 4:3 / 16:9 / 9:16)', '16:9');
+  const styleIdx = await choose('风格', ['写实摄影', '动漫插画', '3D渲染', '数字艺术', '不指定']);
+  const styleMap = ['photographic', 'anime', '3d', 'art', ''];
+  const style = styleMap[styleIdx] || '';
+  const negPrompt = await negPromptFlow(finalPrompt);
+
+  console.log(`\n${C.bold}── 生成配置 ──${C.reset}`);
+  console.log(`  提示词：${finalPrompt}`);
+  if (negPrompt) console.log(`  反向：${negPrompt}`);
+  console.log(`  比例：${ratio}  风格：${style || '默认'}`);
+
+  if (!(await confirm('开始生成'))) return;
+
+  console.log(`\n${C.yellow}[...] 正在生成图片（Agnes ${config.agens.image.model}）...${C.reset}`);
+  try {
+    const { url } = await generateImage({
+      prompt: finalPrompt,
+      params: { ratio, style, negativePrompt: negPrompt },
+    });
+    const { filePath, asset } = await downloadResult(url, 'image', 'image/png', {
+      prompt: finalPrompt, mode: 'image', params: { ratio, style },
+    });
+    await showResult(filePath, asset);
+  } catch (e) {
+    console.log(`${C.red}生成失败：${e.message}${C.reset}`);
+  }
+  await pause();
+}
+
+// ═══════════════════ [2] 图生图 ═══════════════════
+
+async function modeImage2Image() {
+  console.log(`\n${C.bold}─────── 图生图 ───────${C.reset}`);
+  console.log(`${C.gray}输入参考图片的本地文件路径（支持拖拽文件到窗口）${C.reset}`);
+
+  const imgPath = cleanPath(await ask('参考图片路径'));
+  if (!imgPath) { console.log(`${C.yellow}已取消${C.reset}`); return; }
+
+  // 识图
+  console.log(`${C.yellow}[...] 正在识别图片内容（MiMo Vision）...${C.reset}`);
+  let imageDesc = '';
+  try {
+    const dataUri = await imgToVisionUri(imgPath);
+    // 直接调用 mimo vision 获取描述
+    const descResult = await callVisionDirect(dataUri);
+    imageDesc = descResult;
+    console.log(`${C.green}识图结果：${C.reset}${descResult.slice(0, 100)}...`);
+  } catch (e) {
+    console.log(`${C.red}识图失败：${e.message}${C.reset}`);
+    console.log(`${C.gray}将继续使用（提示词优化可能不够精准）${C.reset}`);
+  }
+
+  const rawPrompt = await ask('描述你想要的修改', '转为赛博朋克风格夜景');
+  if (!rawPrompt) { console.log(`${C.yellow}已取消${C.reset}`); return; }
+
+  const finalPrompt = await promptOptimizeFlow(rawPrompt, 'img2img', imageDesc);
+  const ratio = await ask('比例 (1:1 / 3:4 / 4:3 / 16:9 / 9:16)', '1:1');
+  const negPrompt = await negPromptFlow(finalPrompt);
+
+  console.log(`\n${C.bold}── 生成配置 ──${C.reset}`);
+  console.log(`  参考图：${imgPath}`);
+  console.log(`  提示词：${finalPrompt}`);
+  if (negPrompt) console.log(`  反向：${negPrompt}`);
+  console.log(`  比例：${ratio}`);
+
+  if (!(await confirm('开始生成'))) return;
+
+  console.log(`\n${C.yellow}[...] 正在生成图片...${C.reset}`);
+  try {
+    const dataUri = await imgToDataUri(imgPath);
+    const { url } = await generateImage({
+      prompt: finalPrompt,
+      params: { ratio, style: '', negativePrompt: negPrompt, imageUrls: [dataUri] },
+    });
+    const { filePath, asset } = await downloadResult(url, 'image', 'image/png', {
+      prompt: finalPrompt, mode: 'img2img', params: { ratio },
+    });
+    await showResult(filePath, asset);
+  } catch (e) {
+    console.log(`${C.red}生成失败：${e.message}${C.reset}`);
+  }
+  await pause();
+}
+
+/** 直接调用 mimo vision 识图（不依赖 mimoVision.js 的本地路径逻辑） */
+async function callVisionDirect(dataUri) {
+  const cfg = config.mimo;
+  const system = [
+    '你是图像分析专家。请分析图片，严格按以下 6 个维度输出，每个维度一行，格式为「维度名：内容」。',
+    '', '维度：', '主体：画面核心对象是谁/是什么，什么状态/动作', '场景：发生在哪里，背景环境',
+    '构图：画面布局，主体位置', '色调：主色调、光线氛围', '风格：写实/动漫/3D/艺术/摄影等',
+    '可保留元素：图生图时应该保持不变的元素',
+    '', '要求：1. 必须输出全部 6 行 2. 每行简洁（5-30字） 3. 不要输出其它内容',
+  ].join('\n');
+
+  const res = await fetch(`${cfg.baseUrl}${cfg.endpoint}`, {
+    method: 'POST',
+    headers: { 'Content-Type': 'application/json', 'api-key': cfg.apiKey },
+    body: JSON.stringify({
+      model: cfg.visionModel,
+      messages: [
+        { role: 'system', content: system },
+        { role: 'user', content: [
+          { type: 'text', text: '请分析这张图片' },
+          { type: 'image_url', image_url: { url: dataUri } },
+        ]},
+      ],
+      max_completion_tokens: 800, temperature: 0.3, top_p: 0.9, stream: false,
+    }),
+  });
+  const text = await res.text();
+  if (!res.ok) throw new Error(`识图失败 (${res.status})`);
+  const data = JSON.parse(text);
+  return data.choices?.[0]?.message?.content?.trim() || '';
+}
+
+// ═══════════════════ [3] 文生视频 ═══════════════════
+
+async function modeText2Video() {
+  console.log(`\n${C.bold}─────── 文生视频 ───────${C.reset}`);
+
+  const rawPrompt = await ask('描述你想生成的视频场景');
+  if (!rawPrompt) { console.log(`${C.yellow}已取消${C.reset}`); return; }
+
+  const finalPrompt = await promptOptimizeFlow(rawPrompt, 'text2video');
+
+  const ratio = await ask('比例 (16:9 / 9:16 / 1:1 / 4:3 / 3:4)', '16:9');
+  const resolution = await ask('分辨率 (720p / 1080p)', '720p');
+  const duration = await ask('时长（秒，3-10）', '5');
+  const negPrompt = await negPromptFlow(finalPrompt);
+
+  console.log(`\n${C.bold}── 生成配置 ──${C.reset}`);
+  console.log(`  提示词：${finalPrompt}`);
+  if (negPrompt) console.log(`  反向：${negPrompt}`);
+  console.log(`  分辨率：${resolution}  比例：${ratio}  时长：${duration}s`);
+
+  if (!(await confirm('开始生成'))) return;
+
+  console.log(`\n${C.yellow}[...] 正在提交视频任务（Agnes ${config.agens.video.model}）...${C.reset}`);
+  try {
+    const { videoId } = await generateVideo({
+      mode: 'text2video', prompt: finalPrompt, imageUrls: [],
+      params: { ratio, resolution, duration, negativePrompt: negPrompt },
+    });
+    console.log(`  任务 ID：${videoId}`);
+    console.log(`  ${C.yellow}轮询进度（每 ${config.agens.pollInterval / 1000}s 刷新）...${C.reset}\n`);
+
+    const poll = await pollVideo(videoId);
+    if (poll.status === 'success' && poll.videoUrl) {
+      const { filePath, asset } = await downloadResult(poll.videoUrl, 'video', 'video/mp4', {
+        prompt: finalPrompt, mode: 'text2video', params: { ratio, resolution, duration },
+      });
+      await showResult(filePath, asset);
+    } else {
+      console.log(`${C.red}生成失败：${poll.error}${C.reset}`);
+    }
+  } catch (e) {
+    console.log(`${C.red}生成失败：${e.message}${C.reset}`);
+  }
+  await pause();
+}
+
+// ═══════════════════ [4] 图生视频 ═══════════════════
+
+async function modeImage2Video() {
+  console.log(`\n${C.bold}─────── 图生视频 ───────${C.reset}`);
+  console.log(`${C.gray}输入参考图片路径（支持拖拽）${C.reset}`);
+
+  const imgPath = cleanPath(await ask('参考图片路径'));
+  if (!imgPath) { console.log(`${C.yellow}已取消${C.reset}`); return; }
+
+  const rawPrompt = await ask('描述动画效果', '人物轻微呼吸，头发随风轻摆，背景灯光闪烁');
+  if (!rawPrompt) { console.log(`${C.yellow}已取消${C.reset}`); return; }
+
+  const finalPrompt = await promptOptimizeFlow(rawPrompt, 'image2video');
+
+  const ratio = await ask('比例 (16:9 / 9:16 / 1:1)', '16:9');
+  const resolution = await ask('分辨率 (720p / 1080p)', '720p');
+  const duration = await ask('时长（秒，3-10）', '5');
+  const negPrompt = await negPromptFlow(finalPrompt);
+
+  console.log(`\n${C.bold}── 生成配置 ──${C.reset}`);
+  console.log(`  参考图：${imgPath}`);
+  console.log(`  提示词：${finalPrompt}`);
+  if (negPrompt) console.log(`  反向：${negPrompt}`);
+  console.log(`  分辨率：${resolution}  比例：${ratio}  时长：${duration}s`);
+
+  if (!(await confirm('开始生成'))) return;
+
+  console.log(`\n${C.yellow}[...] 正在提交视频任务...${C.reset}`);
+  try {
+    const b64 = await imgToB64(imgPath);
+    const { videoId } = await generateVideo({
+      mode: 'image2video', prompt: finalPrompt, imageUrls: [b64],
+      params: { ratio, resolution, duration, negativePrompt: negPrompt },
+    });
+    console.log(`  任务 ID：${videoId}`);
+    console.log(`  ${C.yellow}轮询进度...${C.reset}\n`);
+
+    const poll = await pollVideo(videoId);
+    if (poll.status === 'success' && poll.videoUrl) {
+      const { filePath, asset } = await downloadResult(poll.videoUrl, 'video', 'video/mp4', {
+        prompt: finalPrompt, mode: 'image2video', params: { ratio, resolution, duration },
+      });
+      await showResult(filePath, asset);
+    } else {
+      console.log(`${C.red}生成失败：${poll.error}${C.reset}`);
+    }
+  } catch (e) {
+    console.log(`${C.red}生成失败：${e.message}${C.reset}`);
+  }
+  await pause();
+}
+
+// ═══════════════════ [5] 多图生视频 ═══════════════════
+
+async function modeMulti2Video() {
+  console.log(`\n${C.bold}─────── 多图生视频 ───────${C.reset}`);
+  console.log(`${C.gray}输入多张图片路径，生成过渡视频（2-5 张）${C.reset}`);
+
+  const images = [];
+  for (let i = 0; i < 5; i++) {
+    const p = cleanPath(await ask(`图片 ${i + 1} 路径${i >= 2 ? '（留空=完成）' : ''}`));
+    if (!p) {
+      if (i < 2) { console.log(`${C.red}至少需要 2 张图片${C.reset}`); return; }
+      break;
+    }
+    images.push(p);
+  }
+
+  console.log(`${C.green}已选择 ${images.length} 张图片${C.reset}`);
+
+  const rawPrompt = await ask('描述过渡效果', '图片之间平滑过渡，镜头缓慢推进');
+  if (!rawPrompt) { console.log(`${C.yellow}已取消${C.reset}`); return; }
+
+  const finalPrompt = await promptOptimizeFlow(rawPrompt, 'multi2video');
+
+  const ratio = await ask('比例 (16:9 / 9:16 / 1:1)', '16:9');
+  const resolution = await ask('分辨率 (720p / 1080p)', '720p');
+  const duration = await ask('时长（秒，5-15）', '5');
+  const negPrompt = await negPromptFlow(finalPrompt);
+
+  console.log(`\n${C.bold}── 生成配置 ──${C.reset}`);
+  console.log(`  图片数量：${images.length}`);
+  images.forEach((img, i) => console.log(`    [${i + 1}] ${img}`));
+  console.log(`  提示词：${finalPrompt}`);
+  if (negPrompt) console.log(`  反向：${negPrompt}`);
+  console.log(`  分辨率：${resolution}  比例：${ratio}  时长：${duration}s`);
+
+  if (!(await confirm('开始生成'))) return;
+
+  console.log(`\n${C.yellow}[...] 正在提交视频任务...${C.reset}`);
+  try {
+    const b64s = await Promise.all(images.map(imgToB64));
+    const { videoId } = await generateVideo({
+      mode: 'multi2video', prompt: finalPrompt, imageUrls: b64s,
+      params: { ratio, resolution, duration, negativePrompt: negPrompt },
+    });
+    console.log(`  任务 ID：${videoId}`);
+    console.log(`  ${C.yellow}轮询进度...${C.reset}\n`);
+
+    const poll = await pollVideo(videoId);
+    if (poll.status === 'success' && poll.videoUrl) {
+      const { filePath, asset } = await downloadResult(poll.videoUrl, 'video', 'video/mp4', {
+        prompt: finalPrompt, mode: 'multi2video', params: { ratio, resolution, duration },
+      });
+      await showResult(filePath, asset);
+    } else {
+      console.log(`${C.red}生成失败：${poll.error}${C.reset}`);
+    }
+  } catch (e) {
+    console.log(`${C.red}生成失败：${e.message}${C.reset}`);
+  }
+  await pause();
+}
+
+// ═══════════════════ [6] 关键帧动画 ═══════════════════
+
+async function modeKeyframe() {
+  console.log(`\n${C.bold}─────── 关键帧动画 ───────${C.reset}`);
+  console.log(`${C.gray}至少输入首帧和尾帧图片（2-5 张关键帧）${C.reset}`);
+
+  const keyframes = [];
+  const labels = ['首帧', '尾帧', '中间帧1', '中间帧2', '中间帧3'];
+  for (let i = 0; i < 5; i++) {
+    const hint = i < 2 ? `（必填）` : `（可选，留空跳过）`;
+    const p = cleanPath(await ask(`${labels[i]} 图片路径 ${hint}`));
+    if (!p) {
+      if (i < 2) { console.log(`${C.red}首帧和尾帧为必填${C.reset}`); return; }
+      break;
+    }
+    keyframes.push(p);
+  }
+
+  console.log(`${C.green}已选择 ${keyframes.length} 个关键帧${C.reset}`);
+
+  const rawPrompt = await ask('描述过渡和运动', '从首帧平滑过渡到尾帧，人物缓慢转头');
+  if (!rawPrompt) { console.log(`${C.yellow}已取消${C.reset}`); return; }
+
+  const finalPrompt = await promptOptimizeFlow(rawPrompt, 'keyframe');
+
+  const ratio = await ask('比例 (16:9 / 9:16 / 1:1)', '16:9');
+  const resolution = await ask('分辨率 (720p / 1080p)', '720p');
+  const duration = await ask('时长（秒，5-15）', '5');
+  const negPrompt = await negPromptFlow(finalPrompt);
+
+  console.log(`\n${C.bold}── 生成配置 ──${C.reset}`);
+  console.log(`  关键帧：${keyframes.length} 个`);
+  keyframes.forEach((kf, i) => console.log(`    [${labels[i]}] ${kf}`));
+  console.log(`  提示词：${finalPrompt}`);
+  if (negPrompt) console.log(`  反向：${negPrompt}`);
+  console.log(`  分辨率：${resolution}  比例：${ratio}  时长：${duration}s`);
+
+  if (!(await confirm('开始生成'))) return;
+
+  console.log(`\n${C.yellow}[...] 正在提交关键帧动画任务...${C.reset}`);
+  try {
+    const b64s = await Promise.all(keyframes.map(imgToB64));
+    const { videoId } = await generateVideo({
+      mode: 'keyframe', prompt: finalPrompt, imageUrls: b64s,
+      params: { ratio, resolution, duration, negativePrompt: negPrompt },
+    });
+    console.log(`  任务 ID：${videoId}`);
+    console.log(`  ${C.yellow}轮询进度...${C.reset}\n`);
+
+    const poll = await pollVideo(videoId);
+    if (poll.status === 'success' && poll.videoUrl) {
+      const { filePath, asset } = await downloadResult(poll.videoUrl, 'video', 'video/mp4', {
+        prompt: finalPrompt, mode: 'keyframe', params: { ratio, resolution, duration },
+      });
+      await showResult(filePath, asset);
+    } else {
+      console.log(`${C.red}生成失败：${poll.error}${C.reset}`);
+    }
+  } catch (e) {
+    console.log(`${C.red}生成失败：${e.message}${C.reset}`);
+  }
+  await pause();
+}
+
+// ═══════════════════ [7] 素材库 ═══════════════════
+
+async function menuAssets() {
+  while (true) {
+    console.log(`\n${C.bold}─────── 素材库 ───────${C.reset}`);
+    console.log(`  ${C.cyan}[1]${C.reset} 全部素材`);
+    console.log(`  ${C.cyan}[2]${C.reset} 只看图片`);
+    console.log(`  ${C.cyan}[3]${C.reset} 只看视频`);
+    console.log(`  ${C.cyan}[4]${C.reset} 搜索素材`);
+    console.log(`  ${C.gray}[0]${C.reset} 返回主菜单`);
+    console.log('');
+
+    const c = await ask('选择');
+    let type = 'all', search = '';
+
+    if (c === '0' || c === '') return;
+    if (c === '2') type = 'image';
+    else if (c === '3') type = 'video';
+    else if (c === '4') search = await ask('搜索关键词');
+    else if (c !== '1') { console.log(`${C.red}无效选择${C.reset}`); continue; }
+
+    const { items, total } = await assetStore.listAssets({ type, search, pageSize: 20 });
+
+    if (items.length === 0) {
+      console.log(`\n${C.gray}暂无素材${total > 0 ? `（共 ${total} 条，当前页为空）` : ''}${C.reset}`);
+      continue;
+    }
+
+    console.log(`\n${C.bold}共 ${total} 条素材：${C.reset}\n`);
+    items.forEach((a, i) => {
+      const icon = a.type === 'video' ? '🎬' : '🖼';
+      const prompt = (a.prompt || '').slice(0, 50);
+      const note = a.note ? ` | ${a.note}` : '';
+      console.log(`  ${C.cyan}${String(i + 1).padStart(2)}.${C.reset} ${icon} ${a.filename}`);
+      console.log(`       ${C.gray}${prompt}${note}${C.reset}`);
+      console.log(`       ${formatSize(a.size)} | ${fmtDate(a.createdAt)} | ${a.kept ? C.green + '保留' + C.reset : C.gray + '未保留' + C.reset}`);
+    });
+
+    // 操作
+    console.log('');
+    const op = await ask('输入序号查看详情/操作，或回车返回');
+    if (!op) continue;
+    const idx = parseInt(op, 10) - 1;
+    if (isNaN(idx) || idx < 0 || idx >= items.length) { console.log(`${C.red}无效序号${C.reset}`); continue; }
+
+    const asset = items[idx];
+    const absPath = path.join(config.dirs.root, asset.path);
+    console.log(`\n${C.bold}素材详情：${C.reset}`);
+    console.log(`  ID：${asset.id}`);
+    console.log(`  文件：${absPath}`);
+    console.log(`  提示词：${asset.prompt || '(无)'}`);
+    console.log(`  模式：${asset.mode || '(无)'}`);
+    console.log(`  大小：${formatSize(asset.size)}`);
+    console.log(`  创建：${fmtDate(asset.createdAt)}`);
+    console.log(`  保留：${asset.kept ? '是' : '否'}`);
+    if (asset.note) console.log(`  备注：${asset.note}`);
+
+    const action = await choose('操作', ['打开文件', '切换保留状态', '添加备注', '删除']);
+    if (action === 0) {
+      openFile(absPath);
+    } else if (action === 1) {
+      await assetStore.updateAsset(asset.id, { kept: !asset.kept });
+      console.log(`${C.green}已${asset.kept ? '取消保留' : '标记保留'}${C.reset}`);
+    } else if (action === 2) {
+      const note = await ask('输入备注');
+      if (note) {
+        await assetStore.updateAsset(asset.id, { note });
+        console.log(`${C.green}备注已保存${C.reset}`);
+      }
+    } else if (action === 3) {
+      if (await confirm(`确认删除 ${asset.filename}`)) {
+        await assetStore.deleteAsset(asset.id);
+        console.log(`${C.green}已删除${C.reset}`);
+      }
+    }
+  }
+}
+
+// ═══════════════════ [8] 查看配置 ═══════════════════
+
+async function menuConfig() {
+  console.log(`\n${C.bold}─────── 当前配置 ───────${C.reset}\n`);
+
+  const mask = (s) => s ? s.slice(0, 10) + '···' + s.slice(-4) : '(未设置)';
+
+  console.log(`${C.cyan}── Agnes 图片 API ──${C.reset}`);
+  console.log(`  base_url：${config.agens.image.baseUrl}`);
+  console.log(`  api_key：${mask(config.agens.image.apiKey)}`);
+  console.log(`  model：${config.agens.image.model}`);
+
+  console.log(`\n${C.cyan}── Agnes 视频 API ──${C.reset}`);
+  console.log(`  base_url：${config.agens.video.baseUrl}`);
+  console.log(`  api_key：${mask(config.agens.video.apiKey)}`);
+  console.log(`  model：${config.agens.video.model}`);
+  console.log(`  query：${config.agens.video.queryBaseUrl}${config.agens.video.queryPath}`);
+
+  console.log(`\n${C.cyan}── MiMo 大模型 ──${C.reset}`);
+  console.log(`  base_url：${config.mimo.baseUrl}`);
+  console.log(`  api_key：${mask(config.mimo.apiKey)}`);
+  console.log(`  text model：${config.mimo.model}`);
+  console.log(`  vision model：${config.mimo.visionModel}`);
+
+  console.log(`\n${C.cyan}── 轮询配置 ──${C.reset}`);
+  console.log(`  间隔：${config.agens.pollInterval / 1000}s`);
+  console.log(`  超时：~${Math.round(config.agens.pollMaxAttempts * config.agens.pollInterval / 60000)} 分钟`);
+
+  console.log(`\n${C.cyan}── 目录 ──${C.reset}`);
+  console.log(`  项目根目录：${config.dirs.root}`);
+  console.log(`  图片素材：${config.dirs.images}`);
+  console.log(`  视频素材：${config.dirs.videos}`);
+  console.log(`  数据目录：${config.dirs.data}`);
+
+  console.log(`\n${C.cyan}── 提示词记忆 ──${C.reset}`);
+  try {
+    const s = await promptMemory.stats();
+    console.log(`  总样本：${s.total}`);
+    console.log(`  正向：${s.positive.total}（点赞 ${s.positive.liked}，采纳 ${s.positive.adopted}）`);
+    console.log(`  反向：${s.negative.total}（点赞 ${s.negative.liked}，采纳 ${s.negative.adopted}）`);
+    if (s.prefs?.positive?.favoriteStyle) {
+      console.log(`  正向偏好风格：${s.prefs.positive.favoriteStyle}`);
+    }
+  } catch {
+    console.log(`${C.gray}  (暂无数据)${C.reset}`);
+  }
+
+  await pause();
+}
+
+// ═══════════════════ 入口 ═══════════════════
+
+async function main() {
+  // Windows 终端 UTF-8 提示
+  if (process.platform === 'win32') {
+    console.log(`${C.gray}提示：如果中文显示异常，请在 cmd 中先执行 chcp 65001${C.reset}`);
+  }
+
+  // 首次运行检测：
+  // - 有 ~/.agens-cli/config.json → 用户已配置过，跳过
+  // - 有 config.local.json → 开发者模式，跳过
+  // - 都没有 + 代码里没硬编码 key → 分发版新用户，显示 wizard
+  const hasUserConfig = !!readUserConfig();
+  const hasDevConfig = !!config._configSource;
+  const hasHardcodedKeys = !!(config.agens.image.apiKey && config.agens.image.apiKey.length > 10);
+  const isFirstRun = !hasUserConfig && !hasDevConfig && !hasHardcodedKeys;
+
+  if (isFirstRun) {
+    await setupWizard(true);
+    return; // setupWizard 会 process.exit(0) 提示重启
+  }
+
+  await init();
+  console.log(`\n${C.green}${C.bold}Agens 创作工作台 CLI 已启动${C.reset}`);
+  const srcHint = config._configSource ? `（配置来源：${config._configSource}）` : '';
+  console.log(`${C.gray}API 状态：${config.agens.image.apiKey ? '已配置' : '未配置'}${srcHint}${C.reset}\n`);
+  await mainMenu();
+}
+
+main().catch((e) => {
+  console.error(`${C.red}启动失败：${e.message}${C.reset}`);
+  console.error(e.stack);
+  process.exit(1);
+});