Update app.py

app.py

@@ -164,153 +164,22 @@ def update_selection(evt: gr.SelectData, selected_indices, loras_state, width, h
             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:
+        selected_indices.pop()  # Remove the last selected LoRA
+
+    selected_info = ["Select a LoRA"] * 4
+    lora_images = [None] * 4
+    lora_scales = [0.5] * 4
+
+    for i in range(min(len(selected_indices), 4)):
+        lora = loras_state[selected_indices[i]]
+        trigger_word = lora.get('trigger_word', '')
+        selected_info[i] = f"### LoRA {i + 1} Selected: [{lora['title']}]({lora['repo']}) ✨ {trigger_word}"
+        lora_images[i] = lora['image']
+
+    return selected_info, selected_indices, lora_scales, lora_images, gr.update()
     
 def add_custom_lora(custom_lora, selected_indices, current_loras, gallery):
     if custom_lora: