输入prompt修饰

src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java

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