aidlab/aida_back
7
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

TASK: 当选择童装时，sketch生成和to product的prompt需添加儿童限定词

Browse Source
This commit is contained in:
xupei
2025-06-17 14:41:27 +08:00
parent 8d7b4adf48
commit 5abe45ff34
4 changed files with 13 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/main/java/com/ai/da/model/dto/GenerateThroughImageTextDTO.java
View File

@@ -58,4 +58,7 @@ public class GenerateThroughImageTextDTO {
@ApiModelProperty("种子 取值范围 0~500")
private String seed;
@ApiModelProperty("年龄区间 Adult || Child")
private String ageGroup = "Adult";
}

1
src/main/java/com/ai/da/model/dto/ToProductImageDTO.java
View File

@@ -15,4 +15,5 @@ public class ToProductImageDTO {
private String direction;
private Double brightenValue;
private String modelName;
private String ageGroup;
}

7
src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
View File

@@ -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:
}

4
src/main/java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java
View File

@@ -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");
}