From 5abe45ff34ef81bbc69464351949486cfbbdcef1 Mon Sep 17 00:00:00 2001
From: xupei <xupei3360@163.com>
Date: Tue, 17 Jun 2025 14:41:27 +0800
Subject: [PATCH] =?UTF-8?q?TASK:=20=E5=BD=93=E9=80=89=E6=8B=A9=E7=AB=A5?=
 =?UTF-8?q?=E8=A3=85=E6=97=B6=EF=BC=8Csketch=E7=94=9F=E6=88=90=E5=92=8Cto?=
 =?UTF-8?q?=20product=E7=9A=84prompt=E9=9C=80=E6=B7=BB=E5=8A=A0=E5=84=BF?=
 =?UTF-8?q?=E7=AB=A5=E9=99=90=E5=AE=9A=E8=AF=8D?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../com/ai/da/model/dto/GenerateThroughImageTextDTO.java   | 3 +++
 src/main/java/com/ai/da/model/dto/ToProductImageDTO.java   | 1 +
 .../java/com/ai/da/service/impl/GenerateServiceImpl.java   | 7 +++++--
 .../com/ai/da/service/impl/UserLikeGroupServiceImpl.java   | 4 ++++
 4 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/src/main/java/com/ai/da/model/dto/GenerateThroughImageTextDTO.java b/src/main/java/com/ai/da/model/dto/GenerateThroughImageTextDTO.java
index 19266a53..d245efc9 100644
--- a/src/main/java/com/ai/da/model/dto/GenerateThroughImageTextDTO.java
+++ b/src/main/java/com/ai/da/model/dto/GenerateThroughImageTextDTO.java
@@ -58,4 +58,7 @@ public class GenerateThroughImageTextDTO {
 
     @ApiModelProperty("种子 取值范围 0~500")
     private String seed;
+
+    @ApiModelProperty("年龄区间 Adult || Child")
+    private String ageGroup = "Adult";
 }
diff --git a/src/main/java/com/ai/da/model/dto/ToProductImageDTO.java b/src/main/java/com/ai/da/model/dto/ToProductImageDTO.java
index 56c8bbbd..e5212e00 100644
--- a/src/main/java/com/ai/da/model/dto/ToProductImageDTO.java
+++ b/src/main/java/com/ai/da/model/dto/ToProductImageDTO.java
@@ -15,4 +15,5 @@ public class ToProductImageDTO {
     private String direction;
     private Double brightenValue;
     private String modelName;
+    private String ageGroup;
 }
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 3c48d3e4..e9504d5e 100644
--- a/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
+++ b/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
@@ -173,7 +173,7 @@ public class GenerateServiceImpl extends ServiceImpl<GenerateMapper, Generate> i
         Long elementId = generateThroughImageTextDTO.getCollectionElementId();
 //        validateGeneraType(generate, text, elementId);
         if (!StringUtil.isNullOrEmpty(text)) {
-            text = modifyPrompt(text, generate, generateThroughImageTextDTO.getLevel1Type());
+            text = modifyPrompt(text, generate, generateThroughImageTextDTO.getLevel1Type(), generateThroughImageTextDTO.getAgeGroup());
         }
 
         // todo 这一步现在还是有必要的吗？
@@ -352,7 +352,7 @@ public class GenerateServiceImpl extends ServiceImpl<GenerateMapper, Generate> i
 
     }
 
-    private String modifyPrompt(String userInput, Generate generate, String level1Type) {
+    private String modifyPrompt(String userInput, Generate generate, String level1Type, String ageGroup) {
         String text = "";
         String prefix = "";
         if (userInput.startsWith("Painting Style")
@@ -384,6 +384,9 @@ public class GenerateServiceImpl extends ServiceImpl<GenerateMapper, Generate> i
             case "Sketchboard":
 //                text = "clear lines, simple outlines monochrome white vector image of " + translated + ", no background, sketch flat, front view display, best quality, ultra-high resolution 8k";
                 text = "a single item of sketch of " + translated + ", 4k, white background";
+                if (!StringUtil.isNullOrEmpty(ageGroup) && ageGroup.equals("Child")){
+                    text = text + ", Children's clothing";
+                }
 //                generate.setText(text);
             default:
         }
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 4165307e..a2a75235 100644
--- a/src/main/java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java
+++ b/src/main/java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java
@@ -447,6 +447,10 @@ public class UserLikeGroupServiceImpl extends ServiceImpl<UserLikeGroupMapper, U
                     toProductImageResult.setResultType("ToProductImage");
                 } else {
                     // 走模型
+                    if (!StringUtil.isNullOrEmpty(toProductImageDTO.getAgeGroup())
+                            && toProductImageDTO.getAgeGroup().equals("Child")){
+                        sb.append(", Children's face");
+                    }
                     pythonService.toProductImage(tDesignPythonOutfit.getDesignUrl(), taskId, sb.toString(), toProductImageDTO.getImageStrength(), productType);
                     toProductImageResult.setModelName("local");
                 }