Updates

app.py

@@ -13,6 +13,8 @@ import copy
 import random
 import time
 
+selected_lora_index = None
+
 # Load LoRAs from JSON file
 with open('loras.json', 'r') as f:
     loras = json.load(f)
@@ -57,6 +59,8 @@ class calculateDuration:
             print(f"Elapsed time: {self.elapsed_time:.6f} seconds")
 
 def update_selection(evt: gr.SelectData, width, height):
+    global selected_lora_index
+    selected_lora_index = evt.index
     selected_lora = loras[evt.index]
     new_placeholder = f"{selected_lora['trigger_word']} {prompt.value}"
     lora_repo = selected_lora["repo"]
@@ -74,7 +78,6 @@ def update_selection(evt: gr.SelectData, width, height):
     return (
         gr.update(value=new_placeholder),
         updated_text,
-        evt.index,
         width,
         height,
         gr.update(interactive=True)  # Enable the Generate button
@@ -99,11 +102,12 @@ def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, lora_scal
         ):
             yield img
 
-def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, width, height, lora_scale, progress=gr.Progress(track_tqdm=True)):
-    if selected_index is None:
+def run_lora(prompt, cfg_scale, steps, randomize_seed, seed, width, height, lora_scale, progress=gr.Progress(track_tqdm=True)):
+    global selected_lora_index
+    if selected_lora_index is None:
         raise gr.Error("You must select a LoRA before proceeding.")
     
-    selected_lora = loras[selected_index]
+    selected_lora = loras[selected_lora_index]
     lora_path = selected_lora["repo"]
     trigger_word = selected_lora["trigger_word"]
     
@@ -196,9 +200,9 @@ with gr.Blocks() as demo:
     progress_bar = gr.Markdown(visible=False)
 
     # Event handlers
-    gallery.select(update_selection, [width, height], [prompt, selected_lora, gr.State(), width, height, generate])
+    gallery.select(update_selection, [width, height], [prompt, selected_lora, width, height, generate])
     randomize_seed.change(lambda x: gr.update(visible=not x), randomize_seed, seed_input)
-    generate_event = generate.click(run_lora, inputs=[prompt, cfg_scale, steps, gr.State(), randomize_seed, seed_input, width, height, lora_scale], outputs=[result, seed_output, progress_bar])
+    generate_event = generate.click(run_lora, inputs=[prompt, cfg_scale, steps, randomize_seed, seed_input, width, height, lora_scale], outputs=[result, seed_output, progress_bar])
     cancel.click(lambda: None, None, None, cancels=[generate_event])
 
 demo.queue().launch()