Update app.py

app.py

@@ -40,8 +40,28 @@ def get_image_size(aspect_ratio):
     else:
         return 640, 640
 
-@spaces.GPU(duration=45)
+def polish_prompt_en(original_prompt):
+    
+    SYSTEM_PROMPT = open("improve_prompt.txt").read()
+    
+    original_prompt = original_prompt.strip()
+    prompt = f"{SYSTEM_PROMPT}\n\nUser Input: {original_prompt}\n\n Rewritten Prompt:"
+    success=False
+    
+    while not success:
+        try:
+            polished_prompt = api(prompt, model='qwen-plus')
+            polished_prompt = polished_prompt.strip()
+            polished_prompt = polished_prompt.replace("\n", " ")
+            success = True
+            
+        except Exception as e:
+            print(f"Error during API call: {e}")
+            
+    return polished_prompt
+
 
+@spaces.GPU(duration=45)
 def infer(
     prompt,
     negative_prompt=" ",