Update app.py

app.py

@@ -15,6 +15,9 @@ import html
 import re
 from threading import Thread
 
+from dotenv import load_dotenv
+load_dotenv()
+
 class Prodia:
     def __init__(self, api_key, base=None):
         self.base = base or "https://api.prodia.com/v1"
@@ -102,7 +105,7 @@ 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>"
+        result = f"{current_prompt} <lora:{lorabtn}:1>"
     return result
 
 def create_grid(image_urls):
@@ -340,8 +343,8 @@ with gr.Blocks(css=css) as demo:
                         loralist = prodia_client.list_loras()
                         with gr.Row():
                             for lora in loralist:
-                                lora_btn = gr.Button(lora)
-                                lora_btn.click(place_lora, inputs=[lora_btn], outputs=prompt)
+                                lora_btn = gr.Button(lora, size="sm")
+                                lora_btn.click(place_lora, inputs=[prompt, 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)