Update app.py

app.py

@@ -65,19 +65,23 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
     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"]
+    if 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
+    if prompt:
+        print("中文提示词: \n", prompt)
+        prompt_trans = translateEN(prompt)
+        print("prompt: \n", prompt_trans)
+        return prompt_trans
+
+process_text("外面做一个测试")
 
 @spaces.GPU
 def generate(
@@ -106,12 +110,18 @@ def generate(
     if not use_negative_prompt_2:
         negative_prompt_2 = None  # type: ignore
 
+    process_text("里面做一个测试")
+    print("prompt是：", prompt)
+    print("negative_prompt是：", negative_prompt)
+    print("prompt_2是：", prompt_2)
+    print("negative_prompt_2是：", negative_prompt_2)
+    
     if not apply_refiner:
         return pipe(
-            prompt=process_text(prompt),
-            negative_prompt=process_text(negative_prompt),
-            prompt_2=process_text(prompt_2),
-            negative_prompt_2=process_text(negative_prompt_2),
+            prompt=prompt,
+            negative_prompt=negative_prompt,
+            prompt_2=prompt_2,
+            negative_prompt_2=negative_prompt_2,
             width=width,
             height=height,
             guidance_scale=guidance_scale_base,
@@ -121,10 +131,10 @@ def generate(
         ).images[0]
     else:
         latents = pipe(
-            prompt=process_text(prompt),
-            negative_prompt=process_text(negative_prompt),
-            prompt_2=process_text(prompt_2),
-            negative_prompt_2=process_text(negative_prompt_2),
+            prompt=prompt,
+            negative_prompt=negative_prompt,
+            prompt_2=prompt_2,
+            negative_prompt_2=negative_prompt_2,
             width=width,
             height=height,
             guidance_scale=guidance_scale_base,
@@ -133,10 +143,10 @@ def generate(
             output_type="latent",
         ).images
         image = refiner(
-            prompt=process_text(prompt),
-            negative_prompt=process_text(negative_prompt),
-            prompt_2=process_text(prompt_2),
-            negative_prompt_2=process_text(negative_prompt_2),
+            prompt=prompt,
+            negative_prompt=negative_prompt,
+            prompt_2=prompt_2,
+            negative_prompt_2=negative_prompt_2,
             guidance_scale=guidance_scale_refiner,
             num_inference_steps=num_inference_steps_refiner,
             image=latents,