翻译prompt

app.py

@@ -62,6 +62,20 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
         seed = random.randint(0, MAX_SEED)
     return seed
 
+def translateEN(zh):
+    result = requests.post(
+        "https://api-free.deepl.com/v2/translate", 
+        params={ 
+            "auth_key": "e8b4d428-ada5-3f8d-f965-bad01e8a06c1:fx", 
+            "target_lang": "EN-US", 
+            "text": zh})
+    return result.json()["translations"][0]["text"]
+
+def process_text(prompt):
+    print("中文提示词: \n", prompt)
+    prompt_trans = translateEN(prompt)
+    print("prompt: \n", prompt_trans)
+    return prompt_trans
 
 @spaces.GPU
 def generate(
@@ -284,10 +298,10 @@ with gr.Blocks(css="style.css") as demo:
     ).then(
         fn=generate,
         inputs=[
-            prompt,
-            negative_prompt,
-            prompt_2,
-            negative_prompt_2,
+            process_text(prompt),
+            process_text(negative_prompt),
+            process_text(prompt_2),
+            process_text(negative_prompt_2),
             use_negative_prompt,
             use_prompt_2,
             use_negative_prompt_2,