Update app.py

app.py

@@ -176,12 +176,12 @@ def add_custom_lora(custom_lora):
                 existing_item_index = len(loras)
                 loras.append(new_item)
         
-            return gr.update(visible=True, value=card), gr.update(visible=True), gr.Gallery(selected_index=None), f"Custom: {path}", existing_item_index
+            return gr.update(visible=True, value=card), gr.update(visible=True), gr.Gallery(selected_index=None), f"Custom: {path}", existing_item_index, trigger_word
         except Exception as e:
             gr.Warning(f"Invalid LoRA: either you entered an invalid link, or a non-FLUX LoRA")
-            return gr.update(visible=True, value=f"Invalid LoRA: either you entered an invalid link, a non-FLUX LoRA"), gr.update(visible=True), gr.update(), "", None
+            return gr.update(visible=True, value=f"Invalid LoRA: either you entered an invalid link, a non-FLUX LoRA"), gr.update(visible=True), gr.update(), "", None, ""
     else:
-        return gr.update(visible=False), gr.update(visible=False), gr.update(), "", None
+        return gr.update(visible=False), gr.update(visible=False), gr.update(), "", None, ""
 
 def remove_custom_lora():
     return gr.update(visible=False), gr.update(visible=False), gr.update(), "", None, ""
@@ -252,7 +252,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as app:
     custom_lora.input(
         add_custom_lora,
         inputs=[custom_lora],
-        outputs=[custom_lora_info, custom_lora_button, gallery, selected_info, selected_index]
+        outputs=[custom_lora_info, custom_lora_button, gallery, selected_info, selected_index, prompt]
     )
     custom_lora_button.click(
         remove_custom_lora,