From 26f50c2acb60e2eec92b499785b218fbb392d9ec Mon Sep 17 00:00:00 2001
From: xupei <xupei3360@163.com>
Date: Thu, 19 Dec 2024 15:26:27 +0800
Subject: [PATCH] =?UTF-8?q?generate=20print=E7=BF=BB=E8=AF=91=E8=BE=93?=
 =?UTF-8?q?=E5=85=A5=E4=BF=AE=E6=94=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../java/com/ai/da/service/impl/GenerateServiceImpl.java | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

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 55900f43..fe8f7994 100644
--- a/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
+++ b/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
@@ -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";