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,9 +78,9 @@ 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
     )
 
 @spaces.GPU(duration=70)
@@ -117,11 +121,12 @@ def generate_image_to_image(prompt_mash, image_input_path, image_strength, steps
     ).images[0]
     return final_image
 
-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"]
     
@@ -201,7 +206,7 @@ with gr.Blocks() as demo:
         )
 
     with gr.Row():
-        generate = gr.Button("Generate", variant="primary")
+        generate = gr.Button("Generate", variant="primary", interactive=False)
         cancel = gr.Button("Cancel")
 
     with gr.Row():
@@ -227,9 +232,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])
+    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()