Update app.py

app.py

@@ -12,24 +12,35 @@ with open('loras.json', 'r') as f:
 # API call function
 def query(payload, api_url, token):
     headers = {"Authorization": f"Bearer {token}"}
-    print(f"Sending API request with payload: {payload}")  # Print the API request payload
+    print(f"Sending API request with payload: {payload}")
     response = requests.post(api_url, headers=headers, json=payload)
-    return io.BytesIO(response.content)
+    if response.status_code == 200:
+        return 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 = loras[0]
+def run_lora(prompt, selected_lora_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}"}
     image_bytes = query(payload, api_url, token)
-    return Image.open(image_bytes)
+    if image_bytes:
+        return Image.open(image_bytes)
+    else:
+        return "API Error"
+
+# Placeholder for gallery.select function
+def update_selection(selected):
+    return selected
 
 # Gradio UI
 with gr.Blocks(css="custom.css") as app:
     title = gr.HTML("<h1>LoRA the Explorer</h1>")
-    selected_state = gr.State()
+    selected_state = gr.State(0)  # Initialize with the index of the first LoRA
     with gr.Row():
         gallery = gr.Gallery(
             [(item["image"], item["title"]) for item in loras],
@@ -39,26 +50,16 @@ with gr.Blocks(css="custom.css") as app:
         )
         with gr.Column():
             prompt_title = gr.Markdown("### Click on a LoRA in the gallery to select it")
-            with gr.Row():
-                prompt = gr.Textbox(label="Prompt", show_label=False, lines=1, max_lines=1, placeholder="Type a prompt after selecting a LoRA")
-                button = gr.Button("Run")
+            prompt = gr.Textbox(label="Prompt", show_label=False, lines=1, max_lines=1, placeholder="Type a prompt after selecting a LoRA")
             result = gr.Image(interactive=False, label="Generated Image")
 
-    # Placeholder for gallery.select function
-    def update_selection():
-        pass
-
     gallery.select(
-        update_selection
+        update_selection,
+        outputs=[selected_state]
     )
     prompt.submit(
         fn=run_lora,
-        inputs=[prompt],
-        outputs=[result]
-    )
-    button.click(
-        fn=run_lora,
-        inputs=[prompt],
+        inputs=[prompt, selected_state],
         outputs=[result]
     )