Update app.py

app.py

@@ -121,41 +121,15 @@ def update_selection(evt: gr.SelectData, selected_indices, loras_state, width, h
             gr.Warning("You can select up to 4 LoRAs, remove one to select a new one.")
             return gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), selected_indices, gr.update(), 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_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']}](https://huggingface.co/{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']}](https://huggingface.co/{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']}](https://huggingface.co/{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']}](https://huggingface.co/{lora4['repo']}) ✨ {trigger_word}"
-        lora_image_4 = lora4['image'] 
+    selected_info = ["Select a LoRA {}".format(i) for i in range(1, 5)]
+    lora_images = [None] * 4
+    lora_scales = [0.5] * 4
+
+    for i in range(len(selected_indices)):
+        lora = loras_state[selected_indices[i]]
+        trigger_word = lora.get('trigger_word', '')
+        selected_info[i] = f"### LoRA {i + 1} Selected: [{lora['title']}](https://huggingface.co/{lora['repo']}) ✨ {trigger_word}"
+        lora_images[i] = lora['image']
 
     if selected_indices:
         last_selected_lora = loras_state[selected_indices[-1]]
@@ -164,11 +138,11 @@ def update_selection(evt: gr.SelectData, selected_indices, loras_state, width, h
         new_placeholder = "Type a prompt after selecting a LoRA"
 
     return (gr.update(placeholder=new_placeholder), 
-        	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,
-        	width, height, 
-        	lora_image_1, lora_image_2, lora_image_3, lora_image_4,
-         	gr.update()
+            *selected_info, selected_indices,
+            *lora_scales,
+            width, height, 
+            *lora_images,
+            gr.update()
     )
 
 def randomize_loras(selected_indices, loras_state):