Update app.py

app.py

@@ -12,7 +12,7 @@ from pathos.multiprocessing import ProcessingPool as ProcessPoolExecutor
 import requests
 from lxml.html import fromstring
 
-pool = ProcessPoolExecutor(4)
+pool = ProcessPoolExecutor(16)
 pool.__enter__()
 
 #model_id = "runwayml/stable-diffusion-v1-5"
@@ -78,14 +78,20 @@ def Piper(_do,_dont):
         height=512,
         width=768,
         negative_prompt=_dont,
-        num_inference_steps=40,
+        num_inference_steps=100,
         guidance_scale=10
     )
 
 def infer(prompt,prompt2):
     name = generate_random_string(12)+".png"
-    _do = f'{ translate(prompt,"english") }, hot colors palette, muted colors, dynamic poze, realistic, accurate, matched, realistic details, award winning photograph, soft natural lighting, deep field, high definition, highly detailed, 8k'.lower()
-    _dont = f'{translate(prompt2,"english")}, ugly, deformed, disfigured, poor details, bad anatomy, labels, text, logo'
+    if prompt == None or prompt.strip() == "":
+        _do = 'hot colors palette, muted colors, dynamic poze, realistic, accurate, matched, realistic details, award winning photograph, soft natural lighting, deep field, high definition, highly detailed, 8k'
+    else:
+        _do = f'{ translate(prompt,"english") }, hot colors palette, muted colors, dynamic poze, realistic, accurate, matched, realistic details, award winning photograph, soft natural lighting, deep field, high definition, highly detailed, 8k'
+    if prompt2 == None or prompt2.strip() == "":
+        _dont = 'ugly, deformed, disfigured, poor details, bad anatomy, including labels, including text, including logo'
+    else:
+        _dont = f'including {translate(prompt2,"english")}, ugly, deformed, disfigured, poor details, bad anatomy, including labels, including text, including logo'
     image = Piper(_do,_dont).images[0].save(name)
     return name
 
@@ -128,6 +134,7 @@ with gr.Blocks(theme=gr.themes.Soft(),css=css,js=js) as demo:
                 rtl=True,
                 max_lines=1
             )
+        with gr.Row():
             prompt2 = gr.Textbox(
                 elem_id="prompt2",
                 placeholder="Forbidden Content",
@@ -138,6 +145,16 @@ with gr.Blocks(theme=gr.themes.Soft(),css=css,js=js) as demo:
         with gr.Row():
             run_button = gr.Button("Run")
         result = gr.Image(elem_id="image-container", label="Result", show_label=False, type='filepath')
+    prompt.submit(
+        fn = infer,
+        inputs = [prompt,prompt2],
+        outputs = [result]
+    )
+    prompt2.submit(
+        fn = infer,
+        inputs = [prompt,prompt2],
+        outputs = [result]
+    )
     run_button.click(
         fn = infer,
         inputs = [prompt,prompt2],