Update app.py

app.py

@@ -72,20 +72,21 @@ def generate_random_string(length):
     return ''.join(random.choice(characters) for _ in range(length))
 
 @spaces.GPU(duration=120)
-def Piper(_do):
+def Piper(_do,_dont):
     return pipe(
         _do,
         height=512,
         width=768,
-        negative_prompt='ugly, deformed, disfigured, poor details, bad anatomy, labels, texts, logos',
+        negative_prompt=_dont,
         num_inference_steps=40,
         guidance_scale=10
     )
 
-def infer(prompt):
+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()
-    image = Piper(_do).images[0].save(name)
+    _dont = f'{translate(prompt2,"english")}, ugly, deformed, disfigured, poor details, bad anatomy, labels, text, logo'
+    image = Piper(_do,_dont).images[0].save(name)
     return name
 
 css="""
@@ -104,6 +105,7 @@ css="""
 js="""
 function custom(){
     document.querySelector("div#prompt input").setAttribute("maxlength","38");
+    document.querySelector("div#prompt2 input").setAttribute("maxlength","38");
 }
 """
 
@@ -121,7 +123,14 @@ with gr.Blocks(theme=gr.themes.Soft(),css=css,js=js) as demo:
         with gr.Row():
             prompt = gr.Textbox(
                 elem_id="prompt",
-                placeholder="Describe the photo",
+                placeholder="Photo Description",
+                container=False,
+                rtl=True,
+                max_lines=1
+            )
+            prompt2 = gr.Textbox(
+                elem_id="prompt2",
+                placeholder="Forbidden Content",
                 container=False,
                 rtl=True,
                 max_lines=1
@@ -129,14 +138,9 @@ 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],
-        outputs = [result]
-    )
     run_button.click(
         fn = infer,
-        inputs = [prompt],
+        inputs = [prompt,prompt2],
         outputs = [result]
     )