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

generate print翻译输入修改

Browse Source
This commit is contained in:
xupei
2024-12-19 15:26:27 +08:00
parent 7194049127
commit 26f50c2acb
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -314,7 +314,14 @@ public class GenerateServiceImpl extends ServiceImpl<GenerateMapper, Generate> i
private String modifyPrompt(String userInput, Generate generate, String level1Type) {
String text = "";
String translated = pythonService.promptTranslate(userInput);
String prefix = "";
if (userInput.startsWith("Painting Style")
|| userInput.startsWith("Illustration Style")
|| userInput.startsWith("Real Style") ){
prefix = userInput.substring(0, userInput.indexOf(",")) + ", ";
userInput = userInput.substring(userInput.indexOf(",") + 1);
}
String translated = prefix + pythonService.promptTranslate(userInput);
switch (level1Type) {
case "Moodboard":
text = translated + ",high quality";