Update app.py

app.py

@@ -1,19 +1,31 @@
+# Existing imports
 import gradio as gr
 import requests
 import io
 from PIL import Image
 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)
 
 # Define the function to run when the button is clicked
-def run_lora(prompt, selected_state):
+def update_selection(selected=None):
+    if selected is None:
+        return None
+    return {'index': selected[0]},
+
+def run_lora(prompt, selected_state, progress=gr.Progress(track_tqdm=True)):
+    logging.debug(f"Inside run_lora, selected_state: {selected_state}")
     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']}"
@@ -29,12 +41,6 @@ def run_lora(prompt, selected_state):
     else:
         return "API Error"
 
-# Placeholder for gallery.select function
-def update_selection(selected=None):
-    if selected is None:
-        return None
-    return {'index': selected[0]},
-
 # Gradio UI
 with gr.Blocks(css="custom.css") as app:
     title = gr.HTML("<h1>LoRA the Explorer</h1>")