Update app.py

app.py

@@ -111,26 +111,43 @@ def update_selection(evt: gr.SelectData, selected_indices, loras_state, width, h
     if selected_index in selected_indices:
         selected_indices.remove(selected_index)
     else:
-        if len(selected_indices) < 2:
+        if len(selected_indices) < 4:
             selected_indices.append(selected_index)
         else:
-            gr.Warning("You can select up to 2 LoRAs, remove one to select a new one.")
-            return gr.update(), gr.update(), gr.update(), selected_indices, gr.update(), gr.update(), width, height, gr.update(), gr.update()
+            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(), gr.update(), gr.update(), selected_indices, gr.update(), gr.update(), gr.update(), gr.update(), width, height, gr.update(), gr.update(), gr.update(), gr.update()
 
     selected_info_1 = "Select a Celebrity as LoRA 1"
     selected_info_2 = "Select a LoRA 2"
-    lora_scale_1 = 0.6
+    selected_info_3 = "Select a LoRA 3"
+    selected_info_4 = "Select a LoRA 4"
+    lora_scale_1 = 1.15
     lora_scale_2 = 1.15
+    lora_scale_3 = 0.65
+    lora_scale_4 = 0.65
     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]]
         selected_info_1 = f"### LoRA 1 Selected: [{lora1['title']}](https://huggingface.co/{lora1['repo']}) ✨"
         lora_image_1 = lora1['image']
+        
     if len(selected_indices) >= 2:
         lora2 = loras_state[selected_indices[1]]
         selected_info_2 = f"### LoRA 2 Selected: [{lora2['title']}](https://huggingface.co/{lora2['repo']}) ✨"
         lora_image_2 = lora2['image']
+        
+    if len(selected_indices) >= 3:
+        lora3 = loras_state[selected_indices[2]]
+        selected_info_3 = f"### LoRA 3 Selected: [{lora3['title']}](https://huggingface.co/{lora3['repo']}) ✨"
+        lora_image_3 = lora3['image']
+        
+    if len(selected_indices) >= 4:
+        lora4 = loras_state[selected_indices[3]]
+        selected_info_4 = f"### LoRA 4 Selected: [{lora4['title']}](https://huggingface.co/{lora4['repo']}) ✨"
+        lora_image_4 = lora4['image']
 
     if selected_indices:
         last_selected_lora = loras_state[selected_indices[-1]]
@@ -138,46 +155,152 @@ def update_selection(evt: gr.SelectData, selected_indices, loras_state, width, h
     else:
         new_placeholder = "Type a prompt after selecting a LoRA"
 
-    return gr.update(placeholder=new_placeholder), selected_info_1, selected_info_2, selected_indices, lora_scale_1, lora_scale_2, width, height, lora_image_1, lora_image_2
+    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
 
 def remove_lora_1(selected_indices, loras_state):
     if len(selected_indices) >= 1:
         selected_indices.pop(0)
     selected_info_1 = "Select a Celebrity as LoRA 1"
     selected_info_2 = "Select a LoRA 2"
-    lora_scale_1 = 0.6
+    selected_info_3 = "Select a LoRA 3"
+    selected_info_4 = "Select a LoRA 4"
+    lora_scale_1 = 1.15
     lora_scale_2 = 1.15
+    lora_scale_3 = 0.65
+    lora_scale_4 = 0.65
     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]]
-        selected_info_1 = f"### LoRA 1 Selected: [{lora1['title']}]({lora1['repo']}) ✨"
+        selected_info_1 = f"### LoRA 1 Selected: [{lora1['title']}](https://huggingface.co/{lora1['repo']}) ✨"
         lora_image_1 = lora1['image']
+        
     if len(selected_indices) >= 2:
         lora2 = loras_state[selected_indices[1]]
-        selected_info_2 = f"### LoRA 2 Selected: [{lora2['title']}]({lora2['repo']}) ✨"
+        selected_info_2 = f"### LoRA 2 Selected: [{lora2['title']}](https://huggingface.co/{lora2['repo']}) ✨"
         lora_image_2 = lora2['image']
-    return selected_info_1, selected_info_2, selected_indices, lora_scale_1, lora_scale_2, lora_image_1, lora_image_2
+        
+    if len(selected_indices) >= 3:
+        lora3 = loras_state[selected_indices[2]]
+        selected_info_3 = f"### LoRA 3 Selected: [{lora3['title']}](https://huggingface.co/{lora3['repo']}) ✨"
+        lora_image_3 = lora3['image']
+        
+    if len(selected_indices) >= 4:
+        lora4 = loras_state[selected_indices[3]]
+        selected_info_4 = f"### LoRA 4 Selected: [{lora4['title']}](https://huggingface.co/{lora4['repo']}) ✨"
+        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
 
 def remove_lora_2(selected_indices, loras_state):
     if len(selected_indices) >= 2:
         selected_indices.pop(1)
     selected_info_1 = "Select a Celebrity as LoRA 1"
-    selected_info_2 = "Select LoRA 2"
-    lora_scale_1 = 0.6
+    selected_info_2 = "Select a LoRA 2"
+    selected_info_3 = "Select a LoRA 3"
+    selected_info_4 = "Select a LoRA 4"
+    lora_scale_1 = 1.15
     lora_scale_2 = 1.15
+    lora_scale_3 = 0.65
+    lora_scale_4 = 0.65
     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]]
-        selected_info_1 = f"### LoRA 1 Selected: [{lora1['title']}]({lora1['repo']}) ✨"
+        selected_info_1 = f"### LoRA 1 Selected: [{lora1['title']}](https://huggingface.co/{lora1['repo']}) ✨"
         lora_image_1 = lora1['image']
+        
     if len(selected_indices) >= 2:
         lora2 = loras_state[selected_indices[1]]
-        selected_info_2 = f"### LoRA 2 Selected: [{lora2['title']}]({lora2['repo']}) ✨"
+        selected_info_2 = f"### LoRA 2 Selected: [{lora2['title']}](https://huggingface.co/{lora2['repo']}) ✨"
         lora_image_2 = lora2['image']
-    return selected_info_1, selected_info_2, selected_indices, lora_scale_1, lora_scale_2, lora_image_1, lora_image_2
-
+        
+    if len(selected_indices) >= 3:
+        lora3 = loras_state[selected_indices[2]]
+        selected_info_3 = f"### LoRA 3 Selected: [{lora3['title']}](https://huggingface.co/{lora3['repo']}) ✨"
+        lora_image_3 = lora3['image']
+        
+    if len(selected_indices) >= 4:
+        lora4 = loras_state[selected_indices[3]]
+        selected_info_4 = f"### LoRA 4 Selected: [{lora4['title']}](https://huggingface.co/{lora4['repo']}) ✨"
+        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
+
+def remove_lora_3(selected_indices, loras_state):
+    if len(selected_indices) >= 3:
+        selected_indices.pop(1)
+    selected_info_1 = "Select a Celebrity as 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 = 1.15
+    lora_scale_2 = 1.15
+    lora_scale_3 = 0.65
+    lora_scale_4 = 0.65
+    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]]
+        selected_info_1 = f"### LoRA 1 Selected: [{lora1['title']}](https://huggingface.co/{lora1['repo']}) ✨"
+        lora_image_1 = lora1['image']
+        
+    if len(selected_indices) >= 2:
+        lora2 = loras_state[selected_indices[1]]
+        selected_info_2 = f"### LoRA 2 Selected: [{lora2['title']}](https://huggingface.co/{lora2['repo']}) ✨"
+        lora_image_2 = lora2['image']
+        
+    if len(selected_indices) >= 3:
+        lora3 = loras_state[selected_indices[2]]
+        selected_info_3 = f"### LoRA 3 Selected: [{lora3['title']}](https://huggingface.co/{lora3['repo']}) ✨"
+        lora_image_3 = lora3['image']
+        
+    if len(selected_indices) >= 4:
+        lora4 = loras_state[selected_indices[3]]
+        selected_info_4 = f"### LoRA 4 Selected: [{lora4['title']}](https://huggingface.co/{lora4['repo']}) ✨"
+        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
+
+def remove_lora_4(selected_indices, loras_state):
+    if len(selected_indices) >= 4:
+        selected_indices.pop(1)
+    selected_info_1 = "Select a Celebrity as 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 = 1.15
+    lora_scale_2 = 1.15
+    lora_scale_3 = 0.65
+    lora_scale_4 = 0.65
+    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]]
+        selected_info_1 = f"### LoRA 1 Selected: [{lora1['title']}](https://huggingface.co/{lora1['repo']}) ✨"
+        lora_image_1 = lora1['image']
+        
+    if len(selected_indices) >= 2:
+        lora2 = loras_state[selected_indices[1]]
+        selected_info_2 = f"### LoRA 2 Selected: [{lora2['title']}](https://huggingface.co/{lora2['repo']}) ✨"
+        lora_image_2 = lora2['image']
+        
+    if len(selected_indices) >= 3:
+        lora3 = loras_state[selected_indices[2]]
+        selected_info_3 = f"### LoRA 3 Selected: [{lora3['title']}](https://huggingface.co/{lora3['repo']}) ✨"
+        lora_image_3 = lora3['image']
+        
+    if len(selected_indices) >= 4:
+        lora4 = loras_state[selected_indices[3]]
+        selected_info_4 = f"### LoRA 4 Selected: [{lora4['title']}](https://huggingface.co/{lora4['repo']}) ✨"
+        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
+    
 def randomize_loras(selected_indices, loras_state):
     if len(loras_state) < 2:
         raise gr.Error("Not enough LoRAs to randomize.")
@@ -191,7 +314,7 @@ def randomize_loras(selected_indices, loras_state):
     lora_image_1 = lora1['image']
     lora_image_2 = lora2['image']
     random_prompt = random.choice(prompt_values)
-    return selected_info_1, selected_info_2, selected_indices, lora_scale_1, lora_scale_2, lora_image_1, lora_image_2, random_prompt
+    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, random_prompt
 
 def add_custom_lora(custom_lora, selected_indices, current_loras, gallery):
     if custom_lora:
@@ -216,44 +339,51 @@ def add_custom_lora(custom_lora, selected_indices, current_loras, gallery):
             # Update gallery
             gallery_items = [(item["image"], item["title"]) for item in current_loras]
             # Update selected_indices if there's room
-            if len(selected_indices) < 2:
+            if len(selected_indices) < 4:
                 selected_indices.append(existing_item_index)
             else:
-                gr.Warning("You can select up to 2 LoRAs, remove one to select a new one.")
+                gr.Warning("You can select up to 4 LoRAs, remove one to select a new one.")
 
             # Update selected_info and images
             selected_info_1 = "Select a Celebrity as LoRA 1"
             selected_info_2 = "Select a LoRA 2"
-            lora_scale_1 = 0.6
+            selected_info_3 = "Select a LoRA 3"
+            selected_info_4 = "Select a LoRA 4"
+            lora_scale_1 = 1.15
             lora_scale_2 = 1.15
+            lora_scale_3 = 0.65
+            lora_scale_4 = 0.65
             lora_image_1 = None
             lora_image_2 = None
+            lora_image_3 = None
+            lora_image_4 = None
             if len(selected_indices) >= 1:
                 lora1 = current_loras[selected_indices[0]]
                 selected_info_1 = f"### LoRA 1 Selected: {lora1['title']} ✨"
                 lora_image_1 = lora1['image'] if lora1['image'] else None
+        
             if len(selected_indices) >= 2:
                 lora2 = current_loras[selected_indices[1]]
                 selected_info_2 = f"### LoRA 2 Selected: {lora2['title']} ✨"
                 lora_image_2 = lora2['image'] if lora2['image'] else None
+        
+            if len(selected_indices) >= 3:
+                lora3 = current_loras[selected_indices[2]]
+                selected_info_3 = f"### LoRA 3 Selected: {lora3['title']} ✨"
+                lora_image_3 = lora3['image'] if lora3['image'] else None
+        
+            if len(selected_indices) >= 4:
+                lora4 = current_loras[selected_indices[3]]
+                selected_info_4 = f"### LoRA 4 Selected: {lora4['title']} ✨"
+                lora_image_4 = lora4['image'] if lora4['image'] else None
             print("Finished adding custom LoRA")
-            return (
-                current_loras,
-                gr.update(value=gallery_items),
-                selected_info_1, 
-                selected_info_2,
-                selected_indices,
-                lora_scale_1,
-                lora_scale_2,
-                lora_image_1,
-                lora_image_2
-            )
+            return (current_loras, gr.update(value=gallery_items), 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)
         except Exception as e:
             print(e)
             gr.Warning(str(e))
-            return current_loras, gr.update(), gr.update(), gr.update(), selected_indices, gr.update(), gr.update(), gr.update(), gr.update()
+            return current_loras, gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), selected_indices, gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update()
     else:
-        return current_loras, gr.update(), gr.update(), gr.update(), selected_indices, gr.update(), gr.update(), gr.update(), gr.update()
+        return current_loras, gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), selected_indices, gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update(), gr.update()
 
 def remove_custom_lora(selected_indices, current_loras, gallery):
     if current_loras:
@@ -269,29 +399,36 @@ def remove_custom_lora(selected_indices, current_loras, gallery):
     # Update selected_info and images
     selected_info_1 = "Select a Celebrity as LoRA 1"
     selected_info_2 = "Select a LoRA 2"
-    lora_scale_1 = 0.6
+    selected_info_3 = "Select a LoRA 3"
+    selected_info_4 = "Select a LoRA 4"
+    lora_scale_1 = 1.15
     lora_scale_2 = 1.15
+    lora_scale_3 = 0.65
+    lora_scale_4 = 0.65
     lora_image_1 = None
     lora_image_2 = None
+    lora_image_3 = None
+    lora_image_4 = None
     if len(selected_indices) >= 1:
-        lora1 = current_loras[selected_indices[0]]
-        selected_info_1 = f"### LoRA 1 Selected: [{lora1['title']}]({lora1['repo']}) ✨"
+        lora1 = loras_state[selected_indices[0]]
+        selected_info_1 = f"### LoRA 1 Selected: [{lora1['title']}](https://huggingface.co/{lora1['repo']}) ✨"
         lora_image_1 = lora1['image']
+        
     if len(selected_indices) >= 2:
-        lora2 = current_loras[selected_indices[1]]
-        selected_info_2 = f"### LoRA 2 Selected: [{lora2['title']}]({lora2['repo']}) ✨"
+        lora2 = loras_state[selected_indices[1]]
+        selected_info_2 = f"### LoRA 2 Selected: [{lora2['title']}](https://huggingface.co/{lora2['repo']}) ✨"
         lora_image_2 = lora2['image']
-    return (
-        current_loras,
-        gr.update(value=gallery_items),
-        selected_info_1,
-        selected_info_2,
-        selected_indices,
-        lora_scale_1,
-        lora_scale_2,
-        lora_image_1,
-        lora_image_2
-    )
+        
+    if len(selected_indices) >= 3:
+        lora3 = loras_state[selected_indices[2]]
+        selected_info_3 = f"### LoRA 3 Selected: [{lora3['title']}](https://huggingface.co/{lora3['repo']}) ✨"
+        lora_image_3 = lora3['image']
+        
+    if len(selected_indices) >= 4:
+        lora4 = loras_state[selected_indices[3]]
+        selected_info_4 = f"### LoRA 4 Selected: [{lora4['title']}](https://huggingface.co/{lora4['repo']}) ✨"
+        lora_image_4 = lora4['image']
+    return (current_loras, gr.update(value=gallery_items), 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)
 
 def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress):
     print("Generating image...")