Update app.py

app.py

@@ -98,8 +98,12 @@ def remove_id_and_ext(text):
         text = text[:-4]
     return text
 
-def place_lora(lorabtn):
-    return lorabtn
+def place_lora(current_prompt, lorabtn):
+    if f"<lora:{lorabtn}:1.0>" in current_prompt:
+        result = current_prompt.replace(f"<lora:{lorabtn}:1.0>", "")
+    else:
+        result = f"{current_prompt}, <lora:{lorabtn}:1.0>"
+    return result
 
 def create_grid(image_urls):
     # Download first image to get size
@@ -334,9 +338,10 @@ with gr.Blocks(css=css) as demo:
 
                     with gr.Tab("Lora"):
                         loralist = prodia_client.list_loras()
-                        for lora in loralist:
-                            lora_btn = gr.Button(lora)
-                            lora_btn.click(place_lora, inputs=[lora_btn], outputs=prompt)
+                        with gr.Row():
+                            for lora in loralist:
+                                lora_btn = gr.Button(lora)
+                                lora_btn.click(place_lora, inputs=[lora_btn], outputs=prompt)
 
                 with gr.Column(scale=2):
                     image_output = gr.Gallery(value=["https://images.prodia.xyz/8ede1a7c-c0ee-4ded-987d-6ffed35fc477.png"], preview=True)