koishi-plugin-aka-ai-generator 0.0.8 → 0.1.0

package/lib/index.js

@@ -521,6 +521,7 @@ var COMMANDS = {
   GENERATE_IMAGE: "生成图像",
   COMPOSE_IMAGE: "合成图像",
   CHANGE_POSE: "改姿势",
+  OPTIMIZE_DESIGN: "优化设计",
   QUERY_QUOTA: "图像额度",
   RECHARGE: "图像充值",
   RECHARGE_HISTORY: "图像充值记录",
@@ -618,6 +619,7 @@ function apply(ctx, config) {
       { name: COMMANDS.GENERATE_IMAGE, description: "使用自定义prompt进行图像处理" },
       { name: COMMANDS.COMPOSE_IMAGE, description: "合成多张图片，使用自定义prompt控制合成效果" },
       { name: COMMANDS.CHANGE_POSE, description: "改变图像主体的姿势造型，保持主体细节和风格不变" },
+      { name: COMMANDS.OPTIMIZE_DESIGN, description: "优化图像主体的结构设计，保持原有设计语言和风格" },
       { name: COMMANDS.QUERY_QUOTA, description: "查询用户额度信息" }
     ],
     // 管理员指令
@@ -1197,6 +1199,15 @@ Prompt: ${prompt}`);
     const posePrompt = "请根据用户提供的图片，在严格保持主体身份、外观特征、服装细节、艺术风格和整体氛围不变的前提下，生成一个新的姿势造型。新姿势应该更加帅气、可爱、有张力或符合主体内容的动态感，展现出更好的视觉表现力。要求：1. 完全保持主体的面部特征、发型、服装、配饰等所有细节不变；2. 完全保持原有的艺术风格（如二次元、写实、手绘等）不变；3. 只改变身体的姿势、动作和姿态，让主体看起来更有活力和表现力；4. 姿势应该自然、协调，符合主体的身份和性格特征；5. 保持背景环境的基本风格不变（可以适当调整视角或构图）。";
     return processImageWithTimeout(session, img, posePrompt, COMMANDS.CHANGE_POSE, options?.num);
   });
+  ctx.command(`${COMMANDS.OPTIMIZE_DESIGN} [img:text]`, "优化图像主体的结构设计，保持原有设计语言和风格").option("num", "-n <num:number> 生成图片数量 (1-4)").action(async ({ session, options }, img) => {
+    if (!session?.userId) return "会话无效";
+    const limitCheck = await checkDailyLimit(session.userId);
+    if (!limitCheck.allowed) {
+      return limitCheck.message;
+    }
+    const designPrompt = "请根据用户提供的图片，在严格保持原有设计语言、视觉风格、功能特征和整体主题不变的前提下，对图像主体的结构设计进行优化。要求：1. 完全保持原有的设计语言和视觉风格（如现代简约、复古、科幻、奇幻等）不变；2. 保持主体的核心功能特征和身份定位不变；3. 可以合理且美观地添加、删改或优化结构元素（如装饰细节、功能组件、线条轮廓、比例关系等），使设计更加完善和美观；4. 所有优化必须符合原有主题的视觉风格，增强设计美感而不破坏原有设计语言；5. 优化后的设计应该更加协调、统一，具有更好的视觉层次和设计完整性；6. 保持色彩方案、材质质感和整体氛围的一致性。";
+    return processImageWithTimeout(session, img, designPrompt, COMMANDS.OPTIMIZE_DESIGN, options?.num);
+  });
   ctx.command(`${COMMANDS.RECHARGE} [content:text]`, "为用户充值次数（仅管理员）").action(async ({ session }, content) => {
     if (!session?.userId) return "会话无效";
     if (!isAdmin(session.userId)) {

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-aka-ai-generator",
   "description": "自用AI生成插件(GPTGod & Yunwu)",
-  "version": "0.0.8",
+  "version": "0.1.0",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "files": [