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 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 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 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