Update app.py

app.py

@@ -119,13 +119,17 @@ def shuffle_images():
     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])
-    
+
+    lora_1_link = f"[{repo_id_1}](https://huggingface.co/{repo_id_1}) ✨"
+    lora_2_link = f"[{repo_id_2}](https://huggingface.co/{repo_id_2}) ✨"
     prompt_description_1 = gr.update(value=description_1, visible=True)
     prompt_description_2 = gr.update(value=description_2, visible=True)
     prompt = gr.update(value=f"{trigger_word_1} {trigger_word_2}")
     scale = gr.update(value=0.7)
+
+    
     
-    return title_1, prompt_description_1, repo_id_1, title_2, prompt_description_2, repo_id_2, prompt, two_shuffled_items, scale, scale
+    return lora_link_1, title_1, prompt_description_1, repo_id_1, lora_link_2, title_2, prompt_description_2, repo_id_2, prompt, two_shuffled_items, scale, scale
 
 def save_preferences(lora_1_id, lora_1_scale, lora_2_id, lora_2_scale, prompt, generated_image, thumbs_direction, seed):
     image_path = IMAGE_DATASET_DIR / f"{uuid4()}.png"
@@ -149,13 +153,15 @@ with gr.Blocks(css=css) as demo:
     with gr.Column(min_width=10, scale=16, elem_classes="plus_column"):
         with gr.Row():
             with gr.Column(min_width=10, scale=4, 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_link = gr.Markdown()
+              lora_1 = gr.Image(interactive=False, show_label=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, elem_id="random_lora_1_id")
               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=4, 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_link = gr.Markdown()
+              lora_2 = gr.Image(interactive=False, show_label=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, elem_id="random_lora_2_id")
               lora_2_prompt = gr.Markdown(visible=False)
             with gr.Column(min_width=10, scale=2, elem_classes="plus_column"):
@@ -186,9 +192,10 @@ with gr.Blocks(css=css) as demo:
       lora_1_scale = gr.Slider(label="LoRA 1 Scale", minimum=0, maximum=1, step=0.1, value=0.7)
       lora_2_scale = gr.Slider(label="LoRa 2 Scale", minimum=0, maximum=1, step=0.1, value=0.7)
   shuffle_button = gr.Button("Reshuffle!")
+  gr.Markdown("For generating with intent visit the [LoRA the Explorer Space](https://huggingface.co/spaces/multimodalart/LoraTheExplorer), but don't forget that sometimes restirctions flourish creativity 🌸")
     
-  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")
+  demo.load(shuffle_images, inputs=[], outputs=[lora_1_link, lora_1, lora_1_prompt, lora_1_id, lora_2_link, 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_link, lora_1, lora_1_prompt, lora_1_id, lora_2_link, 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, seed], outputs=[output_image, post_gen_info, last_used_seed])
   prompt.submit(merge_and_run, inputs=[prompt, negative_prompt, shuffled_items, lora_1_scale, lora_2_scale, seed], outputs=[output_image, post_gen_info, last_used_seed])