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

BUGFIX:generate prompt modify

Browse Source
This commit is contained in:
xupei
2024-04-23 11:40:02 +08:00
parent 3114d3cb36
commit d000eaa6b8
1 changed files with 15 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -123,6 +123,9 @@ public class GenerateServiceImpl extends ServiceImpl<GenerateMapper, Generate> i
String text = generateThroughImageTextDTO.getText(); String text = generateThroughImageTextDTO.getText();
Long elementId = generateThroughImageTextDTO.getCollectionElementId(); Long elementId = generateThroughImageTextDTO.getCollectionElementId();
validateGeneraType(generate, text, elementId, generateType); validateGeneraType(generate, text, elementId, generateType);
if (generateType.equals("text") || generateType.equals("text-image")){
text = modifyPrompt(text, generate, generateThroughImageTextDTO.getLevel1Type());
}
// 2.1 sketch或print在t_collection_element表/t_library表中的信息是否需要更新 如 level2Type // 2.1 sketch或print在t_collection_element表/t_library表中的信息是否需要更新 如 level2Type
CollectionElement collectionElement = collectionElementService.editLevel2Type(elementId, generateThroughImageTextDTO.getLevel2Type(), generateThroughImageTextDTO.getDesignType()); CollectionElement collectionElement = collectionElementService.editLevel2Type(elementId, generateThroughImageTextDTO.getLevel2Type(), generateThroughImageTextDTO.getDesignType());
@@ -232,39 +235,41 @@ public class GenerateServiceImpl extends ServiceImpl<GenerateMapper, Generate> i
if (StringUtil.isNullOrEmpty(text)) { if (StringUtil.isNullOrEmpty(text)) {
throw new BusinessException("please.input.the.caption"); throw new BusinessException("please.input.the.caption");
} }
// generate.setText(text); generate.setText(text);
modifyPrompt(text, generate);
break; break;
case "image": case "image":
if (Objects.isNull(elementId)) { if (Objects.isNull(elementId)) {
throw new BusinessException("please.choose.an.image"); throw new BusinessException("please.choose.an.image");
} }
generate.setElementId(elementId); generate.setElementId(elementId);
modifyPrompt(text, generate);
break; break;
case "text-image": case "text-image":
if (StringUtil.isNullOrEmpty(text) || Objects.isNull(elementId)) { if (StringUtil.isNullOrEmpty(text) || Objects.isNull(elementId)) {
throw new BusinessException("please.input.the.caption.and.choose.an.image"); throw new BusinessException("please.input.the.caption.and.choose.an.image");
} }
// generate.setText(text); generate.setText(text);
generate.setElementId(elementId); generate.setElementId(elementId);
modifyPrompt(text, generate);
default: default:
} }
} }
private void modifyPrompt(String userInput, Generate generate){ private String modifyPrompt(String userInput, Generate generate, String level1Type){
switch (generate.getLevel1Type()) { String text = "";
switch (level1Type) {
case "Moodboard": case "Moodboard":
generate.setText(userInput + ",high quality"); text = userInput + ",high quality";
generate.setText(text);
break; break;
case "Printboard": case "Printboard":
generate.setText(userInput + ", fabric print, high quality"); text = userInput + ", fabric print, high quality";
generate.setText(text);
break; break;
case "Sketchboard": case "Sketchboard":
generate.setText("a single item of " + userInput + "with clean background, hand-drawing sketch style, high quality"); text = "a single item of " + userInput + "with clean background, hand-drawing sketch style, high quality";
generate.setText(text);
default: default:
} }
return text;
} }
@Override @Override