Update app.py

Added Negative Prompting

app.py

@@ -33,7 +33,7 @@ def update_imgbox(choices):
     choices_plus = extend_choices(choices)
     return [gr.Image(None, label=m, visible=(m != 'NA'), elem_id="custom_image") for m in choices_plus]
 
-def gen_fn(model_str, prompt, max_retries=10):
+def gen_fn(model_str, prompt, negative_prompt, max_retries=10):
     if model_str == 'NA':
         return None
     
@@ -41,7 +41,7 @@ def gen_fn(model_str, prompt, max_retries=10):
     while retries < max_retries:
         try:
             noise = str(randint(0, 9999999))
-            result = models_load[model_str](f'{prompt} {noise}')
+            result = models_load[model_str](f'{prompt} {noise}', negative_prompt=negative_prompt)
             return result
         except Exception as e:
             print(f"Error generating image: {e}")
@@ -55,6 +55,7 @@ def make_me():
     with gr.Row():
         with gr.Column(scale=1):
             txt_input = gr.Textbox(label='Your prompt:', lines=3, container=False, elem_id="custom_textbox", placeholder="Prompt")
+            negative_txt_input = gr.Textbox(label='Negative prompt:', lines=3, container=False, elem_id="custom_negative_textbox", placeholder="Negative Prompt")
             with gr.Row():
                 gen_button = gr.Button('Generate images', elem_id="custom_gen_button")
                 stop_button = gr.Button('Stop', variant='secondary', interactive=False, elem_id="custom_stop_button")
@@ -72,7 +73,7 @@ def make_me():
         output = [gr.Image(label=m, min_width=250, height=250, elem_id="custom_image") for m in default_models]
         current_models = [gr.Textbox(m, visible=False) for m in default_models]
         for m, o in zip(current_models, output):
-            gen_event = gen_button.click(gen_fn, [m, txt_input], o)
+            gen_event = gen_button.click(gen_fn, [m, txt_input, negative_txt_input], o)
             stop_button.click(on_stop_click, inputs=None, outputs=[gen_button, stop_button], cancels=[gen_event])
     
     with gr.Accordion('Model selection', elem_id="custom_accordion"):
@@ -120,7 +121,7 @@ body {
     box-sizing: border-box;
     margin-bottom: 20px;
 }
-.custom_textbox {
+.custom_textbox, .custom_negative_textbox {
     background-color: #2d343f;
     border: 1px solid #3b4252;
     color: #7f8184;
@@ -227,7 +228,7 @@ body {
         margin: 0;
         padding: 10px;
     }
-   .custom_textbox,.custom_image,.custom_checkbox_group {
+   .custom_textbox, .custom_negative_textbox, .custom_image, .custom_checkbox_group {
         width: 100%;
         box-sizing: border-box;
     }