Update app.py

app.py

@@ -84,7 +84,7 @@ def Piper(_do):
             _do,
             height=448,
             width=448,
-            num_inference_steps=4,
+            num_inference_steps=8,
             max_sequence_length=256,
             guidance_scale=0
         )
@@ -92,12 +92,14 @@ def Piper(_do):
     except:
         return ""
 
-def infer(prompt_en):
+def infer(p1,p2):
     name = generate_random_string(12)+".png"
-    if prompt_en == "":
-        _do = 'beautiful, playful, photographed,, highly detailed, realistic elements, dynamic poze, deep field, vivid reasonable coloring, rough texture, high sharpness, highres, best quality, focused'
+    if p1 == "":
+        _do = 'beautiful, playful, photographed, highly detailed, realistic elements, dynamic poze, deep field, vivid reasonable coloring, rough texture, high sharpness, highres, best quality, focused'
     else:
-        _do = f'{ prompt_en }, beautiful, playful, photographed, highly detailed, realistic elements, dynamic poze, deep field, vivid reasonable coloring, rough texture, high sharpness, highres, best quality, focused'
+        _do = f'{ p1 }, beautiful, playful, photographed, highly detailed, realistic elements, dynamic poze, deep field, vivid reasonable coloring, rough texture, high sharpness, highres, best quality, focused'
+    if p2 != "":
+        _do = f'{_do}, anti-{p2.replace(" ","-")}!'
     output = Piper(_do)
     if output == "":
         return output
@@ -140,6 +142,7 @@ footer {
 js="""
 function custom(){
     document.querySelector("div#prompt input").setAttribute("maxlength","38")
+    document.querySelector("div#prompt2 input").setAttribute("maxlength","38")
 }
 """
 
@@ -152,7 +155,14 @@ with gr.Blocks(theme=gr.themes.Soft(),css=css,js=js) as demo:
         with gr.Row():
             prompt = gr.Textbox(
                 elem_id="prompt",
-                placeholder="WHAT TO CREATE",
+                placeholder="INCLUDE",
+                container=False,
+                max_lines=1
+            )
+        with gr.Row():
+            prompt2 = gr.Textbox(
+                elem_id="prompt2",
+                placeholder="EXCLUDE",
                 container=False,
                 max_lines=1
             )
@@ -165,17 +175,19 @@ with gr.Blocks(theme=gr.themes.Soft(),css=css,js=js) as demo:
             result.append(gr.Image(interactive=False,elem_classes="image-container", label="Result", show_label=False, type='filepath', show_share_button=False))
             result.append(gr.Image(interactive=False,elem_classes="image-container", label="Result", show_label=False, type='filepath', show_share_button=False))
 
-    def _ret(idx,p1):
-        print(f'Starting {idx}: {p1}')
-        v = infer(p1)
-        print(f'Finished {idx}: {v}')
+    def _ret(idx,p1,p2):
+        print(f'Starting {idx}')
+        v = infer(p1,p2)
+        print(f'Finished {idx}')
         return v
-    def _rets(p1):
+    def _rets(p1,p2):
         p1_en = translate(p1,"english")
+        p2_en = translate(p2,"english")
         ln = len(result)
         idxs = list(range(ln))
         p1s = [p1_en for _ in idxs]
-        return list(Pool(ln).imap( _ret, idxs, p1s ))
-    run_button.click(fn=_rets,inputs=prompt,outputs=result)
+        p2s = [p2_en for _ in idxs]
+        return list(Pool(ln).imap( _ret, idxs, p1s, p2s ))
+    run_button.click(fn=_rets,inputs=[prompt,prompt2],outputs=result)
 
 demo.queue().launch()