koishi-plugin-p-draw 1.5.6 → 1.5.8

package/index.js

@@ -174,8 +174,7 @@ exports.apply = async function apply(ctx, cfg) {
   // 用户自选模型偏好（userid -> unet 文件名），持久化在 p_draw_config.user_models
   if (!cfg.userModels || typeof cfg.userModels !== 'object') cfg.userModels = {}
 
-  const tempDir = fs.mkdtempSync(path.join(os.tmpdir(), 'pdraw-'))
-  // 启动时清理上次运行遗留的临时文件
+  // 启动时清理上次运行遗留的临时目录（先清理，再创建本次目录）
   try {
     const leftovers = fs.readdirSync(os.tmpdir()).filter((n) => n.startsWith('pdraw-'))
     for (const name of leftovers) {
@@ -187,6 +186,8 @@ exports.apply = async function apply(ctx, cfg) {
     }
   } catch (e) { /* ignore */ }
 
+  const tempDir = fs.mkdtempSync(path.join(os.tmpdir(), 'pdraw-'))
+
   const baseUrl = () => normalizeBaseUrl(cfg.comfyuiBaseUrl)
 
   function parseAllowedSizes() {
@@ -478,11 +479,15 @@ exports.apply = async function apply(ctx, cfg) {
     return `已扣除 ${totalPrice} P 点，${modelSeed}`
   }
 
-  function buildGenerationReply(session, { successCount, count, failures, notes = [] }) {
+  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')
+    return reply.filter(Boolean).join('\n')
+  }
+
+  function shouldOptimizePerImage({ count, enabled, admin }) {
+    return count > 1 && (enabled || admin)
   }
 
   async function sendNotices(session, notices, opts = {}) {
@@ -1866,19 +1871,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 优化。
@@ -2169,5 +2180,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.6",
+  "version": "1.5.8",
   "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` 等动态要求可以在不同图片中产生不同结果。
 
 ## 安装