Update app.py

app.py

@@ -1,4 +1,3 @@
-# Existing imports
 import gradio as gr
 import requests
 import io
@@ -7,29 +6,19 @@ import json
 import os
 import logging
 
-# Initialize logging
 logging.basicConfig(level=logging.DEBUG)
 
-# Load LoRAs from JSON
 with open('loras.json', 'r') as f:
     loras = json.load(f)
 
 def update_selection(selected_state: gr.SelectData):
     logging.debug(f"Inside update_selection, selected_state: {selected_state}")
     selected_lora_index = selected_state['index']
-    updated_text = loras[selected_lora_index]['title']
-    instance_prompt = "Your custom instance prompt here"
-    new_placeholder = "Your new placeholder here"
-    use_with_diffusers = "Your use_with_diffusers here"
-    use_with_uis = "Your use_with_uis here"
-    logging.debug(f"Updated selected_state: {selected_state}")
+    selected_lora = loras[selected_lora_index]
+    new_placeholder = f"Type a prompt for {selected_lora['title']}"
     return (
-        updated_text,
-        instance_prompt,
         gr.update(placeholder=new_placeholder),
-        selected_state,
-        use_with_diffusers,
-        use_with_uis,
+        selected_state
     )
 
 def run_lora(prompt, selected_state, progress=gr.Progress(track_tqdm=True)):
@@ -37,6 +26,7 @@ def run_lora(prompt, selected_state, progress=gr.Progress(track_tqdm=True)):
     if not selected_state:
         logging.error("selected_state is None or empty.")
         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']}"
@@ -44,10 +34,16 @@ def run_lora(prompt, selected_state, progress=gr.Progress(track_tqdm=True)):
     token = os.getenv("API_TOKEN")
     payload = {"inputs": f"{prompt} {trigger_word}"}
 
-# Gradio UI
+    headers = {"Authorization": f"Bearer {token}"}
+    response = requests.post(api_url, headers=headers, json=payload)
+    if response.status_code == 200:
+        return Image.open(io.BytesIO(response.content))
+    else:
+        return "API Error"
+
 with gr.Blocks(css="custom.css") as app:
     title = gr.HTML("<h1>LoRA the Explorer</h1>")
-    selected_state = gr.State()  # Initialize with empty state
+    selected_state = gr.State()  
     with gr.Row():
         gallery = gr.Gallery(
             [(item["image"], item["title"]) for item in loras],
@@ -64,7 +60,7 @@ with gr.Blocks(css="custom.css") as app:
 
     gallery.select(
         update_selection,
-        outputs=[selected_state]
+        outputs=[prompt, selected_state]
     )
     button.click(
         fn=run_lora,