Update app.py

app.py

@@ -76,8 +76,11 @@ def shuffle_images():
     description_1, trigger_word_1 = get_description(two_shuffled_items[0])
     description_2, trigger_word_2 = get_description(two_shuffled_items[1])
     
+    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}")
-    return title_1,description_1,title_2,description_2,prompt, two_shuffled_items
+    
+    return title_1, prompt_description_1, title_2, prompt_description_2, prompt, two_shuffled_items
 
 with gr.Blocks(css=css) as demo:
   shuffled_items = gr.State()
@@ -90,12 +93,12 @@ with gr.Blocks(css=css) as demo:
   with gr.Row():
     with gr.Column(min_width=10, scale=6):
       lora_1 = gr.Image(interactive=False, height=300)
-      lora_1_prompt = gr.Markdown()
+      lora_1_prompt = gr.Markdown(visible=False)
     with gr.Column(min_width=10, scale=1, elem_id="plus_column"):
       plus = gr.HTML("+", elem_id="plus_button")
     with gr.Column(min_width=10, scale=6):
       lora_2 = gr.Image(interactive=False, height=300)
-      lora_2_prompt = gr.Markdown()
+      lora_2_prompt = gr.Markdown(visible=False)
   with gr.Row():
     prompt = gr.Textbox(label="Your prompt", info="arrange the trigger words of the two LoRAs in a coherent sentence", interactive=True, elem_id="prompt")
     run_btn = gr.Button("Run", elem_id="run_button")
@@ -108,10 +111,11 @@ 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 LoRAs!")
   
-  demo.load(shuffle_images, inputs=[], outputs=[lora_1,lora_1_prompt,lora_2,lora_2_prompt, prompt, shuffled_items], queue=False, show_progress="hidden")
-  shuffle_button.click(shuffle_images, outputs=[lora_1,lora_1_prompt,lora_2,lora_2_prompt, prompt, shuffled_items], queue=False, show_progress="hidden")
+  demo.load(shuffle_images, inputs=[], outputs=[lora_1, lora_1_prompt, lora_2, lora_2_prompt, prompt, shuffled_items], queue=False, show_progress="hidden")
+  shuffle_button.click(shuffle_images, outputs=[lora_1, lora_1_prompt, lora_2, lora_2_prompt, prompt, shuffled_items], 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])
   prompt.submit(merge_and_run, inputs=[prompt, negative_prompt, shuffled_items, lora_1_scale, lora_2_scale], outputs=[output_image])
+
 demo.queue()
 demo.launch()