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";