Update app.py

app.py

@@ -65,14 +65,18 @@ def Piper(_do, _dont):
         height=512,
         width=512,
         negative_prompt=_dont,
-        num_inference_steps=400,
-        guidance_scale=10
+        num_inference_steps=200,
+        guidance_scale=1.1
     )
 
-def infer(prompt1,prompt2,prompt3):
+def infer(prompt1,prompt2,prompt3,prompt4):
     name = generate_random_string(12)+".png"
-    _do = f'Show an authentic {translate(prompt2,"english").upper()} scene, while focusing on the details, of {translate(prompt1,"english").upper()}, as content.'
-    _dont = f'ANY usage of {translate(prompt3,"english").upper()}...'
+    prompt1 = " and ".join([translate(v,"english").upper() for v in prompt1])
+    prompt2 = " and ".join([translate(v,"english").upper() for v in prompt2])
+    prompt3 = " and ".join([translate(v,"english").upper() for v in prompt3])
+    prompt4 = " or ".join([translate(v,"english").upper() for v in prompt4])
+    _do = f'Show an authentic {prompt3} scene, while focusing on the details, of {prompt1}, as the main elements, and, showing {prompt2} in the background.'
+    _dont = f'ANY usage of {prompt4}...'
     image = Piper(_do, _dont).images[0].save(name)
     return name
 
@@ -99,29 +103,51 @@ with gr.Blocks(css=css) as demo:
         """)
         with gr.Column():
             with gr.Row():
-                prompt2 = gr.Text(
-                    max_lines=1,
-                    placeholder="(The Overall Scene)",
+                prompt1 = gr.Dropdown(
+                    multiselect=True,
+                    allow_custom_value=True,
+                    max_choices=3,
+                    label="Foreground Elements",
+                    show_label=True,
                     container=False,
+                    value="any"
                 )
             with gr.Row():
-                prompt1 = gr.Text(
-                    max_lines=1,
-                    placeholder="(The Content/Details)",
+                prompt2 = gr.Dropdown(
+                    multiselect=True,
+                    allow_custom_value=True,
+                    max_choices=4,
+                    label="Background Elements",
+                    show_label=True,
                     container=False,
+                    value="any"
                 )
             with gr.Row():
-                prompt3 = gr.Text(
-                    max_lines=1,
-                    placeholder="(What Things Are Forbidden?)",
+                prompt3 = gr.Dropdown(
+                    multiselect=True,
+                    allow_custom_value=True,
+                    max_choices=2,
+                    label="Background Events",
+                    show_label=True,
                     container=False,
+                    value="any"
+                )
+            with gr.Row():
+                prompt4 = gr.Dropdown(
+                    multiselect=True,
+                    allow_custom_value=True,
+                    max_choices=6,
+                    label="Forbidden Elements/Events",
+                    show_label=True,
+                    container=False,
+                    value="None"
                 )
             with gr.Row():
                 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 = [prompt1,prompt2,prompt3],
+        inputs = [prompt1,prompt2,prompt3,prompt4],
         outputs = [result]
     )