Update app.py

app.py

@@ -59,21 +59,21 @@ def generate_random_string(length):
     return ''.join(random.choice(characters) for _ in range(length))
 
 @spaces.GPU
-def Piper(english_prompt):
+def Piper(_do, _dont):
     return pipe(
-        english_prompt,
+        _do,
         height=512,
         width=512,
-        negative_prompt="",
+        negative_prompt=_dont,
         num_inference_steps=400,
-        guidance_scale=9
+        guidance_scale=10
     )
 
-def infer(prompt):
+def infer(prompt1,prompt2,prompt3):
     name = generate_random_string(12)+".png"
-    english_prompt = f'REAL {translate(prompt,"english").upper()}:'
-    print(f'Final prompt: {english_prompt}')
-    image = Piper(english_prompt).images[0].save(name)
+    _do = f'BACKGROUND: show an authentic {translate(prompt2,"english")}; FOREGROUND: be focused on a real {translate(prompt1,"english")}.'
+    _dont = f'ANY usage of {translate(prompt3,"english").upper()}...'
+    image = Piper(_do, _dont).images[0].save(name)
     return name
 
 css="""
@@ -98,18 +98,29 @@ with gr.Blocks(css=css) as demo:
             Currently running on {power_device}.
         """)
         with gr.Row():
-            prompt = gr.Text(
-                label="Prompt",
+            prompt2 = gr.Text(
                 show_label=False,
                 max_lines=1,
-                placeholder="Enter your prompt",
+                placeholder="(The Overall Scene)",
+                container=False,
+            )
+            prompt1 = gr.Text(
+                show_label=False,
+                max_lines=1,
+                placeholder="(The Main Content)",
+                container=False,
+            )
+            prompt3 = gr.Text(
+                show_label=False,
+                max_lines=1,
+                placeholder="(What Things Are Forbidden?)",
                 container=False,
             )
             run_button = gr.Button("Run", scale=0)
         result = gr.Image(elem_id="image-container", label="Result", show_label=False, type='filepath')
     run_button.click(
         fn = infer,
-        inputs = [prompt],
+        inputs = [prompt1,prompt2,prompt3],
         outputs = [result]
     )