Update app.py

app.py

@@ -169,7 +169,7 @@ for model in model_tags:
   if model != "ahx-model-1" and model != "ahx-model-2":
     DROPDOWNS[model] = f" in the style of <{model}>"
 
-# def image_prompt(prompt, dropdown, guidance, steps, seed, height, width):
+# def image_prompt(prompt, dropdown, guidance, steps, seed, height, width, negative_prompt=""):
 def image_prompt(prompt, guidance, steps, seed, height, width):
   # prompt = prompt + DROPDOWNS[dropdown]
   square_pixels = height * width
@@ -194,8 +194,8 @@ def image_prompt(prompt, guidance, steps, seed, height, width):
 
   if is_clean:
     return (
-      pipe(prompt=prompt, guidance_scale=guidance, num_inference_steps=steps, generator=generator, height=height, width=width).images[0], 
-      f"prompt: '{prompt}', seed = {int(seed)},\nheight: {height}px, width: {width}px,\nguidance: {guidance}, steps: {steps}"
+      pipe(prompt=prompt, guidance_scale=guidance, num_inference_steps=steps, generator=generator, height=height, width=width, negative_prompt=negative_prompt).images[0], 
+      f"prompt: '{prompt}', seed = {int(seed)},\nheight: {height}px, width: {width}px,\nguidance: {guidance}, steps: {steps}, negative prompt: {negative_prompt}"
     )
   else:
     return (
@@ -248,6 +248,7 @@ with gr.Blocks(css=".gradio-container {max-width: 650px}") as advanced_tab:
     prompt = gr.Textbox(label="image prompt...", elem_id="input-text")
   with gr.Row():
     seed = gr.Slider(0, 99999999999999, label="seed", dtype=int, value=random_seed, interactive=True, step=1)
+    negative_prompt = gr.Textbox(label="negative prompt (optional)", elem_id="input-text")
   with gr.Row():
     with gr.Column():
       guidance = gr.Slider(0, 10, label="guidance", dtype=float, value=default_guidance, step=0.1, interactive=True)
@@ -263,7 +264,7 @@ with gr.Blocks(css=".gradio-container {max-width: 650px}") as advanced_tab:
   go_button = gr.Button("generate image", elem_id="go-button")
   output = gr.Image(elem_id="output-image")
   output_text = gr.Text(elem_id="output-text")
-  go_button.click(fn=image_prompt, inputs=[prompt, guidance, steps, seed, height, width], outputs=[output, output_text])
+  go_button.click(fn=image_prompt, inputs=[prompt, guidance, steps, seed, height, width, negative_prompt], outputs=[output, output_text])
   gr.Markdown("For a complete list of usable models and beta concepts check out the dropdown selectors in the welcome and beta concepts tabs or the project's main website or our discord.\n\nhttp://www.astronaut.horse/concepts")