From a25a7dc023b897395f14c5492bb5bf9918f0a4fe Mon Sep 17 00:00:00 2001
From: xupei <xupei3360@163.com>
Date: Mon, 2 Dec 2024 14:34:10 +0800
Subject: [PATCH] =?UTF-8?q?high=20||=20fast=20=E7=94=9F=E6=88=90=E6=AC=A1?=
 =?UTF-8?q?=E6=95=B0=E4=BF=AE=E6=94=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../da/service/impl/GenerateServiceImpl.java  | 23 +++++++++++--------
 1 file changed, 13 insertions(+), 10 deletions(-)

diff --git a/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java b/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
index b8725f5c..70f039c8 100644
--- a/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
+++ b/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
@@ -503,6 +503,13 @@ public class GenerateServiceImpl extends ServiceImpl<GenerateMapper, Generate> i
                 // 校验后获取
                 generateThroughImageTextDTO.setGenerateType(generate.getGenerateType());
                 creditsEventsEnum = CreditsEventsEnum.PATTERN;
+
+                // 模型迁移SD1.? -> flux,从而产生了不同模型的选择，
+                // high -> 生成图片质量高，但生成速度慢，每次生成只返回一张图片
+                // fast -> 生成图片质量低，但生成速度快，每次生成返回四张图片
+                if (!StringUtil.isNullOrEmpty(generateThroughImageTextDTO.getVersion()) && generateThroughImageTextDTO.getVersion().equals("high")){
+                    times = 1;
+                }
             }
             // Slogan 参数校验
             if (generateThroughImageTextDTO.getLevel2Type().equals(CollectionLevel2TypeEnum.SLOGAN.getRealName())) {
@@ -559,8 +566,14 @@ public class GenerateServiceImpl extends ServiceImpl<GenerateMapper, Generate> i
             }
         } else if (generateThroughImageTextDTO.getLevel1Type().equals(MOOD_BOARD.getRealName())) {
             creditsEventsEnum = CreditsEventsEnum.MOOD_BOARD;
+            if (!StringUtil.isNullOrEmpty(generateThroughImageTextDTO.getVersion()) && generateThroughImageTextDTO.getVersion().equals("high")){
+                times = 1;
+            }
         } else if (generateThroughImageTextDTO.getLevel1Type().equals(SKETCH_BOARD.getRealName())) {
             creditsEventsEnum = CreditsEventsEnum.SKETCH_BOARD;
+            if (!StringUtil.isNullOrEmpty(generateThroughImageTextDTO.getVersion()) && generateThroughImageTextDTO.getVersion().equals("high")){
+                times = 1;
+            }
         }
 
         // 2、判断用户当前积分是否够本次生成消耗
@@ -572,16 +585,6 @@ public class GenerateServiceImpl extends ServiceImpl<GenerateMapper, Generate> i
         // 3、生成唯一id 使用uuid,由于uuid重复的几率很小，故取消对uuid重复性的校验
         String uuid = UUID.randomUUID().toString();
 
-
-        // 模型迁移SD1.? -> flux,从而产生了不同模型的选择，
-        // high -> 生成图片质量高，但生成速度慢，每次生成只返回一张图片
-        // fast -> 生成图片质量低，但生成速度快，每次生成返回四张图片
-        if (!StringUtil.isNullOrEmpty(generateThroughImageTextDTO.getVersion()) && generateThroughImageTextDTO.getVersion().equals("high")){
-            times = 1;
-        }else if (!StringUtil.isNullOrEmpty(generateThroughImageTextDTO.getVersion()) && generateThroughImageTextDTO.getVersion().equals("fast")){
-            times = 4;
-        }
-
         // 除了 Moodboard || Printboard->Pattern(可以区分三种风格) || Sketchboard(Generate Sketch)这三个地方需要区分high || fast之外，其他地方保持原样
         if (generateThroughImageTextDTO.getLevel1Type().equals("Printboard") && !generateThroughImageTextDTO.getLevel2Type().equals("Pattern")){
             generateThroughImageTextDTO.setVersion(null);