Update app.py

app.py

@@ -9,44 +9,36 @@ import os
 with open('loras.json', 'r') as f:
     loras = json.load(f)
 
-# API call function
-def query(payload, api_url, token):
+# Define the function to run when the button is clicked
+def run_lora(prompt, selected_state):
+    if not selected_state:
+        raise gr.Error("You must select a LoRA")
+    
+    selected_lora_index = selected_state['index']
+    selected_lora = loras[selected_lora_index]
+    api_url = f"https://api-inference.huggingface.co/models/{selected_lora['repo']}"
+    trigger_word = selected_lora["trigger_word"]
+    token = os.getenv("API_TOKEN")
+    payload = {"inputs": f"{prompt} {trigger_word}"}
+    
+    # API call
     headers = {"Authorization": f"Bearer {token}"}
-    print(f"Sending API request with payload: {payload}")
     response = requests.post(api_url, headers=headers, json=payload)
     if response.status_code == 200:
-        return io.BytesIO(response.content)
+        return Image.open(io.BytesIO(response.content))
     else:
-        print(f"API Error: {response.text}")
-        return None
-
-# Define the function to run when the button is clicked
-def run_lora(prompt, selected_lora_index):
-    if selected_lora_index is not None:
-        selected_lora_index = selected_lora_index[0]  # Unpack the tuple
-        selected_lora = loras[selected_lora_index]
-        api_url = f"https://api-inference.huggingface.co/models/{selected_lora['repo']}"
-        trigger_word = selected_lora["trigger_word"]
-        token = os.getenv("API_TOKEN")
-        payload = {"inputs": f"{prompt} {trigger_word}"}
-        
-        # API call
-        image_bytes = query(payload, api_url, token)
-        
-        if image_bytes:
-            return Image.open(image_bytes)
-    return "API Error or No LoRA selected"
+        return "API Error"
 
 # Placeholder for gallery.select function
 def update_selection(selected=None):
     if selected is None:
-        return None,
-    return selected[0],
+        return None
+    return {'index': selected[0]},
 
 # Gradio UI
 with gr.Blocks(css="custom.css") as app:
     title = gr.HTML("<h1>LoRA the Explorer</h1>")
-    selected_state = gr.State(0)  # Initialize with the index of the first LoRA
+    selected_state = gr.State()  # Initialize with empty state
     with gr.Row():
         gallery = gr.Gallery(
             [(item["image"], item["title"]) for item in loras],