Update app.py

app.py

@@ -94,9 +94,10 @@ def shuffle_images():
     compatible_items = [item for item in sdxl_loras if item['is_compatible']]
     random.shuffle(compatible_items)
     two_shuffled_items = compatible_items[:2]
-    title_1  = gr.update(label=two_shuffled_items[0]['title'], value=two_shuffled_items[0]['image'])
+    title_1 = gr.update(label=two_shuffled_items[0]['title'], value=two_shuffled_items[0]['image'])
     title_2 = gr.update(label=two_shuffled_items[1]['title'], value=two_shuffled_items[1]['image'])
-
+    repo_id_1 = gr.update(value=two_shuffled_items[0]['repo'])
+    repo_id_2 = gr.update(value=two_shuffled_items[1]['repo'])
     description_1, trigger_word_1 = get_description(two_shuffled_items[0])
     description_2, trigger_word_2 = get_description(two_shuffled_items[1])
     
@@ -105,7 +106,7 @@ def shuffle_images():
     prompt = gr.update(value=f"{trigger_word_1} {trigger_word_2}")
     scale = gr.update(value=0.7)
     
-    return title_1, prompt_description_1, title_2, prompt_description_2, prompt, two_shuffled_items, scale, scale
+    return title_1, prompt_description_1, repo_id_1, title_2, prompt_description_2, repo_id_2, prompt, two_shuffled_items, scale, scale
 
 with gr.Blocks(css=css) as demo:
   shuffled_items = gr.State()
@@ -120,11 +121,13 @@ with gr.Blocks(css=css) as demo:
         with gr.Row():
             with gr.Column(min_width=10, scale=3, elem_classes="random_column"):
               lora_1 = gr.Image(interactive=False, height=150, elem_classes="selected_random", elem_id="randomLoRA_1", show_share_button=False, show_download_button=False)
+              lora_1_id = gr.Textbox(visible=False)
               lora_1_prompt = gr.Markdown(visible=False)
             with gr.Column(min_width=10, scale=1, elem_classes="plus_column"):
               plus = gr.HTML("+", elem_classes="plus_button")
             with gr.Column(min_width=10, scale=3, elem_classes="random_column"):
               lora_2 = gr.Image(interactive=False, height=150, elem_classes="selected_random", elem_id="randomLoRA_2", show_share_button=False, show_download_button=False)
+              lora_2_id = gr.Textbox(visible=False)
               lora_2_prompt = gr.Markdown(visible=False)
             with gr.Column(min_width=10, scale=2, elem_classes="plus_column"):
                equal = gr.HTML("=", elem_classes="plus_button")
@@ -152,8 +155,8 @@ with gr.Blocks(css=css) as demo:
       lora_2_scale = gr.Slider(label="LoRa 2 Scale", minimum=0, maximum=1, step=0.1, value=0.7)
   shuffle_button = gr.Button("Reshuffle!")
     
-  demo.load(shuffle_images, inputs=[], outputs=[lora_1, lora_1_prompt, lora_2, lora_2_prompt, prompt, shuffled_items, lora_1_scale, lora_2_scale], queue=False, show_progress="hidden")
-  shuffle_button.click(shuffle_images, outputs=[lora_1, lora_1_prompt, lora_2, lora_2_prompt, prompt, shuffled_items, lora_1_scale, lora_2_scale], queue=False, show_progress="hidden")
+  demo.load(shuffle_images, inputs=[], outputs=[lora_1, lora_1_prompt, lora_1_id, lora_2, lora_2_prompt, lora_2_id, prompt, shuffled_items, lora_1_scale, lora_2_scale], queue=False, show_progress="hidden")
+  shuffle_button.click(shuffle_images, outputs=[lora_1, lora_1_prompt, lora_1_id, lora_2, lora_2_prompt, lora_2_id, prompt, shuffled_items, lora_1_scale, lora_2_scale], queue=False, show_progress="hidden")
 
   run_btn.click(merge_and_run, inputs=[prompt, negative_prompt, shuffled_items, lora_1_scale, lora_2_scale], outputs=[output_image, post_gen_info])
   prompt.submit(merge_and_run, inputs=[prompt, negative_prompt, shuffled_items, lora_1_scale, lora_2_scale], outputs=[output_image, post_gen_info])