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

输入prompt修饰

Browse Source
This commit is contained in:
xupei
2024-04-22 15:32:59 +08:00
parent 825439255b
commit 19d170c1bf
1 changed files with 19 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -232,20 +232,37 @@ public class GenerateServiceImpl extends ServiceImpl<GenerateMapper, Generate> i
if (StringUtil.isNullOrEmpty(text)) {
throw new BusinessException("please.input.the.caption");
}
generate.setText(text);
// generate.setText(text);
modifyPrompt(text, generate);
break;
case "image":
if (Objects.isNull(elementId)) {
throw new BusinessException("please.choose.an.image");
}
generate.setElementId(elementId);
modifyPrompt(text, generate);
break;
case "text-image":
if (StringUtil.isNullOrEmpty(text) || Objects.isNull(elementId)) {
throw new BusinessException("please.input.the.caption.and.choose.an.image");
}
generate.setText(text);
// generate.setText(text);
generate.setElementId(elementId);
modifyPrompt(text, generate);
default:
}
}
private void modifyPrompt(String userInput, Generate generate){
switch (generate.getLevel1Type()) {
case "Moodboard":
generate.setText(userInput + ",high quality");
break;
case "Printboard":
generate.setText(userInput + ", fabric print, high quality");
break;
case "Sketchboard":
generate.setText("a single item of " + userInput + "with clean background, hand-drawing sketch style, high quality");
default:
}
}