From 8d7b4adf482bc6f1ade570218bb831a03be89f4f Mon Sep 17 00:00:00 2001
From: xupei <xupei3360@163.com>
Date: Mon, 16 Jun 2025 18:51:09 +0800
Subject: [PATCH] =?UTF-8?q?BUGFIX:=201=E3=80=81sketch=20extract=20?=
 =?UTF-8?q?=E6=B2=A1=E8=BF=94=E5=9B=9EtaskId=202=E3=80=81relight=E8=BE=93?=
 =?UTF-8?q?=E5=85=A5=E6=96=87=E6=9C=AC=E5=A4=AA=E9=95=BF=E5=AF=BC=E8=87=B4?=
 =?UTF-8?q?=E6=95=B0=E6=8D=AE=E5=AD=98=E5=82=A8=E6=8A=A5=E9=94=99?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../da/service/impl/GenerateServiceImpl.java  |  3 +-
 .../impl/UserLikeGroupServiceImpl.java        | 42 ++++++++++++-------
 2 files changed, 30 insertions(+), 15 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 199ce1e3..3c48d3e4 100644
--- a/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
+++ b/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
@@ -962,7 +962,8 @@ public class GenerateServiceImpl extends ServiceImpl<GenerateMapper, Generate> i
 
         String clothCategory = pythonService.getClothCategory(sketchPath, gender);
 
-        return new GenerateResultVO(generateDetail.getId(), minioUtil.getPreSignedUrl(sketchPath, CommonConstant.MINIO_IMAGE_EXPIRE_TIME), "Success", clothCategory);
+        return new GenerateResultVO(generate.getUniqueId(), generateDetail.getId(),
+                minioUtil.getPreSignedUrl(sketchPath, CommonConstant.MINIO_IMAGE_EXPIRE_TIME), "Success", clothCategory);
     }
 
     public void doCreditsSubtract(Long accountId, CreditsEventsEnum event){
diff --git a/src/main/java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java b/src/main/java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java
index 215cbf3c..4165307e 100644
--- a/src/main/java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java
+++ b/src/main/java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java
@@ -382,6 +382,18 @@ public class UserLikeGroupServiceImpl extends ServiceImpl<UserLikeGroupMapper, U
             toProductImageRecord.setUserLikeGroupId(userLikeGroupId);
         }
 
+        // 翻译
+        String prompt = toProductImageDTO.getPrompt();
+        StringBuilder sb = new StringBuilder("The best quality, masterpiece, real image.");
+        if (!StringUtil.isNullOrEmpty(prompt)) {
+            prompt = pythonService.promptTranslate(prompt);
+        }
+
+        String[] words = prompt.split("\\s+");
+        if (words.length > 200) {
+            throw new BusinessException("Please keep your input text under 200 words. Thanks!");
+        }
+
         toProductImageRecord.setCreateTime(LocalDateTime.now());
         if (!StringUtils.isEmpty(toProductImageDTO.getPrompt())) {
             toProductImageRecord.setPrompt(toProductImageDTO.getPrompt());
@@ -391,12 +403,7 @@ public class UserLikeGroupServiceImpl extends ServiceImpl<UserLikeGroupMapper, U
         List<ToProductImageResult> result = new ArrayList<>();
 
         int i = 0;
-        // 翻译
-        String prompt = toProductImageDTO.getPrompt();
-        StringBuilder sb = new StringBuilder("The best quality, masterpiece, real image.");
-        if (!StringUtil.isNullOrEmpty(prompt)) {
-            prompt = pythonService.promptTranslate(prompt);
-        }
+
 //        else {
 //            s = "best quality, masterpiece. detailed, high-res, simple background, studio photography, extremely detailed, updo, detailed face, face, close-up, HDR, UHD, 8K realistic, Highly detailed, simple background, Studio lighting";
 //        }
@@ -901,6 +908,20 @@ public class UserLikeGroupServiceImpl extends ServiceImpl<UserLikeGroupMapper, U
             toProductImageRecord.setUserLikeGroupId(userLikeGroupId);
         }
 
+        // 翻译
+        String prompt = toProductImageDTO.getPrompt();
+        String s = "";
+        if (!StringUtil.isNullOrEmpty(prompt)) {
+            s = pythonService.promptTranslate(prompt);
+        }else {
+            s = "Snow moutain, snowy day, natural light";
+        }
+        // 对输入文本进行长度限制
+        String[] words = prompt.split("\\s+");
+        if (words.length > 200) {
+            throw new BusinessException("Please keep your input text under 200 words. Thanks!");
+        }
+
         toProductImageRecord.setCreateTime(LocalDateTime.now());
         if (!StringUtils.isEmpty(toProductImageDTO.getPrompt())) {
             toProductImageRecord.setPrompt(toProductImageDTO.getPrompt());
@@ -910,14 +931,7 @@ public class UserLikeGroupServiceImpl extends ServiceImpl<UserLikeGroupMapper, U
         List<ToProductImageResult> result = new ArrayList<>();
 
         int i = 0;
-        // 翻译
-        String prompt = toProductImageDTO.getPrompt();
-        String s = "";
-        if (!StringUtil.isNullOrEmpty(prompt)) {
-            s = pythonService.promptTranslate(prompt);
-        }else {
-            s = "Snow moutain, snowy day, natural light";
-        }
+
         for (ToProductImageVO toProductImageVO : toProductImageDTO.getToProductImageVOList()) {
             String taskId = UUID.randomUUID() + "-" + i + "-" + userHolder.getId();
             i ++;