From d000eaa6b8f31018cf0bb87f7ebc355de56e9795 Mon Sep 17 00:00:00 2001
From: xupei <xupei3360@163.com>
Date: Tue, 23 Apr 2024 11:40:02 +0800
Subject: [PATCH] BUGFIX:generate prompt modify

---
 .../da/service/impl/GenerateServiceImpl.java  | 25 +++++++++++--------
 1 file changed, 15 insertions(+), 10 deletions(-)

diff --git a/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java b/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
index d0613304..ba286732 100644
--- a/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
+++ b/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
@@ -123,6 +123,9 @@ public class GenerateServiceImpl extends ServiceImpl<GenerateMapper, Generate> i
         String text = generateThroughImageTextDTO.getText();
         Long elementId = generateThroughImageTextDTO.getCollectionElementId();
         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
         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)) {
                     throw new BusinessException("please.input.the.caption");
                 }
-//                generate.setText(text);
-                modifyPrompt(text, generate);
+                generate.setText(text);
                 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()) {
+    private String modifyPrompt(String userInput, Generate generate, String level1Type){
+        String text = "";
+        switch (level1Type) {
             case "Moodboard":
-                generate.setText(userInput + ",high quality");
+                text = userInput + ",high quality";
+                generate.setText(text);
                 break;
             case "Printboard":
-                generate.setText(userInput + ", fabric print, high quality");
+                text = userInput + ", fabric print, high quality";
+                generate.setText(text);
                 break;
             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:
         }
+        return text;
     }
 
     @Override