Update app.py

app.py

@@ -99,7 +99,7 @@ def update_selection(evt: gr.SelectData, width, height):
     )
 
 @spaces.GPU(duration=70)
-def generate_image(prompt, trigger_word, steps, seed, cfg_scale, width, height, lora_scale, progress):
+def generate_image(prompt, trigger_word, steps, seed, cfg_scale, width, height, negative_prompt, lora_scale, progress):
     pipe.to("cuda")
     generator = torch.Generator(device="cuda").manual_seed(seed)
     
@@ -107,6 +107,7 @@ def generate_image(prompt, trigger_word, steps, seed, cfg_scale, width, height,
         # Generate image
         image = pipe(
             prompt=f"{prompt} {trigger_word}",
+            negative_prompt=negative_prompt,
             num_inference_steps=steps,
             guidance_scale=cfg_scale,
             width=width,
@@ -116,7 +117,9 @@ def generate_image(prompt, trigger_word, steps, seed, cfg_scale, width, height,
         ).images[0]
     return image
 
-def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, width, height, lora_scale, progress=gr.Progress(track_tqdm=True)):
+def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, width, height, negative_prompt, lora_scale, progress=gr.Progress(track_tqdm=True)):
+    if negative_prompt == "":
+        negative_prompt = None    
     if selected_index is None:
         raise gr.Error("You must select a LoRA before proceeding.")
 
@@ -136,7 +139,7 @@ def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, wid
         if randomize_seed:
             seed = random.randint(0, MAX_SEED)
     
-    image = generate_image(prompt, trigger_word, steps, seed, cfg_scale, width, height, lora_scale, progress)
+    image = generate_image(prompt, trigger_word, steps, seed, cfg_scale, width, height, negative_prompt, lora_scale, progress)
     pipe.to("cpu")
     pipe.unload_lora_weights()
     return image, seed  
@@ -167,6 +170,9 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as app:
     with gr.Row():
         with gr.Column(scale=3):
             prompt = gr.Textbox(label="Prompt", lines=1, placeholder="Select LoRa/Style & type prompt!")
+    with gr.Row():
+        with gr.Column(scale=3):
+            negative_prompt = gr.Textbox(label="Negative Prompt", lines=1, placeholder="List unwanted conditions, open-fluxedly!")
         with gr.Column(scale=1, elem_id="gen_column"):
             generate_button = gr.Button("Generate", variant="primary", elem_id="gen_btn")
     with gr.Row():
@@ -208,7 +214,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as app:
     gr.on(
         triggers=[generate_button.click, prompt.submit],
         fn=run_lora,
-        inputs=[prompt, cfg_scale, steps, selected_index, randomize_seed, seed, width, height, lora_scale],
+        inputs=[prompt, cfg_scale, steps, selected_index, randomize_seed, seed, width, height, negative_prompt, lora_scale],
         outputs=[result, seed]
     )