agens-studio 0.1.6 → 0.1.7

package/cli.js

@@ -26,6 +26,8 @@ import { optimizePrompt, optimizeNegativePrompt, dimensionsToText } from './src/
 import { describeImage, descriptionToText } from './src/services/mimoVision.js';
 import * as assetStore from './src/services/assetStore.js';
 import * as promptMemory from './src/services/promptMemory.js';
+import * as petStore from './src/services/petStore.js';
+import * as petArt from './src/services/petArt.js';
 
 // ═══════════════════ ANSI 色彩 ═══════════════════
 
@@ -170,11 +172,12 @@ async function promptOptimizeFlow(prompt, mode, imageDesc) {
   const wantOpt = await confirm('是否用 AI 优化提示词');
   if (!wantOpt) return prompt;
 
-  console.log(`${C.yellow}[...] 正在优化提示词（MiMo ${config.mimo.model}）...${C.reset}`);
+  const stopSpin = startPetSpinner(`正在优化提示词（MiMo ${config.mimo.model}）...`);
   try {
     const result = await optimizePrompt({
       text: prompt, mode, imageDescription: imageDesc,
     });
+    stopSpin();
     const optText = result.dimensions
       ? dimensionsToText(result.dimensions)
       : result.optimized;
@@ -191,6 +194,7 @@ async function promptOptimizeFlow(prompt, mode, imageDesc) {
     const use = await confirm('使用优化后的提示词');
     return use ? optText : prompt;
   } catch (e) {
+    stopSpin();
     console.log(`${C.red}优化失败：${e.message}${C.reset}`);
     return prompt;
   }
@@ -206,12 +210,14 @@ async function negPromptFlow(positive) {
   if (!wantNeg) return '';
   const negInput = await ask('描述不想出现的内容', '模糊, 低质量, 变形, 多余手指');
   if (!negInput) return '';
-  console.log(`${C.yellow}[...] 优化反向提示词...${C.reset}`);
+  const stopSpin = startPetSpinner('优化反向提示词...');
   try {
     const result = await optimizeNegativePrompt({ text: negInput, mode: 'image', positive });
+    stopSpin();
     console.log(`${C.green}反向提示词：${C.reset}${result.optimized || result}`);
     return result.optimized || result;
   } catch (e) {
+    stopSpin();
     console.log(`${C.red}反向优化失败：${e.message}${C.reset}`);
     return negInput;
   }
@@ -225,31 +231,60 @@ async function negPromptFlow(positive) {
 async function pollVideo(videoId) {
   const maxAttempts = config.agens.pollMaxAttempts || 120;
   const interval = config.agens.pollInterval || 5000;
+  const isTTY = !!process.stdout.isTTY;
+
+  // 单行 \r 重绘（不依赖多行光标回退，不会因为文案换行而错位堆叠）。
+  // 网络轮询（每 pollInterval 一次）和动画帧率（120ms 一次）完全独立，
+  // 轮询回调只更新 barText。颜文字每 6 帧（约 720ms）轮播一次，比旋转
+  // 指示符慢一档，制造"正在想"的层次感，同时不会显得单调。
+  const tier = petStore.getTierNow();
+  let spinI = 0;
+  let barText = `${makeBar(0, 30)}   0%`;
+  let animTimer = null;
+
+  const draw = () => {
+    const spin = petArt.SPINNER_FRAMES[spinI % petArt.SPINNER_FRAMES.length];
+    const face = petArt.faceAt(tier, Math.floor(spinI / 6));
+    process.stdout.write(`\r  ${face} ${C.yellow}${spin}${C.reset} ${barText}\x1b[K`);
+    spinI++;
+  };
+
+  if (isTTY) {
+    draw();
+    animTimer = setInterval(draw, 120);
+  } else {
+    console.log(`${C.yellow}[...] 正在生成视频，请稍候...${C.reset}`);
+  }
 
-  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 stop = () => {
+    if (animTimer) clearInterval(animTimer);
+    if (isTTY) process.stdout.write('\r\x1b[K');
+  };
 
-    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)}%  `);
+  try {
+    for (let i = 0; i < maxAttempts; i++) {
+      await sleep(interval);
+      let info;
+      try {
+        info = await getTaskStatus(videoId);
+      } catch {
+        continue;
+      }
 
-    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 || '生成失败' };
+      const pct = info.progress || Math.min(95, 5 + i * 3);
+      barText = `${makeBar(pct, 30)} ${String(Math.round(pct)).padStart(3)}%`;
+
+      if (info.status === 'success') {
+        return { status: 'success', videoUrl: info.result?.videoUrl };
+      }
+      if (info.status === 'failed') {
+        return { status: 'failed', error: info.result?.error || '生成失败' };
+      }
     }
+    return { status: 'failed', error: '轮询超时（约 10 分钟）' };
+  } finally {
+    stop();
   }
-  return { status: 'failed', error: '轮询超时（约 10 分钟）' };
 }
 
 function makeBar(pct, w) {
@@ -257,6 +292,60 @@ function makeBar(pct, w) {
   return `${C.green}${'█'.repeat(filled)}${C.gray}${'░'.repeat(w - filled)}${C.reset}`;
 }
 
+/**
+ * 等待期间的颜文字动画（类似 Claude Code 的"思考中"效果）。
+ * 单行 \r 重绘（和视频进度条同一种模式），不依赖多行光标回退，
+ * 不会因为文案换行而错位堆叠。
+ * - TTY 下：旋转指示符每 120ms 切换一次，颜文字每 6 帧（约 720ms）
+ *   轮播一次同档位的另一个表情，两个节奏叠在一起，不会显得单调
+ * - 非 TTY（管道/重定向）：退化为原来的静态文字提示，不写转义序列
+ * @param {string} caption 等待文案，如「正在生成图片...」
+ * @returns {() => void} stop 函数，调用后清空该行
+ */
+function startPetSpinner(caption) {
+  if (!process.stdout.isTTY) {
+    console.log(`${C.yellow}[...] ${caption}${C.reset}`);
+    return () => {};
+  }
+
+  const tier = petStore.getTierNow();
+  let i = 0;
+  let stopped = false;
+
+  const draw = () => {
+    const spin = petArt.SPINNER_FRAMES[i % petArt.SPINNER_FRAMES.length];
+    const face = petArt.faceAt(tier, Math.floor(i / 6));
+    process.stdout.write(`\r  ${face} ${C.yellow}${spin} ${caption}${C.reset}\x1b[K`);
+    i++;
+  };
+
+  draw();
+  const timer = setInterval(draw, 120);
+
+  // stop() 可能在成功/失败两条路径上都被调用（如 try 里调用一次，catch 里再兜底调用一次），
+  // 用 stopped 保证只清一次，避免第二次调用把 stop 之后新打印的内容误删。
+  return () => {
+    if (stopped) return;
+    stopped = true;
+    clearInterval(timer);
+    process.stdout.write('\r\x1b[K');
+  };
+}
+
+/**
+ * 根据 asset.type 用 config.dirs.images/videos 重建绝对路径。
+ * 不能用 path.join(config.dirs.root, asset.path)：那个写法假设 root 和
+ * images/videos 同根，一旦用户自定义了 storage_dir（素材目录可以指到
+ * 别的盘/目录），root 和 images/videos 就分家了，会算出错误路径。
+ * @param {{type: string, path: string}} asset
+ * @returns {string} 绝对路径
+ */
+function resolveAssetPath(asset) {
+  const subdir = asset.type === 'video' ? 'videos' : 'images';
+  const filename = path.basename(asset.path);
+  return path.join(config.dirs[subdir], filename);
+}
+
 /**
  * 下载远程文件到本地 assets
  * - 支持 data: URI 直接写入（无需网络请求）
@@ -290,7 +379,7 @@ async function downloadResult(url, kind, mime, meta) {
 
   try {
     const asset = await assetStore.downloadAndRegister({ url, kind, mime, meta });
-    const absPath = path.join(config.dirs.root, asset.path);
+    const absPath = resolveAssetPath(asset);
     return { filePath: absPath, asset };
   } catch (dlErr) {
     // 下载失败：尝试直接 fetch 带超时
@@ -336,6 +425,12 @@ async function showResult(filePath, asset, w, h) {
   console.log(`  文件：${filePath}`);
   console.log(`  大小：${formatSize(asset.size)}`);
   if (w && h) console.log(`  尺寸：${w} x ${h}`);
+
+  try {
+    const react = await petStore.onGenerateSuccess(asset.type);
+    if (react?.line) console.log(`  ${C.cyan}${react.line}${C.reset}`);
+  } catch { /* 宠物状态更新失败不影响主流程 */ }
+
   console.log('');
   if (await confirm('打开文件')) {
     openFile(filePath);
@@ -403,6 +498,13 @@ async function setupWizard(isFirstRun) {
   console.log(`\n${C.green}${C.bold}配置已保存到 ${USER_CONFIG_FILE}${C.reset}`);
 
   if (isFirstRun) {
+    console.log(`\n${C.cyan}── 你的专属创作机器人已激活 ──${C.reset}`);
+    console.log(`${C.gray}它会陪着你一起创作，每次生成成功都会给它充电～${C.reset}`);
+    const petName = await ask('给它起个名字', '阿吉');
+    await petStore.init();
+    await petStore.setName(petName);
+    console.log(`${C.green}[^_^] ${petName} 已就位！${C.reset}`);
+
     console.log(`\n${C.yellow}请重新启动 CLI 以加载新配置：${C.reset}`);
     console.log(`  ${C.cyan}node cli.js${C.reset}`);
     process.exit(0);
@@ -412,6 +514,117 @@ async function setupWizard(isFirstRun) {
   await pause();
 }
 
+// ═══════════════════ [9] 修改参数 ═══════════════════
+
+/**
+ * 修改参数子菜单：挑一项单独改，而不是像 setupWizard 那样每次从头问一遍。
+ * 复用现有的 readUserConfig/saveUserConfig 存到 ~/.agens-cli/config.json，
+ * 大部分字段需要重启 CLI 才能生效；机器人名字例外（走 pet.json，立即生效）。
+ */
+async function menuParams() {
+  while (true) {
+    console.log(`\n${C.bold}─────── 修改参数 ───────${C.reset}`);
+    console.log(`  ${C.cyan}[1]${C.reset} Agnes API Key`);
+    console.log(`  ${C.cyan}[2]${C.reset} Agnes 图片（base_url / model）`);
+    console.log(`  ${C.cyan}[3]${C.reset} Agnes 视频（base_url / model）`);
+    console.log(`  ${C.cyan}[4]${C.reset} MiMo API Key`);
+    console.log(`  ${C.cyan}[5]${C.reset} MiMo（base_url / text model / vision model）`);
+    console.log(`  ${C.cyan}[6]${C.reset} 存储目录（图片/视频/数据保存位置）`);
+    console.log(`  ${C.cyan}[7]${C.reset} 机器人名字`);
+    console.log(`  ${C.gray}[0]${C.reset} 返回主菜单`);
+    console.log('');
+
+    const c = await ask('选择要修改的参数');
+    if (c === '0' || c === '') return;
+
+    const cfg = readUserConfig() || {};
+    let saved = false;
+    let needRestart = true;
+
+    switch (c) {
+      case '1': {
+        console.log(`\n${C.cyan}── Agnes API Key ──${C.reset}`);
+        console.log(`${C.gray}图片和视频生成共用同一个 key${C.reset}`);
+        const v = await ask('Agnes API Key', cfg.agens_api_key || '');
+        if (v) { cfg.agens_api_key = v; saveUserConfig(cfg); saved = true; }
+        break;
+      }
+      case '2': {
+        console.log(`\n${C.cyan}── Agnes 图片 ──${C.reset}`);
+        cfg.agens_image_base_url = await ask('base_url', cfg.agens_image_base_url || config.agens.image.baseUrl);
+        cfg.agens_image_model = await ask('model', cfg.agens_image_model || config.agens.image.model);
+        saveUserConfig(cfg);
+        saved = true;
+        break;
+      }
+      case '3': {
+        console.log(`\n${C.cyan}── Agnes 视频 ──${C.reset}`);
+        cfg.agens_video_base_url = await ask('base_url', cfg.agens_video_base_url || config.agens.video.baseUrl);
+        cfg.agens_video_model = await ask('model', cfg.agens_video_model || config.agens.video.model);
+        saveUserConfig(cfg);
+        saved = true;
+        break;
+      }
+      case '4': {
+        console.log(`\n${C.cyan}── MiMo API Key ──${C.reset}`);
+        const v = await ask('MiMo API Key (tp- 开头)', cfg.mimo_api_key || '');
+        if (v) { cfg.mimo_api_key = v; saveUserConfig(cfg); saved = true; }
+        break;
+      }
+      case '5': {
+        console.log(`\n${C.cyan}── MiMo ──${C.reset}`);
+        cfg.mimo_base_url = await ask('base_url', cfg.mimo_base_url || config.mimo.baseUrl);
+        cfg.mimo_model = await ask('text model', cfg.mimo_model || config.mimo.model);
+        cfg.mimo_vision_model = await ask('vision model', cfg.mimo_vision_model || config.mimo.visionModel);
+        saveUserConfig(cfg);
+        saved = true;
+        break;
+      }
+      case '6': {
+        console.log(`\n${C.cyan}── 存储目录 ──${C.reset}`);
+        console.log(`${C.gray}当前：${config.dirs.storageRoot}${C.reset}`);
+        console.log(`${C.gray}只影响以后新生成的文件存到哪，已有素材不会自动搬家${C.reset}`);
+        const dir = await ask('新的存储目录（绝对路径）', config.dirs.storageRoot);
+        if (dir && path.resolve(dir) !== config.dirs.storageRoot) {
+          const resolved = path.resolve(dir);
+          try {
+            await fs.mkdir(resolved, { recursive: true });
+            cfg.storage_dir = resolved;
+            saveUserConfig(cfg);
+            saved = true;
+          } catch (e) {
+            console.log(`${C.red}目录无效或无法创建：${e.message}${C.reset}`);
+          }
+        }
+        break;
+      }
+      case '7': {
+        console.log(`\n${C.cyan}── 机器人名字 ──${C.reset}`);
+        const pet = petStore.getState();
+        const name = await ask('给它起个新名字', pet?.name || '阿吉');
+        if (name) {
+          await petStore.setName(name);
+          console.log(`${C.green}好，以后就叫 ${name} 了～${C.reset}`);
+          saved = true;
+          needRestart = false;
+        }
+        break;
+      }
+      default:
+        console.log(`${C.red}无效选择${C.reset}`);
+        continue;
+    }
+
+    if (saved) {
+      console.log(`${C.green}已保存${C.reset}`);
+      if (needRestart) {
+        console.log(`${C.gray}提示：这项改动需要重启 CLI 才能生效${C.reset}`);
+      }
+    }
+    await pause();
+  }
+}
+
 // ═══════════════════ 初始化 ═══════════════════
 
 async function init() {
@@ -434,6 +647,8 @@ async function mainMenu() {
     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}`);
+    const petLine = petStore.renderBanner();
+    if (petLine) console.log(petLine);
     console.log('');
     console.log(`  ${C.cyan}[1]${C.reset} 文生图`);
     console.log(`  ${C.cyan}[2]${C.reset} 图生图`);
@@ -443,7 +658,7 @@ async function mainMenu() {
     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.cyan}[9]${C.reset} 修改参数`);
     console.log(`  ${C.gray}[0]${C.reset} 退出`);
     console.log('');
 
@@ -458,7 +673,7 @@ async function mainMenu() {
       case '6': await modeKeyframe(); break;
       case '7': await menuAssets(); break;
       case '8': await menuConfig(); break;
-      case '9': await setupWizard(false); break;
+      case '9': await menuParams(); break;
       case '0':
         console.log(`\n${C.green}再见！${C.reset}\n`);
         process.exit(0);
@@ -508,17 +723,20 @@ async function modeText2Image() {
 
   if (!(await confirm('开始生成'))) return;
 
-  console.log(`\n${C.yellow}[...] 正在生成图片（Agnes ${config.agens.image.model}）...${C.reset}`);
+  console.log('');
+  const stopGen = startPetSpinner(`正在生成图片（Agnes ${config.agens.image.model}）...`);
   try {
     const { url } = await generateImage({
       prompt: finalPrompt,
       params: { ratio, style, negativePrompt: negPrompt },
     });
+    stopGen();
     const { filePath, asset } = await downloadResult(url, 'image', 'image/png', {
       prompt: finalPrompt, mode: 'image', params: { ratio, style },
     });
     await showResult(filePath, asset);
   } catch (e) {
+    stopGen();
     console.log(`${C.red}生成失败：${e.message}${C.reset}`);
   }
   await pause();
@@ -535,15 +753,17 @@ async function modeImage2Image() {
   if (!imgPath) { console.log(`${C.yellow}已取消${C.reset}`); return; }
 
   // 识图
-  console.log(`${C.yellow}[...] 正在识别图片内容（MiMo Vision）...${C.reset}`);
+  const stopVision = startPetSpinner('正在识别图片内容（MiMo Vision）...');
   let imageDesc = '';
   try {
     const dataUri = await imgToVisionUri(imgPath);
     // 直接调用 mimo vision 获取描述
     const descResult = await callVisionDirect(dataUri);
+    stopVision();
     imageDesc = descResult;
     console.log(`${C.green}识图结果：${C.reset}${descResult.slice(0, 100)}...`);
   } catch (e) {
+    stopVision();
     console.log(`${C.red}识图失败：${e.message}${C.reset}`);
     console.log(`${C.gray}将继续使用（提示词优化可能不够精准）${C.reset}`);
   }
@@ -563,18 +783,21 @@ async function modeImage2Image() {
 
   if (!(await confirm('开始生成'))) return;
 
-  console.log(`\n${C.yellow}[...] 正在生成图片...${C.reset}`);
+  console.log('');
+  const stopGen = startPetSpinner('正在生成图片...');
   try {
     const dataUri = await imgToDataUri(imgPath);
     const { url } = await generateImage({
       prompt: finalPrompt,
       params: { ratio, style: '', negativePrompt: negPrompt, imageUrls: [dataUri] },
     });
+    stopGen();
     const { filePath, asset } = await downloadResult(url, 'image', 'image/png', {
       prompt: finalPrompt, mode: 'img2img', params: { ratio },
     });
     await showResult(filePath, asset);
   } catch (e) {
+    stopGen();
     console.log(`${C.red}生成失败：${e.message}${C.reset}`);
   }
   await pause();
@@ -635,12 +858,14 @@ async function modeText2Video() {
 
   if (!(await confirm('开始生成'))) return;
 
-  console.log(`\n${C.yellow}[...] 正在提交视频任务（Agnes ${config.agens.video.model}）...${C.reset}`);
+  console.log('');
+  const stopSubmit = startPetSpinner(`正在提交视频任务（Agnes ${config.agens.video.model}）...`);
   try {
     const { videoId } = await generateVideo({
       mode: 'text2video', prompt: finalPrompt, imageUrls: [],
       params: { ratio, resolution, duration, negativePrompt: negPrompt },
     });
+    stopSubmit();
     console.log(`  任务 ID：${videoId}`);
     console.log(`  ${C.yellow}轮询进度（每 ${config.agens.pollInterval / 1000}s 刷新）...${C.reset}\n`);
 
@@ -654,6 +879,7 @@ async function modeText2Video() {
       console.log(`${C.red}生成失败：${poll.error}${C.reset}`);
     }
   } catch (e) {
+    stopSubmit();
     console.log(`${C.red}生成失败：${e.message}${C.reset}`);
   }
   await pause();
@@ -687,13 +913,15 @@ async function modeImage2Video() {
 
   if (!(await confirm('开始生成'))) return;
 
-  console.log(`\n${C.yellow}[...] 正在提交视频任务...${C.reset}`);
+  console.log('');
+  const stopSubmit = startPetSpinner('正在提交视频任务...');
   try {
     const b64 = await imgToB64(imgPath);
     const { videoId } = await generateVideo({
       mode: 'image2video', prompt: finalPrompt, imageUrls: [b64],
       params: { ratio, resolution, duration, negativePrompt: negPrompt },
     });
+    stopSubmit();
     console.log(`  任务 ID：${videoId}`);
     console.log(`  ${C.yellow}轮询进度...${C.reset}\n`);
 
@@ -707,6 +935,7 @@ async function modeImage2Video() {
       console.log(`${C.red}生成失败：${poll.error}${C.reset}`);
     }
   } catch (e) {
+    stopSubmit();
     console.log(`${C.red}生成失败：${e.message}${C.reset}`);
   }
   await pause();
@@ -750,13 +979,15 @@ async function modeMulti2Video() {
 
   if (!(await confirm('开始生成'))) return;
 
-  console.log(`\n${C.yellow}[...] 正在提交视频任务...${C.reset}`);
+  console.log('');
+  const stopSubmit = startPetSpinner('正在提交视频任务...');
   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 },
     });
+    stopSubmit();
     console.log(`  任务 ID：${videoId}`);
     console.log(`  ${C.yellow}轮询进度...${C.reset}\n`);
 
@@ -770,6 +1001,7 @@ async function modeMulti2Video() {
       console.log(`${C.red}生成失败：${poll.error}${C.reset}`);
     }
   } catch (e) {
+    stopSubmit();
     console.log(`${C.red}生成失败：${e.message}${C.reset}`);
   }
   await pause();
@@ -815,13 +1047,15 @@ async function modeKeyframe() {
 
   if (!(await confirm('开始生成'))) return;
 
-  console.log(`\n${C.yellow}[...] 正在提交关键帧动画任务...${C.reset}`);
+  console.log('');
+  const stopSubmit = startPetSpinner('正在提交关键帧动画任务...');
   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 },
     });
+    stopSubmit();
     console.log(`  任务 ID：${videoId}`);
     console.log(`  ${C.yellow}轮询进度...${C.reset}\n`);
 
@@ -835,6 +1069,7 @@ async function modeKeyframe() {
       console.log(`${C.red}生成失败：${poll.error}${C.reset}`);
     }
   } catch (e) {
+    stopSubmit();
     console.log(`${C.red}生成失败：${e.message}${C.reset}`);
   }
   await pause();
@@ -916,7 +1151,7 @@ async function menuAssets() {
 }
 
 async function showAssetDetail(asset) {
-  const absPath = path.join(config.dirs.root, asset.path);
+  const absPath = resolveAssetPath(asset);
   console.log(`\n${C.bold}素材详情：${C.reset}`);
   console.log(`  ID：${asset.id}`);
   console.log(`  文件：${absPath}`);
@@ -977,6 +1212,7 @@ async function menuConfig() {
 
   console.log(`\n${C.cyan}── 目录 ──${C.reset}`);
   console.log(`  项目根目录：${config.dirs.root}`);
+  console.log(`  素材存储目录：${config.dirs.storageRoot}${config.dirs.storageRoot === config.dirs.root ? C.gray + '（默认，未自定义）' + C.reset : ''}`);
   console.log(`  图片素材：${config.dirs.images}`);
   console.log(`  视频素材：${config.dirs.videos}`);
   console.log(`  数据目录：${config.dirs.data}`);
@@ -994,6 +1230,16 @@ async function menuConfig() {
     console.log(`${C.gray}  (暂无数据)${C.reset}`);
   }
 
+  console.log(`\n${C.cyan}── 我的机器人 ──${C.reset}`);
+  const pet = petStore.getState();
+  if (pet) {
+    console.log(`  名字：${pet.name}`);
+    console.log(`  电量：${pet.energy}%（${pet.statusLabel}）`);
+    console.log(`  累计陪伴生成：${pet.totalGenerated} 次`);
+  } else {
+    console.log(`${C.gray}  (暂无数据)${C.reset}`);
+  }
+
   await pause();
 }
 
@@ -1023,6 +1269,10 @@ async function main() {
   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`);
+
+  const greet = await petStore.init();
+  if (greet) console.log(`${C.cyan}${greet}${C.reset}\n`);
+
   await mainMenu();
 }
 

package/config.js

@@ -92,16 +92,23 @@ export const config = {
   },
 
   // 目录
-  dirs: {
-    root: __dirname,
-    public: path.join(__dirname, 'public'),
-    assets: path.join(__dirname, 'assets'),
-    images: path.join(__dirname, 'assets', 'images'),
-    videos: path.join(__dirname, 'assets', 'videos'),
-    data: path.join(__dirname, 'data'),
-    indexFile: path.join(__dirname, 'data', 'index.json'),
-    uploads: path.join(__dirname, 'data', 'uploads'),
-  },
+  // storageRoot：素材/数据实际存放的根目录，支持通过 storage_dir 自定义
+  // （默认等于项目代码目录 __dirname，和 root 相同；一旦自定义，root 和
+  // storageRoot 会分家 —— root 始终是代码位置，storageRoot 是素材位置）
+  dirs: (() => {
+    const storageRoot = ext_cfg.storage_dir ? path.resolve(ext_cfg.storage_dir) : __dirname;
+    return {
+      root: __dirname,
+      storageRoot,
+      public: path.join(__dirname, 'public'),
+      assets: path.join(storageRoot, 'assets'),
+      images: path.join(storageRoot, 'assets', 'images'),
+      videos: path.join(storageRoot, 'assets', 'videos'),
+      data: path.join(storageRoot, 'data'),
+      indexFile: path.join(storageRoot, 'data', 'index.json'),
+      uploads: path.join(storageRoot, 'data', 'uploads'),
+    };
+  })(),
 
   // 上传参考图限制
   upload: {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agens-studio",
-  "version": "0.1.6",
+  "version": "0.1.7",
   "description": "Agens 创作工作台 —— AI 图片/视频生成 CLI + Web 工作台",
   "type": "module",
   "main": "server.js",

package/src/services/petArt.js

@@ -0,0 +1,79 @@
+/**
+ * 机器人「阿吉」渲染器（颜文字丰富版）
+ * ------------------------------------------------------------------
+ * 试过多行像素画（8x6 / 14x14 / 9x9 网格），结论是：不管怎么缩小，
+ * "每次回到主菜单都要重新打印一次"的多行画面天生和 CLI 菜单的空间
+ * 预算冲突——缩到还能看出天线/眼睛/手臂/腿，就不可能再小了。
+ *
+ * 改回单行颜文字，但不是回退到单一固定的 [o_o]：每个电量档位配一组
+ * 颜文字，随机/轮播挑选 + 配色，解决"太单调"的问题，同时保持单行
+ * \r 重绘（不会有多行光标回退错位的风险）。
+ * ------------------------------------------------------------------
+ */
+
+const RESET = '\x1b[0m';
+
+// 电量档位 → 前景色
+const COLOR = {
+  full: 214, // 满电：明黄橙
+  ok: 43, // 元气满满：青绿
+  low: 130, // 电量不足：暗橙褐
+  empty: 240, // 快没电了：灰色
+};
+
+const TIER_LABEL = {
+  full: '满电',
+  ok: '元气满满',
+  low: '电量不足',
+  empty: '快没电了',
+};
+
+// 每个档位一组颜文字，避免总是同一个表情
+const KAOMOJI = {
+  full: ['(≧◡≦)', '(★ω★)', '(๑>ᴗ<๑)', '(*≧▽≦)'],
+  ok: ['(´・ω・`)', '(◕‿◕)', '(^‿^)', '( ˘ᴗ˘ )'],
+  low: ['(￣﹃￣)', '(¬_¬)', '(-_-)'],
+  empty: ['(x_x)', '(-.-)zzZ', '(u_u)'],
+};
+
+export function getTier(energy) {
+  if (energy >= 80) return 'full';
+  if (energy >= 50) return 'ok';
+  if (energy >= 20) return 'low';
+  return 'empty';
+}
+
+export function tierLabel(tier) {
+  return TIER_LABEL[tier] || '';
+}
+
+function fg(colorCode) {
+  return `\x1b[38;5;${colorCode}m`;
+}
+
+/**
+ * 随机挑一个该档位的颜文字（配色跟着档位走）。主菜单展示用——
+ * 每次重绘菜单都随机挑一次，避免每次看到的都是同一个表情。
+ * @param {string} tier
+ * @returns {string} 带颜色转义码的颜文字
+ */
+export function pickFace(tier) {
+  const pool = KAOMOJI[tier] || KAOMOJI.ok;
+  const face = pool[Math.floor(Math.random() * pool.length)];
+  return `${fg(COLOR[tier] ?? COLOR.ok)}${face}${RESET}`;
+}
+
+/**
+ * 取该档位第 index 个颜文字（按顺序轮播，供等待动画用）。
+ * @param {string} tier
+ * @param {number} index
+ * @returns {string} 带颜色转义码的颜文字
+ */
+export function faceAt(tier, index) {
+  const pool = KAOMOJI[tier] || KAOMOJI.ok;
+  const face = pool[index % pool.length];
+  return `${fg(COLOR[tier] ?? COLOR.ok)}${face}${RESET}`;
+}
+
+/** 等待动画用的旋转指示符（经典 braille spinner，逐帧切换） */
+export const SPINNER_FRAMES = ['⠋', '⠙', '⠹', '⠸', '⠼', '⠴', '⠦', '⠧', '⠇', '⠏'];

package/src/services/petStore.js

@@ -0,0 +1,141 @@
+import fs from 'node:fs/promises';
+import fsSync from 'node:fs';
+import path from 'node:path';
+import { USER_CONFIG_DIR } from '../../config.js';
+import { getTier, tierLabel, pickFace } from './petArt.js';
+
+/**
+ * 机器人宠物「阿吉」
+ * ------------------------------------------------------------------
+ * 轻量陪伴：靠你的创作「充电」——生成成功涨电量，长期不打开 CLI 会掉电量。
+ * 纯 ASCII 颜文字展示，不用 emoji，避免不同终端/字体下宽度错位。
+ *
+ * 数据结构：~/.agens-cli/pet.json
+ * {
+ *   name, energy(0-100), totalGenerated, createdAt, lastSeenAt
+ * }
+ * ------------------------------------------------------------------
+ */
+
+const PET_FILE = path.join(USER_CONFIG_DIR, 'pet.json');
+
+const DEFAULT_NAME = '阿吉';
+const MAX_ENERGY = 100;
+const DECAY_PER_DAY = 5; // 每隔一天没打开 CLI，掉多少电量
+const GAIN_PER_SUCCESS = 6; // 每次成功生成，涨多少电量
+
+const SUCCESS_LINES = [
+  '叮～能量 +{gain}，创作真快乐！',
+  '又完成一件作品，我悄悄充了点电～',
+  '嗡嗡嗡，电路都被这波灵感点亮了！',
+  '不错不错，我的电量条又长长了一截。',
+  '创作即燃料，我现在浑身是电！',
+];
+
+const WELCOME_BACK_LINES = [
+  '你好久没来啦，电量都快耗尽了，快来创作充充电吧！',
+  '滴……检测到你离开了 {days} 天，我有点想你了。',
+  '电量掉了不少，赶紧生成点作品给我充充电～',
+];
+
+let _state = null;
+
+function nowTs() { return Date.now(); }
+
+function defaultState() {
+  const ts = nowTs();
+  return {
+    name: DEFAULT_NAME,
+    energy: 70,
+    totalGenerated: 0,
+    createdAt: ts,
+    lastSeenAt: ts,
+  };
+}
+
+function pick(pool) { return pool[Math.floor(Math.random() * pool.length)]; }
+
+async function save() {
+  if (!_state) return;
+  await fs.mkdir(USER_CONFIG_DIR, { recursive: true });
+  await fs.writeFile(PET_FILE, JSON.stringify(_state, null, 2), 'utf8');
+}
+
+async function load() {
+  try {
+    const raw = await fs.readFile(PET_FILE, 'utf8');
+    const parsed = JSON.parse(raw);
+    if (parsed && typeof parsed === 'object' && parsed.name) return parsed;
+  } catch { /* 不存在或损坏，走默认值 */ }
+  return defaultState();
+}
+
+/**
+ * 启动时调用：加载状态，按离开天数衰减电量，更新 lastSeenAt。
+ * @returns {Promise<string|null>} 若离开 ≥1 天，返回一条「久别重逢」台词；否则 null
+ */
+export async function init() {
+  _state = await load();
+
+  const days = Math.floor((nowTs() - _state.lastSeenAt) / 86400000);
+  let welcomeLine = null;
+  if (days >= 1) {
+    _state.energy = Math.max(0, _state.energy - days * DECAY_PER_DAY);
+    welcomeLine = pick(WELCOME_BACK_LINES).replace('{days}', String(days));
+  }
+  _state.lastSeenAt = nowTs();
+  await save();
+  return welcomeLine;
+}
+
+/** 重命名机器人 */
+export async function setName(name) {
+  if (!_state) await init();
+  const n = (name || '').trim();
+  if (n) _state.name = n;
+  await save();
+}
+
+/**
+ * 一次成功生成后调用：涨电量 + 累计计数
+ * @param {'image'|'video'} type
+ * @returns {Promise<{line: string}>}
+ */
+export async function onGenerateSuccess(type) {
+  if (!_state) await init();
+  _state.energy = Math.min(MAX_ENERGY, _state.energy + GAIN_PER_SUCCESS);
+  _state.totalGenerated += 1;
+  await save();
+  return { line: pick(SUCCESS_LINES).replace('{gain}', String(GAIN_PER_SUCCESS)) };
+}
+
+/** 简易电量条 */
+function energyBar(energy, w = 10) {
+  const filled = Math.round((w * energy) / 100);
+  return '[' + '#'.repeat(filled) + '.'.repeat(w - filled) + ']';
+}
+
+/**
+ * 主菜单展示用：单行颜文字 + 名字/电量条/状态词。
+ * 每次调用随机挑一个该档位的颜文字，同样是"元气满满"也不会总看到
+ * 同一张脸，缓解单行方案容易显得单调的问题。
+ * @returns {string} 单行文本，直接 console.log 即可
+ */
+export function renderBanner() {
+  if (!_state) return '';
+  const tier = getTier(_state.energy);
+  const face = pickFace(tier);
+  return `  ${face}  ${_state.name}  ${energyBar(_state.energy)} ${_state.energy}%  ${tierLabel(tier)}`;
+}
+
+/** 配置页展示用的详细状态 */
+export function getState() {
+  if (!_state) return null;
+  return { ..._state, statusLabel: tierLabel(getTier(_state.energy)) };
+}
+
+/** 当前电量档位（供 cli.js 的等待动画取用，避免动画循环里重复算电量） */
+export function getTierNow() {
+  if (!_state) return 'ok';
+  return getTier(_state.energy);
+}