Update app.py

app.py

@@ -131,9 +131,10 @@ def update_selection(evt: gr.SelectData, selected_indices, loras_state, width, h
             gr.Warning("You can select up to 3 LoRAs, remove one to select a new one.")
             return gr.update(), gr.update(), gr.update(), gr.update(), selected_indices, gr.update(), gr.update(), gr.update(), width, height, gr.update(), gr.update(), gr.update()
 
-    selected_info_1 = "Select a LoRA 1"
-    selected_info_2 = "Select a LoRA 2"
-    selected_info_3 = "Select a LoRA 3"
+    selected_info_1 = gr.Markdown("LoRA 1 선택")
+    selected_info_2 = gr.Markdown("LoRA 2 선택")
+    selected_info_3 = gr.Markdown("LoRA 3 선택")
+
     
     lora_scale_1 = 1.15
     lora_scale_2 = 1.15
@@ -756,38 +757,51 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, delete_cache=(60, 3600)) as a
         upscale_result = gr.Image(label="Upscaled Image", type="pil")
         upscale_seed_output = gr.Number(label="Seed Used", precision=0)
 
-
     gallery.select(
         update_selection,
         inputs=[selected_indices, loras_state, width, height],
-        outputs=[prompt, selected_info_1, selected_info_2,selected_info_3, selected_indices, lora_scale_1, lora_scale_2,lora_scale_3, width, height, lora_image_1, lora_image_2,lora_image_3]
+        outputs=[prompt, selected_info_1, selected_info_2, selected_info_3, selected_indices, lora_scale_1, lora_scale_2, lora_scale_3, width, height, lora_image_1, lora_image_2, lora_image_3]
     )
+
     remove_button_1.click(
         remove_lora_1,
         inputs=[selected_indices, loras_state],
-        outputs=[selected_info_1, selected_info_2, selected_indices, lora_scale_1, lora_scale_2, lora_image_1, lora_image_2]
+        outputs=[selected_info_1, selected_info_2, selected_info_3, selected_indices, lora_scale_1, lora_scale_2, lora_scale_3, lora_image_1, lora_image_2, lora_image_3]
     )
+
     remove_button_2.click(
         remove_lora_2,
         inputs=[selected_indices, loras_state],
-        outputs=[selected_info_1, selected_info_2, selected_indices, lora_scale_1, lora_scale_2, lora_image_1, lora_image_2]
+        outputs=[selected_info_1, selected_info_2, selected_info_3, selected_indices, lora_scale_1, lora_scale_2, lora_scale_3, lora_image_1, lora_image_2, lora_image_3]
     )
+
+    remove_button_3.click(
+        remove_lora_3,
+        inputs=[selected_indices, loras_state],
+        outputs=[selected_info_1, selected_info_2, selected_info_3, selected_indices, lora_scale_1, lora_scale_2, lora_scale_3, lora_image_1, lora_image_2, lora_image_3]
+    )
+
     randomize_button.click(
         randomize_loras,
         inputs=[selected_indices, loras_state],
-        outputs=[selected_info_1, selected_info_2, selected_indices, lora_scale_1, lora_scale_2, lora_image_1, lora_image_2, prompt]
+        outputs=[selected_info_1, selected_info_2, selected_info_3, selected_indices, lora_scale_1, lora_scale_2, lora_scale_3, lora_image_1, lora_image_2, lora_image_3, prompt]
     )
+
     add_custom_lora_button.click(
         add_custom_lora,
         inputs=[custom_lora, selected_indices, loras_state],
-        outputs=[loras_state, gallery, selected_info_1, selected_info_2, selected_indices, lora_scale_1, lora_scale_2, lora_image_1, lora_image_2]
+        outputs=[loras_state, gallery, selected_info_1, selected_info_2, selected_info_3, selected_indices, lora_scale_1, lora_scale_2, lora_scale_3, lora_image_1, lora_image_2, lora_image_3]
     )
+
     remove_custom_lora_button.click(
         remove_custom_lora,
         inputs=[selected_indices, loras_state],
-        outputs=[loras_state, gallery, selected_info_1, selected_info_2, selected_indices, lora_scale_1, lora_scale_2, lora_image_1, lora_image_2]
+        outputs=[loras_state, gallery, selected_info_1, selected_info_2, selected_info_3, selected_indices, lora_scale_1, lora_scale_2, lora_scale_3, lora_image_1, lora_image_2, lora_image_3]
     )
 
+
+
+
     gr.on(
         triggers=[generate_button.click, prompt.submit],
         fn=run_lora,