Update app.py

app.py

@@ -163,7 +163,154 @@ def update_selection(evt: gr.SelectData, selected_indices, loras_state, width, h
             *lora_images,
             gr.update()
     )
-
+def remove_lora_1(selected_indices, loras_state):
+    if len(selected_indices) >= 1:
+        selected_indices.pop(0)
+    selected_info_1 = "Select a LoRA 1"
+    selected_info_2 = "Select a LoRA 2"
+    selected_info_3 = "Select a LoRA 3"
+    selected_info_4 = "Select a LoRA 4"
+    lora_scale_1 = 0.5
+    lora_scale_2 = 0.5
+    lora_scale_3 = 0.5
+    lora_scale_4 = 0.5
+    lora_image_1 = None
+    lora_image_2 = None
+    lora_image_3 = None
+    lora_image_4 = None
+    if len(selected_indices) >= 1:
+        lora1 = loras_state[selected_indices[0]]
+        trigger_word = lora1.get('trigger_word', '')  # Get actual trigger word from LoRA 1
+        selected_info_1 = f"### LoRA 1 Selected: [{lora1['title']}]({lora1['repo']}) ✨ {trigger_word}"
+        lora_image_1 = lora1['image']
+    if len(selected_indices) >= 2:
+        lora2 = loras_state[selected_indices[1]]
+        trigger_word = lora2.get('trigger_word', '')  # Get actual trigger word from LoRA 2
+        selected_info_2 = f"### LoRA 2 Selected: [{lora2['title']}]({lora2['repo']}) ✨ {trigger_word}"
+        lora_image_2 = lora2['image']
+    if len(selected_indices) >= 3:
+        lora3 = loras_state[selected_indices[2]]
+        trigger_word = lora3.get('trigger_word', '')  # Get actual trigger word from LoRA 3
+        selected_info_3 = f"### LoRA 3 Selected: [{lora3['title']}]({lora3['repo']}) ✨ {trigger_word}"
+        lora_image_3 = lora3['image']
+    if len(selected_indices) >= 4:
+        lora4 = loras_state[selected_indices[3]]
+        trigger_word = lora4.get('trigger_word', '')  # Get actual trigger word from LoRA 4
+        selected_info_4 = f"### LoRA 4 Selected: [{lora4['title']}]({lora4['repo']}) ✨ {trigger_word}"
+        lora_image_4 = lora4['image']
+    return selected_info_1, selected_info_2, selected_info_3, selected_info_4, selected_indices, lora_scale_1, lora_scale_2, lora_scale_3, lora_scale_4, lora_image_1, lora_image_2, lora_image_3, lora_image_4, gr.update()
+    
+def remove_lora_2(selected_indices, loras_state):
+    if len(selected_indices) >= 2:
+        selected_indices.pop(1)
+    selected_info_1 = "Select a LoRA 1"
+    selected_info_2 = "Select a LoRA 2"
+    selected_info_3 = "Select a LoRA 3"
+    selected_info_4 = "Select a LoRA 4"
+    lora_scale_1 = 0.5
+    lora_scale_2 = 0.5
+    lora_scale_3 = 0.5
+    lora_scale_4 = 0.5
+    lora_image_1 = None
+    lora_image_2 = None
+    lora_image_3 = None
+    lora_image_4 = None
+    if len(selected_indices) >= 1:
+        lora1 = loras_state[selected_indices[0]]
+        trigger_word = lora1.get('trigger_word', '')  # Get actual trigger word from LoRA 1
+        selected_info_1 = f"### LoRA 1 Selected: [{lora1['title']}]({lora1['repo']}) ✨ {trigger_word}"
+        lora_image_1 = lora1['image']
+    if len(selected_indices) >= 2:
+        lora2 = loras_state[selected_indices[1]]
+        trigger_word = lora2.get('trigger_word', '')  # Get actual trigger word from LoRA 2
+        selected_info_2 = f"### LoRA 2 Selected: [{lora2['title']}]({lora2['repo']}) ✨ {trigger_word}"
+        lora_image_2 = lora2['image']
+    if len(selected_indices) >= 3:
+        lora3 = loras_state[selected_indices[2]]
+        trigger_word = lora3.get('trigger_word', '')  # Get actual trigger word from LoRA 3
+        selected_info_3 = f"### LoRA 3 Selected: [{lora3['title']}]({lora3['repo']}) ✨ {trigger_word}"
+        lora_image_3 = lora3['image']
+    if len(selected_indices) >= 4:
+        lora4 = loras_state[selected_indices[3]]
+        trigger_word = lora4.get('trigger_word', '')  # Get actual trigger word from LoRA 4
+        selected_info_4 = f"### LoRA 4 Selected: [{lora4['title']}]({lora4['repo']}) ✨ {trigger_word}"
+        lora_image_4 = lora4['image']
+    return selected_info_1, selected_info_2, selected_info_3, selected_info_4, selected_indices, lora_scale_1, lora_scale_2, lora_scale_3, lora_scale_4, lora_image_1, lora_image_2, lora_image_3, lora_image_4, gr.update()
+    
+def remove_lora_3(selected_indices, loras_state):
+    if len(selected_indices) >= 3:
+        selected_indices.pop(2)
+    selected_info_1 = "Select a LoRA 1"
+    selected_info_2 = "Select a LoRA 2"
+    selected_info_3 = "Select a LoRA 3"
+    selected_info_4 = "Select a LoRA 4"
+    lora_scale_1 = 0.5
+    lora_scale_2 = 0.5
+    lora_scale_3 = 0.5
+    lora_scale_4 = 0.5
+    lora_image_1 = None
+    lora_image_2 = None
+    lora_image_3 = None
+    lora_image_4 = None
+    if len(selected_indices) >= 1:
+        lora1 = loras_state[selected_indices[0]]
+        trigger_word = lora1.get('trigger_word', '')  # Get actual trigger word from LoRA 1
+        selected_info_1 = f"### LoRA 1 Selected: [{lora1['title']}]({lora1['repo']}) ✨ {trigger_word}"
+        lora_image_1 = lora1['image']
+    if len(selected_indices) >= 2:
+        lora2 = loras_state[selected_indices[1]]
+        trigger_word = lora2.get('trigger_word', '')  # Get actual trigger word from LoRA 2
+        selected_info_2 = f"### LoRA 2 Selected: [{lora2['title']}]({lora2['repo']}) ✨ {trigger_word}"
+        lora_image_2 = lora2['image']
+    if len(selected_indices) >= 3:
+        lora3 = loras_state[selected_indices[2]]
+        trigger_word = lora3.get('trigger_word', '')  # Get actual trigger word from LoRA 3
+        selected_info_3 = f"### LoRA 3 Selected: [{lora3['title']}]({lora3['repo']}) ✨ {trigger_word}"
+        lora_image_3 = lora3['image']
+    if len(selected_indices) >= 4:
+        lora4 = loras_state[selected_indices[3]]
+        trigger_word = lora4.get('trigger_word', '')  # Get actual trigger word from LoRA 4
+        selected_info_4 = f"### LoRA 4 Selected: [{lora4['title']}]({lora4['repo']}) ✨ {trigger_word}"
+        lora_image_4 = lora4['image']
+    return selected_info_1, selected_info_2, selected_info_3, selected_info_4, selected_indices, lora_scale_1, lora_scale_2, lora_scale_3, lora_scale_4, lora_image_1, lora_image_2, lora_image_3, lora_image_4, gr.update()
+    
+def remove_lora_4(selected_indices, loras_state):
+    if len(selected_indices) >= 4:
+        selected_indices.pop(3)
+    selected_info_1 = "Select a LoRA 1"
+    selected_info_2 = "Select a LoRA 2"
+    selected_info_3 = "Select a LoRA 3"
+    selected_info_4 = "Select a LoRA 4"
+    lora_scale_1 = 0.5
+    lora_scale_2 = 0.5
+    lora_scale_3 = 0.5
+    lora_scale_4 = 0.5
+    lora_image_1 = None
+    lora_image_2 = None
+    lora_image_3 = None
+    lora_image_4 = None
+    if len(selected_indices) >= 1:
+        lora1 = loras_state[selected_indices[0]]
+        trigger_word = lora1.get('trigger_word', '')  # Get actual trigger word from LoRA 1
+        selected_info_1 = f"### LoRA 1 Selected: [{lora1['title']}]({lora1['repo']}) ✨ {trigger_word}"
+        lora_image_1 = lora1['image']
+    if len(selected_indices) >= 2:
+        lora2 = loras_state[selected_indices[1]]
+        trigger_word = lora2.get('trigger_word', '')  # Get actual trigger word from LoRA 2
+        selected_info_2 = f"### LoRA 2 Selected: [{lora2['title']}]({lora2['repo']}) ✨ {trigger_word}"
+        lora_image_2 = lora2['image']
+    if len(selected_indices) >= 3:
+        lora3 = loras_state[selected_indices[2]]
+        trigger_word = lora3.get('trigger_word', '')  # Get actual trigger word from LoRA 3
+        selected_info_3 = f"### LoRA 3 Selected: [{lora3['title']}]({lora3['repo']}) ✨ {trigger_word}"
+        lora_image_3 = lora3['image']
+    if len(selected_indices) >= 4:
+        lora4 = loras_state[selected_indices[3]]
+        trigger_word = lora4.get('trigger_word', '')  # Get actual trigger word from LoRA 4
+        selected_info_4 = f"### LoRA 4 Selected: [{lora4['title']}]({lora4['repo']}) ✨ {trigger_word}"
+        lora_image_4 = lora4['image']
+    return selected_info_1, selected_info_2, selected_info_3, selected_info_4, selected_indices, lora_scale_1, lora_scale_2, lora_scale_3, lora_scale_4, lora_image_1, lora_image_2, lora_image_3, lora_image_4, gr.update()
+    
 def remove_lora(selected_indices, loras_state):
     # Remove the LoRA based on the index
     if selected_indices: