koishi-plugin-p-draw 1.5.5 → 1.5.7

package/index.js

@@ -478,13 +478,16 @@ exports.apply = async function apply(ctx, cfg) {
     return `已扣除 ${totalPrice} P 点，${modelSeed}`
   }
 
-  function buildGenerationReply(session, { successCount, count, failures, notes = [], chargeNotice = '' }) {
-    const reply = []
-    if (chargeNotice) reply.push(chargeNotice)
-    if (notes.length) reply.push(notes.join('\n'))
-    if (failures.length) reply.push(session.text('.batch-partial', [successCount, count, failures.length, failures.join('；')]))
+  function buildGenerationReply(session, { successCount, count, failures, notes = [] }) {
+    const reply = []
+    if (notes.length) reply.push(notes.join('\n'))
+    if (failures.length) reply.push(session.text('.batch-partial', [successCount, count, failures.length, failures.join('；')]))
     return reply.filter(Boolean).join('\n')
   }
+
+  function shouldOptimizePerImage({ count, enabled, admin }) {
+    return count > 1 && (enabled || admin)
+  }
 
   async function sendNotices(session, notices, opts = {}) {
     const content = notices.filter(Boolean).join('\n')
@@ -1492,10 +1495,12 @@ exports.apply = async function apply(ctx, cfg) {
     // 发图：合并转发，不引用原指令
     await sendImagesAsForward(session, forwardOutputs)
 
-    return buildGenerationReply(session, { successCount, count, failures, notes, chargeNotice })
+    await sendNotices(session, [chargeNotice])
+
+    return buildGenerationReply(session, { successCount, count, failures, notes })
   }
 
-  // ---------------- 权限 ----------------
+  // ---------------- 权限 ----------------
   function readableOptimizeReason(reason) {
     if (!reason) return ''
     if (reason === 'llm_not_configured') return '（未配置 LLM 模型名或接口地址）'
@@ -1865,19 +1870,25 @@ exports.apply = async function apply(ctx, cfg) {
     if (!raw) {
       const globalOpt = cfg.promptOptimizeEnabled
       const adminOpt = !globalOpt && isAdmin && cfg.llmModel && cfg.llmBaseUrl
-      const tokenOpt = !globalOpt && !isAdmin && cfg.llmModel && cfg.llmBaseUrl
-      if (globalOpt) {
-        // 全局优化开启：一次优化，整批复用同一提示词
-        const optimized = await optimizePrompt(session, userPrompt, false)
-        finalPrompt = optimized.prompt
-        degraded = !optimized.ok
-        optimizedReason = optimized.reason || ''
-      } else if (adminOpt) {
-        // 管理员在全局关闭时也免费优化（不耗券）
-        const optimized = await optimizePrompt(session, userPrompt, true)
-        finalPrompt = optimized.prompt
-        degraded = !optimized.ok
-        optimizedReason = optimized.reason || ''
+      const tokenOpt = !globalOpt && !isAdmin && cfg.llmModel && cfg.llmBaseUrl
+      if (globalOpt) {
+        if (shouldOptimizePerImage({ count, enabled: true, admin: false })) {
+          perImageOptimize = true
+        } else {
+          const optimized = await optimizePrompt(session, userPrompt, false)
+          finalPrompt = optimized.prompt
+          degraded = !optimized.ok
+          optimizedReason = optimized.reason || ''
+        }
+      } else if (adminOpt) {
+        if (shouldOptimizePerImage({ count, enabled: false, admin: true })) {
+          perImageOptimize = true
+        } else {
+          const optimized = await optimizePrompt(session, userPrompt, true)
+          finalPrompt = optimized.prompt
+          degraded = !optimized.ok
+          optimizedReason = optimized.reason || ''
+        }
       } else if (tokenOpt) {
         // 全局优化关闭：使用 p-shop 的「提示词优化券」（p_system.llmToken）。
         // 券一次性，按张数扣：x3 扣 3 张，每张图独立做一次 LLM 优化。
@@ -2020,8 +2031,10 @@ exports.apply = async function apply(ctx, cfg) {
     // 发图：合并转发，不引用原指令
     await sendImagesAsForward(session, forwardOutputs)
 
-    return buildGenerationReply(session, { successCount, count, failures, chargeNotice })
-  }
+    await sendNotices(session, [chargeNotice])
+
+    return buildGenerationReply(session, { successCount, count, failures })
+  }
   // ---------------- 提示词优化券交互式确认 ----------------
   // 仅全局优化关闭 + 非管理员 + 已配置 LLM（tokenOpt 分支）时进入。
   // 返回：{ cancelled, tokenUsedCount, perImageOptimize, noOptimizeReason }
@@ -2166,5 +2179,5 @@ exports.apply = async function apply(ctx, cfg) {
   })
 
   // 暴露内部接口供自动化测试调用（Koishi 忽略 apply 返回值，不影响生产行为）
-  return { couponConfirmFlow, buyCouponsAndConsume, normalizeConfirm, resolveCouponPrice, buildChargeNotice, buildGenerationReply, sendNotices, sendImagesAsForward, executeBatch }
+  return { couponConfirmFlow, buyCouponsAndConsume, normalizeConfirm, resolveCouponPrice, buildChargeNotice, buildGenerationReply, shouldOptimizePerImage, sendNotices, sendImagesAsForward, executeBatch }
 }

package/package.json

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

package/readme.md

@@ -21,6 +21,7 @@ p点-绘图插件，连接本地 ComfyUI 生图，参考 AstrBot 的 [anima 绘
 - `p-draw 状态` 查看 ComfyUI 连接状态与模型可用性。
 - `p-draw help`（或 `帮助`）查看全部指令帮助，包含生成 / 多人 / 批量张数 / 尺寸 / 画师组 / 状态等。
 - 发图时会以合并转发发送，每张图片后附对应的 Positive / Negative 提示词；生成完成消息会显示实际模型和 seed（如 `已扣除 750 P 点，当前模型：animagine-xl-3.1.safetensors，--seed=123456`）。管理员/免扣费用户不显示扣费金额，但仍显示模型和 seed。
+- 批量生成（如 `x5` / `x10`）开启 LLM 优化时，每张图会在生成前单独重新优化一次提示词；因此 `random color hair`、`random color eyes` 等动态要求可以在不同图片中产生不同结果。
 
 ## 安装