koishi-plugin-p-draw 1.0.0 → 1.2.0

package/index.js

@@ -28,7 +28,11 @@ exports.usage = `
 - **联网搜索：** 描述中带 \`联网\` / \`搜索\` / \`查一下\` 等词时，会先联网搜索补充角色设定（需配置 Tavily Key）。
 - **画师组：** \`创建画师组 名称=tags\` \`切换画师组 名称\` \`查看画师组\` \`删除画师组 名称\`
 - **固定角色：** \`添加角色 名称=tags\`
-- **以图生图：** \`p-draw i2i <描述>\`，并在同一条消息里附一张原图（文件/截图/链接均可），保留原图构图与主体，按描述调整风格与细节（去噪强度由 \`img2imgDenoise\` 控制，默认 0.55）。
+- **以图生图：** \`p-draw i2i <描述>\`，并在同一条消息里附一张原图（文件/截图/链接均可）。发送后会询问处理模式：
+  - **① 换风格（漫画化）**：保留原图构图，把图片转成二次元画风（用 ControlNet 锁结构）
+  - **② 换装换姿势**：保留角色长相，重新设计服装/姿势/场景（用 IP-Adapter 保脸）
+  - **③ 取消**：不生成
+  - 可加 \`--denoise 0.6\` 单独调整强度；未装 ControlNet/IPAdapter 时自动回退普通 img2img。
 `;
 
 const zhCN = {
@@ -74,7 +78,19 @@ const zhCN = {
   multiPrice: { $description: '多人指令（p-draw 多人）单张消耗的 P 点' },
   couponPrice: { $description: '提示词优化券单价（P 点/张，购买询问时显示）' },
   couponAskTimeout: { $description: '提示词优化券确认等待时间（秒）' },
-  img2imgDenoise: { $description: '以图生图（p-draw i2i）的去噪强度，越小越接近原图（建议 0.4-0.7）' },
+  img2imgDenoise: { $description: '普通以图生图（p-draw i2i）的去噪强度，越小越接近原图（建议 0.4-0.7）' },
+  i2iMode: { $description: 'i2i 模式选择方式：ask=每次询问 / style=直接换风格（漫画化）/ ootd=直接换装换姿势 / plain=普通 img2img' },
+  i2iAskTimeout: { $description: 'i2i 模式询问等待时间（秒）' },
+  i2iStyleDenoise: { $description: '换风格模式（漫画化）的去噪强度，越大风格变化越彻底（建议 0.7-0.85）' },
+  i2iOotdDenoise: { $description: '换装换姿势模式（保留角色）的去噪强度（建议 0.5-0.6）' },
+  i2iControlNetStrength: { $description: '换风格模式的 ControlNet 强度，越大构图锁得越死（建议 0.5-0.8）' },
+  i2iIPAdapterPath: { $description: 'Anima IP-Adapter 模型文件路径（换装换姿势模式保脸用；需安装 comfyui-anima-ipadapter 节点并把模型路径填到这里）' },
+  i2iIPAdapterWeight: { $description: '换装换姿势模式的 IP-Adapter 权重，越大角色特征保留越强（建议 0.6-1.0）' },
+  controlNetModel: { $description: 'ControlNet 模型文件名（留空自动检测 Qwen/Anima 系 ControlNet；需放到 ComfyUI/models/controlnet）' },
+  taggerEnabled: { $description: 'i2i 前自动识图（需 ComfyUI 安装 WD14 Tagger 节点与模型；识别出的标签会注入提示词优化）' },
+  taggerModel: { $description: '识图模型名（WD14 Tagger 节点里可选模型）' },
+  taggerThreshold: { $description: '识图标签置信度阈值' },
+  taggerCharacterThreshold: { $description: '识图角色标签置信度阈值' },
   adminUsers: { $description: '免 P 点管理员用户 ID 列表' },
   outputLogs: { $description: '是否在控制台输出详细日志' },
   multiVerifyEnabled: { $description: '多人图生成后启用视觉校验（需配置下方视觉模型）' },
@@ -94,7 +110,7 @@ const zhCN = {
       description: '连接本地 ComfyUI 生图，消耗 P 点',
       messages: {
         'not-permitted': 'ComfyUI 助手已关闭，或当前用户没有使用权限。',
-        usage: 'p-draw 帮助（本指令名可自行更换，如 /anm）：\n\n【生成】\n  p-draw <描述>\n  例：p-draw 一个女孩，白色裙子，立绘，简单背景\n  可加 --seed 数字 固定种子（单张 / 批量 / 连续图均支持）\n\n【多人生成】\n  p-draw 多人 <描述>（2-4 人画面）\n  例：p-draw 多人 左边若叶睦抱着吉他，右边千早爱音牵着她的手\n\n【连续图】\n  p-draw 连续 <角色>：<阶段1> → <阶段2> → ...\n  例：p-draw 连续 少女：清纯校服 → 换上晚礼服 → 华丽登场\n  全阶段共用同一 seed，角色外观尽量一致；可加 --seed 数字 固定种子\n  阶段分隔：→ / -> / |\n\n【批量张数】\n  描述后加 x3 / ×3 / 3张 / 三张 / --数量 3，例：p-draw 一个女孩 x3\n  多张按 张数×单价 一次性扣 P 点，余额不足则不生成\n\n【尺寸】\n  竖图 / 横图 / 方图 / 长竖图 / 宽屏，或 1024x1536：描述 / --尺寸 1216x832\n  例：p-draw 竖图：狐莉站在梨花树下\n\n【原样模式】\n  p-draw 无优化 masterpiece, best quality, 1girl, solo\n\n【提示词优化】\n  配置 LLM（llmBaseUrl/llmModel）后：\n  全局优化开启 → 每次自动把中文描述转成 Danbooru tags\n  全局优化关闭 → 生图时询问是否使用「提示词优化券」（p-shop 购买，每张图扣 1 张）：\n    有券 → 确认后消耗券并优化，拒绝则取消本次生图\n    没券 → 询问是否购买（显示价格），确认后购买并消耗、优化生图；\n          拒绝一次会再次警告，再拒绝则直接生图（不优化）；\n          P 点不足买不起券时，会询问是否仍然生图\n  无优化 <tags> 原样生图，不耗券\n  连续图逐阶段强制优化；多人指令依赖 LLM 规划（未配置会提示）\n\n【画师组】\n  创建画师组 名称=tags / 追加画师组 名称=tags / 切换画师组 名称 / 查看画师组 / 删除画师组 名称\n\n【固定角色】\n  添加角色 名称=tags\n\n【以图生图】\n  p-draw i2i <描述>，并在同一条消息里附一张原图（文件/截图/链接均可）\n  例：p-draw i2i 换成晚礼服，背景换成舞台灯光\n  保留原图构图与主体，按描述调整风格与细节\n\n【模型】\n  p-draw 模型（查看当前与可用模型） / p-draw 模型 名称（切换，支持模糊匹配）/ p-draw 模型 默认（重置）\n  例：p-draw 模型 anima-aesthetic\n\n【状态】\n  p-draw 状态（查看 ComfyUI 连接状态与模型可用性）',
+        usage: 'p-draw 帮助（本指令名可自行更换，如 /anm）：\n\n【生成】\n  p-draw <描述>\n  例：p-draw 一个女孩，白色裙子，立绘，简单背景\n  可加 --seed 数字 固定种子（单张 / 批量 / 连续图均支持）\n\n【多人生成】\n  p-draw 多人 <描述>（2-4 人画面）\n  例：p-draw 多人 左边若叶睦抱着吉他，右边千早爱音牵着她的手\n\n【连续图】\n  p-draw 连续 <角色>：<阶段1> → <阶段2> → ...\n  例：p-draw 连续 少女：清纯校服 → 换上晚礼服 → 华丽登场\n  全阶段共用同一 seed，角色外观尽量一致；可加 --seed 数字 固定种子\n  阶段分隔：→ / -> / |\n\n【批量张数】\n  描述后加 x3 / ×3 / 3张 / 三张 / --数量 3，例：p-draw 一个女孩 x3\n  多张按 张数×单价 一次性扣 P 点，余额不足则不生成\n\n【尺寸】\n  竖图 / 横图 / 方图 / 长竖图 / 宽屏，或 1024x1536：描述 / --尺寸 1216x832\n  例：p-draw 竖图：狐莉站在梨花树下\n\n【原样模式】\n  p-draw 无优化 masterpiece, best quality, 1girl, solo\n\n【提示词优化】\n  配置 LLM（llmBaseUrl/llmModel）后：\n  全局优化开启 → 每次自动把中文描述转成 Danbooru tags\n  全局优化关闭 → 生图时询问是否使用「提示词优化券」（p-shop 购买，每张图扣 1 张）：\n    有券 → 确认后消耗券并优化，拒绝则取消本次生图\n    没券 → 询问是否购买（显示价格），确认后购买并消耗、优化生图；\n          拒绝一次会再次警告，再拒绝则直接生图（不优化）；\n          P 点不足买不起券时，会询问是否仍然生图\n  无优化 <tags> 原样生图，不耗券\n  连续图逐阶段强制优化；多人指令依赖 LLM 规划（未配置会提示）\n\n【画师组】\n  创建画师组 名称=tags / 追加画师组 名称=tags / 切换画师组 名称 / 查看画师组 / 删除画师组 名称\n\n【固定角色】\n  添加角色 名称=tags\n\n【以图生图】\n  p-draw i2i <描述>，并在同一条消息里附一张原图（文件/截图/链接均可）\n  例：p-draw i2i 换成晚礼服，背景换成舞台灯光\n  发送后会询问处理模式（可配置 i2iMode 固定模式跳过询问）：\n    ① 换风格（漫画化）：保留原图构图，转成二次元画风（需 ControlNet）\n    ② 换装换姿势：保留角色长相，重新设计服装/姿势/场景（需 Anima IP-Adapter）\n    ③ 取消：不生成\n  可加 --denoise 0.6 单独调整强度；未装 ControlNet/IPAdapter 时自动回退普通 img2img\n\n【模型】\n  p-draw 模型（查看当前与可用模型） / p-draw 模型 名称（切换，支持模糊匹配）/ p-draw 模型 默认（重置）\n  例：p-draw 模型 anima-aesthetic\n\n【状态】\n  p-draw 状态（查看 ComfyUI 连接状态与模型可用性）',
         'account-notExists': '君现在还没有 p 点，请先签到哦',
         'no-enough-p': '君的 p 点不够 {0}p 哦，先去签个到吧qwq',
         'no-prompt': '请提供画面描述，例如：p-draw 一个女孩，白色裙子',
@@ -151,6 +167,13 @@ const zhCN = {
         'i2i-no-image': 'i2i（以图生图）需要附一张原图。用法：p-draw i2i <描述>，并在同一条消息里带上图片。',
         'i2i-upload-fail': '原图上传 ComfyUI 失败：{0}',
         'i2i-no-custom-workflow': 'i2i（以图生图）暂不支持自定义工作流（customWorkflowEnabled），请关闭后再试。',
+        'i2i-mode-ask': '请选择 i2i 处理模式：\n① 换风格（漫画化）——保留原图构图，转成二次元画风\n② 换装换姿势——保留角色长相，重新设计服装/姿势/场景\n③ 取消——不生成\n回复 1 / 2 / 3 或对应名称即可。',
+        'i2i-mode-invalid': '没有理解你的选择。请回复 ①换风格（漫画化） / ②换装换姿势 / ③取消。',
+        'i2i-mode-cancelled': '已取消本次 i2i 生图。',
+        'i2i-mode-style': '已选择【换风格（漫画化）】：保留原图构图，转成二次元画风。',
+        'i2i-mode-ootd': '已选择【换装换姿势】：保留角色长相，重新设计服装/姿势/场景。',
+        'i2i-no-controlnet': '未检测到可用的 ControlNet 模型，本次「换风格」将使用普通 img2img，构图保留效果会弱一些。',
+        'i2i-no-ipadapter': '未检测到可用的 Anima IP-Adapter，本次「换装换姿势」将使用普通 img2img，角色保留效果会弱一些。',
         'multi-no-llm': '多人指令需要 LLM 规划，但当前未配置 llmBaseUrl / llmModel。请管理员在配置中填写后使用。',
       },
     },
@@ -210,7 +233,19 @@ exports.Config = Schema.object({
   multiPrice: Schema.number().default(900).description('多人指令（p-draw 多人）单张消耗的 P 点'),
   couponPrice: Schema.number().default(3000).description('提示词优化券单价（P 点/张，购买询问时显示；可自动读取 data/p-shop.json 里的价格覆盖）'),
   couponAskTimeout: Schema.number().default(60).description('提示词优化券确认等待时间（秒）'),
-  img2imgDenoise: Schema.number().default(0.55).description('以图生图（p-draw i2i）的去噪强度，越小越接近原图（建议 0.4-0.7）'),
+  img2imgDenoise: Schema.number().default(0.55).description('普通以图生图（p-draw i2i）的去噪强度，越小越接近原图（建议 0.4-0.7）'),
+  i2iMode: Schema.string().default('ask').description('i2i 模式选择方式：ask=每次询问 / style=直接换风格（漫画化）/ ootd=直接换装换姿势 / plain=普通 img2img'),
+  i2iAskTimeout: Schema.number().default(60).description('i2i 模式询问等待时间（秒）'),
+  i2iStyleDenoise: Schema.number().default(0.75).description('换风格模式（漫画化）的去噪强度，越大风格变化越彻底（建议 0.7-0.85）'),
+  i2iOotdDenoise: Schema.number().default(0.55).description('换装换姿势模式（保留角色）的去噪强度（建议 0.5-0.6）'),
+  i2iControlNetStrength: Schema.number().default(0.7).description('换风格模式的 ControlNet 强度，越大构图锁得越死（建议 0.5-0.8）'),
+  i2iIPAdapterPath: Schema.string().default('').description('Anima IP-Adapter 模型文件路径（换装换姿势模式保脸用；需安装 comfyui-anima-ipadapter 节点并把模型路径填到这里）'),
+  i2iIPAdapterWeight: Schema.number().default(0.8).description('换装换姿势模式的 IP-Adapter 权重，越大角色特征保留越强（建议 0.6-1.0）'),
+  controlNetModel: Schema.string().default('').description('ControlNet 模型文件名（留空自动检测 Qwen/Anima 系 ControlNet；需放到 ComfyUI/models/controlnet）'),
+  taggerEnabled: Schema.boolean().default(false).description('i2i 前自动识图（需 ComfyUI 安装 WD14 Tagger 节点与模型；识别出的标签会注入提示词优化）'),
+  taggerModel: Schema.string().default('wd-v1-4-convnext-tagger-v2').description('识图模型名（WD14 Tagger 节点里可选模型）'),
+  taggerThreshold: Schema.number().default(0.35).description('识图标签置信度阈值'),
+  taggerCharacterThreshold: Schema.number().default(0.85).description('识图角色标签置信度阈值'),
   adminUsers: Schema.array(Schema.string()).default([]).description('免 P 点管理员用户 ID 列表'),
   outputLogs: Schema.boolean().default(true).description('是否在控制台输出详细日志'),
 
@@ -381,6 +416,18 @@ function parseSeed(text) {
   return { seed, prompt: cleaned }
 }
 
+// 去噪强度解析：--denoise 0.3 / --denoise=0.6 / --去噪 0.4，并从提示词里剥离
+function parseDenoise(text) {
+  const prompt = String(text || '').trim()
+  const m = prompt.match(/(?<!\S)--(?:denoise|去噪)\s*(?:=|＝|:|：)?\s*(\d+(?:\.\d+)?)/i)
+  if (!m) return { denoise: null, prompt }
+  const denoise = Math.min(1, Math.max(0, parseFloat(m[1])))
+  const cleaned = (prompt.slice(0, m.index) + ' ' + prompt.slice(m.index + m[0].length)).trim()
+    .replace(/^[\s,，;；:：]+|[\s,，;；:：]+$/g, '')
+    .replace(/\s+/g, ' ')
+  return { denoise, prompt: cleaned }
+}
+
 // ------------------------------------------------------------------
 // 提示词辅助（移植自 anima prompt_presets）
 // ------------------------------------------------------------------
@@ -519,6 +566,109 @@ function animaI2IWorkflow(cfg, prompt, negativePrompt, width, height, steps, cfg
   }
 }
 
+// 「换风格（漫画化）」i2i 工作流：LoadImage → Canny → ControlNetApplyAdvanced 锁构图，
+// VAEEncode 原图作起点，KSampler 用较高 denoise 整体转二次元风格。
+function animaStyleI2IWorkflow(cfg, prompt, negativePrompt, width, height, steps, cfgVal, seed, inputImage, controlNetModel, controlNetStrength, denoise) {
+  return {
+    '44': { class_type: 'UNETLoader', inputs: { unet_name: cfg.unetName, weight_dtype: 'fp8_e4m3fn' } },
+    '45': { class_type: 'CLIPLoader', inputs: { clip_name: cfg.clipName, type: 'stable_diffusion', device: 'default' } },
+    '15': { class_type: 'VAELoader', inputs: { vae_name: cfg.vaeName } },
+    '13': { class_type: 'LoadImage', inputs: { image: inputImage } },
+    '30': { class_type: 'VAEEncode', inputs: { pixels: ['13', 0], vae: ['15', 0] } },
+    '11': { class_type: 'CLIPTextEncode', inputs: { text: prompt, clip: ['45', 0] } },
+    '12': { class_type: 'CLIPTextEncode', inputs: { text: negativePrompt, clip: ['45', 0] } },
+    '70': { class_type: 'Canny', inputs: { image: ['13', 0], low_threshold: 0.4, high_threshold: 0.8 } },
+    '71': { class_type: 'ControlNetLoader', inputs: { controlnet_name: controlNetModel } },
+    '72': { class_type: 'ControlNetApplyAdvanced', inputs: { positive: ['11', 0], negative: ['12', 0], control_net: ['71', 0], image: ['70', 0], strength: controlNetStrength, start_percent: 0, end_percent: 1 } },
+    '19': {
+      class_type: 'KSampler',
+      inputs: {
+        model: ['44', 0],
+        positive: ['72', 0],
+        negative: ['72', 1],
+        latent_image: ['30', 0],
+        seed,
+        steps,
+        cfg: cfgVal,
+        sampler_name: cfg.samplerName,
+        scheduler: cfg.scheduler,
+        denoise,
+      },
+    },
+    '8': { class_type: 'VAEDecodeTiled', inputs: { samples: ['19', 0], vae: ['15', 0], tile_size: 512, overlap: 64, temporal_size: 64, temporal_overlap: 8 } },
+    '9': { class_type: 'SaveImage', inputs: { images: ['8', 0], filename_prefix: 'pdraw/anm_i2i_style' } },
+  }
+}
+
+// 「换装换姿势」i2i 工作流：LoadImage → AnimaSiglipeEncodeImage → AnimaIPAdapterApply 保角色特征，
+// VAEEncode 原图作起点，KSampler 用中等 denoise 重画服装/姿势/场景。
+function animaOotdI2IWorkflow(cfg, prompt, negativePrompt, width, height, steps, cfgVal, seed, inputImage, ipAdapterPath, ipAdapterWeight, denoise) {
+  return {
+    '44': { class_type: 'UNETLoader', inputs: { unet_name: cfg.unetName, weight_dtype: 'fp8_e4m3fn' } },
+    '45': { class_type: 'CLIPLoader', inputs: { clip_name: cfg.clipName, type: 'stable_diffusion', device: 'default' } },
+    '15': { class_type: 'VAELoader', inputs: { vae_name: cfg.vaeName } },
+    '13': { class_type: 'LoadImage', inputs: { image: inputImage } },
+    '30': { class_type: 'VAEEncode', inputs: { pixels: ['13', 0], vae: ['15', 0] } },
+    '11': { class_type: 'CLIPTextEncode', inputs: { text: prompt, clip: ['45', 0] } },
+    '12': { class_type: 'CLIPTextEncode', inputs: { text: negativePrompt, clip: ['45', 0] } },
+    '80': { class_type: 'AnimaIPAdapterLoader', inputs: { ipadapter_path: ipAdapterPath } },
+    '81': { class_type: 'AnimaSiglipeEncodeImage', inputs: { image: ['13', 0] } },
+    '82': { class_type: 'AnimaIPAdapterApply', inputs: { model: ['44', 0], ipadapter: ['80', 0], siglip_features: ['81', 0], start_at: 0, end_at: 1, weight: ipAdapterWeight } },
+    '19': {
+      class_type: 'KSampler',
+      inputs: {
+        model: ['82', 0],
+        positive: ['11', 0],
+        negative: ['12', 0],
+        latent_image: ['30', 0],
+        seed,
+        steps,
+        cfg: cfgVal,
+        sampler_name: cfg.samplerName,
+        scheduler: cfg.scheduler,
+        denoise,
+      },
+    },
+    '8': { class_type: 'VAEDecodeTiled', inputs: { samples: ['19', 0], vae: ['15', 0], tile_size: 512, overlap: 64, temporal_size: 64, temporal_overlap: 8 } },
+    '9': { class_type: 'SaveImage', inputs: { images: ['8', 0], filename_prefix: 'pdraw/anm_i2i_ootd' } },
+  }
+}
+
+// 根据 i2i 模式与已检测到的节点能力，决定最终工作流与去噪强度。
+// mode：style=换风格（漫画化）/ ootd=换装换姿势 / 其他=基础 img2img
+// caps：{ controlNet: { available, model }, ipAdapter: { available } }（由 detectI2ICapabilities 返回）
+// 返回 { promptBody, kind, denoise }，kind 为 plain / controlnet / ipadapter。
+function buildI2IWorkflow(cfg, prompt, negativePrompt, width, height, steps, cfgVal, seed, inputImage, mode, denoiseOverride, caps) {
+  const fallbackDenoise = Number(cfg.img2imgDenoise) || 0.55
+  if (mode === 'style') {
+    if (caps && caps.controlNet && caps.controlNet.available && caps.controlNet.model) {
+      const denoise = denoiseOverride != null ? denoiseOverride : (Number(cfg.i2iStyleDenoise) || 0.75)
+      return {
+        kind: 'controlnet',
+        denoise,
+        promptBody: animaStyleI2IWorkflow(cfg, prompt, negativePrompt, width, height, steps, cfgVal, seed, inputImage, caps.controlNet.model, Number(cfg.i2iControlNetStrength) || 0.7, denoise),
+      }
+    }
+    // 没有 ControlNet：换风格又不想崩构图，denoise 自动压到 0.5 以内
+    const denoise = denoiseOverride != null ? denoiseOverride : Math.min(Number(cfg.i2iStyleDenoise) || 0.75, 0.5)
+    return { kind: 'plain', denoise, promptBody: animaI2IWorkflow(cfg, prompt, negativePrompt, width, height, steps, cfgVal, seed, inputImage, denoise) }
+  }
+  if (mode === 'ootd') {
+    if (caps && caps.ipAdapter && caps.ipAdapter.available && String(cfg.i2iIPAdapterPath || '').trim()) {
+      const denoise = denoiseOverride != null ? denoiseOverride : (Number(cfg.i2iOotdDenoise) || 0.55)
+      return {
+        kind: 'ipadapter',
+        denoise,
+        promptBody: animaOotdI2IWorkflow(cfg, prompt, negativePrompt, width, height, steps, cfgVal, seed, inputImage, String(cfg.i2iIPAdapterPath).trim(), Number(cfg.i2iIPAdapterWeight) || 0.8, denoise),
+      }
+    }
+    const denoise = denoiseOverride != null ? denoiseOverride : (Number(cfg.i2iOotdDenoise) || 0.55)
+    return { kind: 'plain', denoise, promptBody: animaI2IWorkflow(cfg, prompt, negativePrompt, width, height, steps, cfgVal, seed, inputImage, denoise) }
+  }
+  const denoise = denoiseOverride != null ? denoiseOverride : fallbackDenoise
+  return { kind: 'plain', denoise, promptBody: animaI2IWorkflow(cfg, prompt, negativePrompt, width, height, steps, cfgVal, seed, inputImage, denoise) }
+}
+
 function customWorkflow(cfg, prompt, negativePrompt, width, height, steps, cfgVal, seed, explicitSize) {
   const rawPath = path.resolve(__dirname, cfg.customWorkflowPath)
   let raw
@@ -1447,8 +1597,9 @@ exports.apply = async function apply(ctx, cfg) {
     if (i2iImage && cfg.customWorkflowEnabled && cfg.customWorkflowPath) {
       return { ok: false, message: 'i2i（以图生图）暂不支持自定义工作流（customWorkflowEnabled），请关闭后再试。' }
     }
+    const i2iOpts = overrides.i2i || {}
     const promptBody = i2iImage
-      ? animaI2IWorkflow(workCfg, prompt, negativePrompt, width, height, steps, cfgVal, seed, i2iImage, Number(cfg.img2imgDenoise) || 0.55)
+      ? buildI2IWorkflow(workCfg, prompt, negativePrompt, width, height, steps, cfgVal, seed, i2iImage, i2iOpts.mode || 'plain', i2iOpts.denoise != null ? i2iOpts.denoise : null, i2iOpts.caps || null).promptBody
       : buildWorkflow(workCfg, prompt, negativePrompt, width, height, steps, cfgVal, seed, Boolean(size))
 
     const clientId = crypto.randomUUID()
@@ -2875,7 +3026,84 @@ exports.apply = async function apply(ctx, cfg) {
       return session.text('.i2i-upload-fail', [e.message])
     }
     if (cfg.outputLogs) logger.info(`[p-draw] i2i ${session.userId} 原图已上传：${uploadName}`)
-    return await handleGenerate(session, rawText, uploadName)
+
+    // 能力检测（ControlNet / Anima IP-Adapter）
+    const caps = await detectI2ICapabilities()
+
+    // 处理模式：i2iMode=ask 时交互询问；style/ootd/plain 直接固定
+    let mode = 'plain'
+    const cfgMode = String(cfg.i2iMode || 'ask').toLowerCase()
+    if (cfgMode === 'style' || cfgMode === 'ootd') {
+      mode = cfgMode
+    } else if (cfgMode === 'ask' && typeof session.prompt === 'function') {
+      const chosen = await askI2IMode(session)
+      if (chosen === 'cancel') return session.text('.i2i-mode-cancelled')
+      if (chosen === null) {
+        await session.send(session.text('.i2i-mode-cancelled'))
+        return ''
+      }
+      mode = chosen
+      if (mode === 'style') await session.send(session.text('.i2i-mode-style'))
+      if (mode === 'ootd') await session.send(session.text('.i2i-mode-ootd'))
+    }
+
+    // 所选模式依赖的节点缺失时给出提示（仍会回退普通 img2img，不中断）
+    const notices = []
+    if (mode === 'style' && (!caps.controlNet.available || !caps.controlNet.model)) {
+      notices.push(session.text('.i2i-no-controlnet'))
+    }
+    if (mode === 'ootd' && (!caps.ipAdapter.available || !String(cfg.i2iIPAdapterPath || '').trim())) {
+      notices.push(session.text('.i2i-no-ipadapter'))
+    }
+    if (notices.length) {
+      try { await session.send(notices.join('\n')) } catch (e) { logger.warn(`发送 i2i 提示失败：${e.message}`) }
+    }
+
+    // 识图：仅当启用、非无优化模式、且已配置 LLM（识图结果会注入提示词优化）时才执行
+    let taggerTags = ''
+    if (cfg.taggerEnabled && cfg.llmModel && cfg.llmBaseUrl && !stripRawPrefix(rawText).raw) {
+      try {
+        taggerTags = await taggerImage(uploadName)
+        if (cfg.outputLogs) logger.info(`[p-draw] i2i ${session.userId} 识图完成：${taggerTags.slice(0, 120)}${taggerTags.length > 120 ? '...' : ''}`)
+      } catch (e) {
+        logger.warn(`识图失败（继续生图）：${e.message}`)
+      }
+    }
+    return await handleGenerate(session, rawText, uploadName, taggerTags, { mode, caps })
+  }
+
+  // 识图预工作流：LoadImage → WD14Tagger|pysssss，读回原图 tags
+  async function taggerImage(uploadName) {
+    const model = String(cfg.taggerModel || 'wd-v1-4-convnext-tagger-v2').trim()
+    const taggerNodeId = '61'
+    const promptBody = {
+      '60': { class_type: 'LoadImage', inputs: { image: uploadName } },
+      [taggerNodeId]: {
+        class_type: 'WD14Tagger|pysssss',
+        inputs: {
+          image: ['60', 0],
+          model,
+          threshold: Number(cfg.taggerThreshold) || 0.35,
+          character_threshold: Number(cfg.taggerCharacterThreshold) || 0.85,
+          replace_underscore: true,
+          trailing_comma: false,
+          exclude_tags: '',
+        },
+      },
+    }
+    const clientId = crypto.randomUUID()
+    const submit = await comfyPost('/prompt', { prompt: promptBody, client_id: clientId }, 20000)
+    const promptId = submit && submit.prompt_id
+    if (!promptId) throw new Error('识图工作流提交失败')
+    const timeoutMs = Math.max(1, parseInt(cfg.timeout) || 300) * 1000
+    const pollMs = Math.max(1, parseInt(cfg.pollInterval) || 2) * 1000
+    const history = await waitComfyResult(ctx, comfyGet, baseUrl(), promptId, clientId, timeoutMs, pollMs)
+    if (!history) throw new Error('识图超时')
+    const nodeOutput = history.outputs && history.outputs[taggerNodeId]
+    const tags = (nodeOutput && Array.isArray(nodeOutput.tags) ? nodeOutput.tags : []).filter(Boolean)
+    const first = String(tags[0] || '').trim()
+    if (!first) throw new Error('识图未返回标签')
+    return first
   }
 
   // 从会话消息里取第一张图片并下载为内存 Buffer
@@ -2925,12 +3153,62 @@ exports.apply = async function apply(ctx, cfg) {
     return data.name
   }
 
-  async function handleGenerate(session, rawText, i2iImage = null) {
+  // 检测 i2i 可用能力：ControlNet 模型（ControlNetLoader 的 control_net_name 列表）与
+  // Anima IP-Adapter 专用节点（comfyui-anima-ipadapter）。检测失败全部视为不可用（回退普通 img2img）。
+  async function detectI2ICapabilities() {
+    const caps = { controlNet: { available: false, model: null }, ipAdapter: { available: false } }
+    let info
+    try {
+      info = await getObjectInfoCached()
+    } catch (e) {
+      return caps
+    }
+    if (!info || typeof info !== 'object') return caps
+    const cnModels = availableModels(info, 'ControlNetLoader', 'control_net_name')
+    if (cnModels.length) {
+      caps.controlNet.available = true
+      const configured = String(cfg.controlNetModel || '').trim()
+      if (configured && cnModels.includes(configured)) {
+        caps.controlNet.model = configured
+      } else {
+        const norm = (s) => String(s).toLowerCase().replace(/[-_ ]/g, '')
+        const prefers = ['qwen-image-instantx-controlnet-union', 'anima-instantx-controlnet-union', 'instantx-controlnet-union', 'qwen-image-controlnet-union', 'controlnet-union']
+        const pool = cnModels.filter((m) => /qwen|anima|instant|union|canny/i.test(String(m)))
+        const scored = pool.map((m) => ({ m, s: prefers.reduce((acc, p, i) => acc + (norm(m) === norm(p) ? prefers.length - i : 0), 0) }))
+        scored.sort((a, b) => b.s - a.s)
+        caps.controlNet.model = (scored[0] && scored[0].m) || null
+      }
+    }
+    if (info.AnimaIPAdapterLoader && info.AnimaIPAdapterApply && info.AnimaSiglipeEncodeImage) {
+      caps.ipAdapter.available = true
+    }
+    if (cfg.outputLogs) logger.info(`[p-draw] i2i 能力检测：ControlNet=${JSON.stringify(caps.controlNet)} IPAdapter=${JSON.stringify(caps.ipAdapter)}`)
+    return caps
+  }
+
+  // 交互式询问 i2i 处理模式，返回 'style' / 'ootd' / 'cancel' / null（超时或无法询问时 null）
+  async function askI2IMode(session, i18nAsk) {
+    if (typeof session.prompt !== 'function') return null
+    await session.send(i18nAsk || session.text('.i2i-mode-ask'))
+    const reply = await session.prompt((cfg.i2iAskTimeout || 60) * 1000).catch(() => null)
+    const ans = String((reply && (reply.content != null ? reply.content : reply)) || '').trim()
+    if (/^(1|①|换风格|风格|漫画|漫画化|style|stylechange)$/i.test(ans)) return 'style'
+    if (/^(2|②|换装|换装换姿势|换衣服|换姿势|ootd|outfit)$/i.test(ans)) return 'ootd'
+    if (/^(3|③|取消|不生成|不要|算了|cancel|no)$/i.test(ans)) return 'cancel'
+    if (!ans) return null
+    await session.send(session.text('.i2i-mode-invalid'))
+    return askI2IMode(session, i18nAsk)
+  }
+
+  async function handleGenerate(session, rawText, i2iImage = null, taggerTags = '', i2iOpts = null) {
     const USERID = session.userId
     const isAdmin = isAdminUser(session)
     const unet = await resolveUnet(USERID)
     const i2iRule = i2iImage
       ? '这是以图生图（img2img）：原图的构图、主体与色调会被保留。请主要描述你希望发生的变化（风格、服装、表情、场景改造、细节调整等），不要重复描述原图已有的细节。'
+        + (taggerTags
+          ? `\n\n原图内容参考（本地识图自动识别出的标签，用于了解原图已包含的内容；不要照抄全部标签，只参考与用户改动需求相关的部分）：\n${taggerTags}`
+          : '')
       : ''
     if (cfg.outputLogs) {
       logger.info(`[p-draw] 请求 userId=${USERID} isAdmin=${isAdmin} adminUsers=${JSON.stringify(cfg.adminUsers || [])} normalizeId=${normalizeId(USERID)}`)
@@ -2945,8 +3223,11 @@ exports.apply = async function apply(ctx, cfg) {
     const parsedBatch = parseBatchCount(parsedSize.prompt, cfg.batchMax)
     // 固定种子解析（--seed:xxx / --seed xxx / --seed=xxx），并从提示词中剥离
     const parsedSeed = parseSeed(parsedBatch.prompt)
-    const text = parsedSeed.prompt
+    // i2i 去噪强度解析（--denoise 0.6 / --去噪 0.4），并从提示词中剥离
+    const parsedDenoise = i2iImage ? parseDenoise(parsedSeed.prompt) : { denoise: null, prompt: parsedSeed.prompt }
+    const text = parsedDenoise.prompt
     const seed = parsedSeed.seed
+    const denoise = parsedDenoise.denoise
     const count = parsedBatch.count
 
     // P 点校验（按总价 = 张数 × 单价）
@@ -3041,6 +3322,11 @@ exports.apply = async function apply(ctx, cfg) {
       if (cfg.outputLogs) logger.info(`[p-draw] ${USERID} 已扣除 ${count * cfg.price} P 点（${count} 张 × ${cfg.price}），余额 ${saving - count * cfg.price}`)
     }
 
+    // i2i 运行参数：模式（style/ootd/plain）、--denoise 覆盖值、检测到的能力
+    const i2iRun = i2iImage
+      ? { i2iImage, i2i: { mode: (i2iOpts && i2iOpts.mode) || 'plain', denoise: denoise != null ? denoise : null, caps: (i2iOpts && i2iOpts.caps) || null } }
+      : {}
+
     // 队列：预排队全部任务，占满即退回总价
     const queuedTasks = []
     let firstPosition = null
@@ -3052,7 +3338,7 @@ exports.apply = async function apply(ctx, cfg) {
             const optimized = await optimizePrompt(session, userPrompt, true, i2iRule)
             p = composePrompt(optimized.prompt || userPrompt, raw).prompt
           }
-          return runComfyGenerate(p, parsedSize.size, { unet, seed, i2iImage })
+          return runComfyGenerate(p, parsedSize.size, Object.assign({ unet, seed }, i2iRun))
         })
         if (!q.ok) {
           if (!isAdmin) await refundP(USERID, count * cfg.price)
@@ -3108,7 +3394,7 @@ exports.apply = async function apply(ctx, cfg) {
       if (cfg.queueEnabled) {
         try { result = await queuedTasks[i] } catch (e) { result = { ok: false, message: `生成失败：${e.message}` } }
       } else {
-        try { result = await runComfyGenerate(p, parsedSize.size, { unet, seed, i2iImage }) } catch (e) { result = { ok: false, message: `生成失败：${e.message}` } }
+        try { result = await runComfyGenerate(p, parsedSize.size, Object.assign({ unet, seed }, i2iRun)) } catch (e) { result = { ok: false, message: `生成失败：${e.message}` } }
       }
       return result
     }
@@ -3283,5 +3569,5 @@ exports.apply = async function apply(ctx, cfg) {
   })
 
   // 暴露内部接口供自动化测试调用（Koishi 忽略 apply 返回值，不影响生产行为）
-  return { couponConfirmFlow, buyCouponsAndConsume, normalizeConfirm, resolveCouponPrice, handleGenerateI2I, extractImageFromSession, uploadImageToComfyui, animaI2IWorkflow }
+  return { couponConfirmFlow, buyCouponsAndConsume, normalizeConfirm, resolveCouponPrice, handleGenerateI2I, extractImageFromSession, uploadImageToComfyui, taggerImage, animaI2IWorkflow, animaStyleI2IWorkflow, animaOotdI2IWorkflow, buildI2IWorkflow, detectI2ICapabilities, parseDenoise }
 }

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-p-draw",
   "description": "p点-绘图，连接本地 ComfyUI 生图。支持自然语言优化、尺寸选择、画师组、固定角色、生成队列，需要配合 p-qiandao 使用",
-  "version": "1.0.0",
+  "version": "1.2.0",
   "main": "index.js",
   "files": [
     "index.js",

package/readme.md

@@ -14,7 +14,8 @@ p点-绘图插件，连接本地 ComfyUI 生图，参考 AstrBot 的 [anima 绘
 - 尺寸选择：`竖图` `横图` `方图` `长竖图` `宽屏`，或 `1024x1536：描述` / `--尺寸 1216x832`。
 - 画师组管理（创建 / 追加 / 切换 / 查看 / 删除）。
 - 固定角色（`添加角色 名称=tags`）。
-- **以图生图**：`p-draw i2i <描述>` 并在同一条消息附一张原图（文件 / 截图 / 链接均可），保留原图构图与主体，按描述调整风格与细节。使用 ComfyUI 内置 `LoadImage → VAEEncode → KSampler(denoise<1)` 标准节点，**无需修改 ComfyUI**；去噪强度由 `img2imgDenoise` 控制（默认 0.55，越小越接近原图）。不支持自定义工作流（`customWorkflowEnabled`）时使用 i2i 会提示关闭。
+- **以图生图**：`p-draw i2i <描述>` 并在同一条消息附一张原图（文件 / 截图 / 链接均可），发送后会让用户选择处理方式（可用 `i2iMode` 固定跳过询问）：**① 换风格（漫画化）**——保留原图构图，整体转成二次元画风，优先用 ControlNet（Canny 锁构图）；**② 换装换姿势**——保留角色长相，重新设计服装 / 姿势 / 场景，优先用 Anima IP-Adapter（保脸）；**③ 取消**——不生成。支持 `--denoise 0.6` / `--去噪 0.4` 单独调整强度；未安装对应节点 / 模型时自动回退普通 img2img（纯 `LoadImage → VAEEncode → KSampler(denoise<1)`，**无需修改 ComfyUI**），去噪强度由 `img2imgDenoise` 控制（默认 0.55）。不支持自定义工作流（`customWorkflowEnabled`）时使用 i2i 会提示关闭。
+- **i2i 自动识图**：配置 `taggerEnabled` 后，i2i 会先通过 ComfyUI 的 **WD14 Tagger** 节点对原图识图，把识别出的标签注入提示词优化，让不支持读图的 LLM（如 DeepSeek）也能"看到"原图内容。需要在 ComfyUI `custom_nodes` 安装 [ComfyUI-WD14-Tagger](https://github.com/pythongosssss/ComfyUI-WD14-Tagger)，并把模型 `.onnx` + `.csv` 放进 `custom_nodes/ComfyUI-WD14-Tagger/models/`（模型名由 `taggerModel` 指定，默认 `wd-v1-4-convnext-tagger-v2`）。仅当已配置 LLM（`llmBaseUrl`/`llmModel`）且非 `无优化` 模式时才会识图；识图失败会跳过并正常生图。
 - **用户自选模型**：`p-draw 模型` 查看当前与可用模型（自动读取 ComfyUI 的 UNET 列表），`p-draw 模型 <名称>` 切换（支持模糊匹配，如 `anima-aesthetic`），`p-draw 模型 默认` 恢复默认。偏好按用户保存（`p_draw_config.user_models`），单人 / 多人 / 连续图都生效；未自选时用 `unetName` 默认模型。自定义工作流（`customWorkflowEnabled`）不参与模型切换。
 - 画师组与固定角色**保存在数据库 `p_draw_config` 表中**，管理指令不触发插件重载（不会打断正在生成的图，也不会把配置恢复成默认）。数据库中的值在重启后覆盖配置里的同名项。
 - 自定义 ComfyUI 工作流 JSON。
@@ -51,7 +52,7 @@ plugins:
 画图 创建画师组 千代风格=@artist_a, @artist_b,
 画图 切换画师组 千代风格
 画图 添加角色 狐莉=1girl, solo, fox girl
-画图 i2i 换成晚礼服，背景换成舞台灯光（同一条消息附原图）
+画图 i2i 换成晚礼服，背景换成舞台灯光 --denoise 0.6（同一条消息附原图，随后回复 1=换风格 / 2=换装换姿势 / 3=取消）
 画图 多人 左边若叶睦抱着吉他，右边千早爱音牵着她的手
 画图 多人 若叶睦与千早爱音十指相扣 x2
 画图 连续 圣女：白色长发红瞳、圣洁白裙、端庄微笑 → 发梢染黑、眼神迷离、白裙开裂 → 黑发红瞳、妖艳邪笑、黑色破损礼服
@@ -75,7 +76,16 @@ plugins:
 | `webSearchEnabled` / `tavilyApiKey` / `webSearchMaxResults` / `webSearchDepth` / `webSearchQueryTemplate` | 联网搜索（指令带 `联网`/`搜索` 触发，需 Tavily Key） |
 | `fixedCharacters` / `artistPresets` / `activeArtistPreset` / `styleTags` | 角色、画师组与画风 |
 | `queueEnabled` / `queueMaxRequests` | 生成队列 |
-| `img2imgDenoise` | 以图生图（`p-draw i2i`）去噪强度，默认 0.55（建议 0.4-0.7，越小越接近原图） |
+| `img2imgDenoise` | 普通以图生图（`p-draw i2i`）去噪强度，默认 0.55（建议 0.4-0.7，越小越接近原图） |
+| `i2iMode` | i2i 模式选择方式：`ask`=每次询问（默认） / `style`=直接换风格（漫画化） / `ootd`=直接换装换姿势 / `plain`=普通 img2img |
+| `i2iAskTimeout` | i2i 处理方式询问等待时间（秒，默认 60） |
+| `i2iStyleDenoise` | 换风格（漫画化）去噪强度，默认 0.75（建议 0.7-0.85，越大风格变化越彻底；无 ControlNet 时自动压到 0.5 防崩） |
+| `i2iOotdDenoise` | 换装换姿势去噪强度，默认 0.55（建议 0.5-0.6） |
+| `i2iControlNetStrength` | 换风格模式的 ControlNet 强度，默认 0.7（建议 0.5-0.8，越大构图锁得越死） |
+| `controlNetModel` | ControlNet 模型文件名（留空自动检测 Qwen/Anima 系，如 `Qwen-Image-InstantX-ControlNet-Union.safetensors`；需放到 ComfyUI `models/controlnet/`） |
+| `i2iIPAdapterPath` | Anima IP-Adapter 模型文件**路径**（换装换姿势保脸用，如 `E:/anima/ipadapter.safetensors`；需安装 [comfyui-anima-ipadapter](https://github.com/Wenaka2004/comfyui-anima-ipadapter) 节点） |
+| `i2iIPAdapterWeight` | 换装换姿势模式的 IP-Adapter 权重，默认 0.8（建议 0.6-1.0，越大角色特征保留越强） |
+| `taggerEnabled` / `taggerModel` / `taggerThreshold` / `taggerCharacterThreshold` | i2i 自动识图开关、识图模型名、标签阈值 / 角色标签阈值（需 ComfyUI 安装 WD14 Tagger 节点与模型） |
 | `couponPrice` / `couponAskTimeout` | 提示词优化券单价（P 点/张，默认 3000，可被 `data/p-shop.json` 覆盖）与交互确认等待时间（秒，默认 60） |
 | `batchMax` | 单次指令最多生成的张数（`x3` / `3张` / `--数量 3` 等写法），默认 4 |
 | `price` / `multiPrice` / `adminUsers` | 单张 / 多人单张消耗的 P 点与免单管理员 |
@@ -101,4 +111,5 @@ plugins:
   - 券为一次性、**按张数消耗**：`x3` 需 3 张，每张图独立做一次 LLM 优化；券不足时该批不优化也不扣券（提示 `token-short`）。等待回复时间由 `couponAskTimeout`（秒）控制。
   - **管理员在全局关闭时自动免费优化**（不耗券）；`无优化` 原样模式不消耗、不询问。平台不支持交互确认时退回自动消耗逻辑。
 - **未使用优化的提示**：单人生图没走 LLM 优化时会明确提示原因——未配置 LLM、未消耗优化券、或 `无优化` 原样模式；多人指令强依赖 LLM 规划，未配置 `llmBaseUrl`/`llmModel` 时会直接提示「多人指令需要 LLM 规划」。
+- **i2i 增强模式依赖的节点**：换风格（漫画化）需要 ComfyUI `models/controlnet/` 里有 Qwen/Anima 系 ControlNet 模型（`ControlNetLoader` 下拉可见）；换装换姿势需要安装 [comfyui-anima-ipadapter](https://github.com/Wenaka2004/comfyui-anima-ipadapter) 自定义节点并配置 `i2iIPAdapterPath`。两者缺失时都会自动回退普通 img2img 并提示，不影响出图。
 - 本地模型不一定认识新角色，可用 `添加角色` 手动补充角色 tags。