Update app.py

app.py

@@ -118,25 +118,30 @@ def download_file(url, directory=None):
         file.write(response.content)
     
     return filepath
-            
+
 def update_selection(evt: gr.SelectData, selected_indices, loras_state, width, height):
     selected_index = evt.index
     selected_indices = selected_indices or []
     if selected_index in selected_indices:
         selected_indices.remove(selected_index)
     else:
-        if len(selected_indices) < 2:
+        if len(selected_indices) < 3:  # 변경: 2에서 3으로
             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 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"
+    
     lora_scale_1 = 1.15
     lora_scale_2 = 1.15
+    lora_scale_3 = 1.15    
     lora_image_1 = None
     lora_image_2 = None
+    lora_image_2 = 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']}) ✨"
@@ -145,14 +150,19 @@ def update_selection(evt: gr.SelectData, selected_indices, loras_state, width, h
         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 selected_indices:
         last_selected_lora = loras_state[selected_indices[-1]]
         new_placeholder = f"Type a prompt for {last_selected_lora['title']}"
     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_indices, lora_scale_1, lora_scale_2, lora_scale_3, width, height, lora_image_1, lora_image_2, lora_image_3
+
 
 def remove_lora_1(selected_indices, loras_state):
     if len(selected_indices) >= 1:
@@ -350,7 +360,7 @@ def generate_image_to_image(prompt_mash, image_input_path, image_strength, steps
     ).images[0]
     return final_image
 
-def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_indices, lora_scale_1, lora_scale_2, randomize_seed, seed, width, height, loras_state, progress=gr.Progress(track_tqdm=True)):
+def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_indices, lora_scale_1, lora_scale_2, lora_scale_3, randomize_seed, seed, width, height, loras_state, progress=gr.Progress(track_tqdm=True)):
     try:
         # 한글 감지 및 번역
         if any('\u3131' <= char <= '\u318E' or '\uAC00' <= char <= '\uD7A3' for char in prompt):
@@ -390,7 +400,7 @@ def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_ind
             for idx, lora in enumerate(selected_loras):
                 lora_name = f"lora_{idx}"
                 lora_names.append(lora_name)
-                lora_weights.append(lora_scale_1 if idx == 0 else lora_scale_2)
+                lora_weights.append(lora_scale_1 if idx == 0 else lora_scale_2 if idx == 1 else lora_scale_3)
                 lora_path = lora['repo']
                 weight_name = lora.get("weights")
                 print(f"Lora Path: {lora_path}")
@@ -435,12 +445,12 @@ def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_ind
             raise Exception("Failed to generate image")
         
         return final_image, seed, gr.update(visible=False)
+
+
     except Exception as e:
         print(f"Error in run_lora: {str(e)}")
         return None, seed, gr.update(visible=False)
 
-
-
 run_lora.zerogpu = True
 
 def get_huggingface_safetensors(link):
@@ -636,6 +646,9 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, delete_cache=(60, 3600)) as a
             prompt = gr.Textbox(label="Prompt", lines=1, placeholder="Type a prompt after selecting a LoRA")
         with gr.Column(scale=1):
             generate_button = gr.Button("Generate", variant="primary", elem_classes=["button_total"])
+
+
+
     
     with gr.Row(elem_id="loaded_loras"):
         with gr.Column(scale=1, min_width=25):
@@ -650,16 +663,18 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, delete_cache=(60, 3600)) as a
                     lora_scale_1 = gr.Slider(label="LoRA 1 Scale", minimum=0, maximum=3, step=0.01, value=1.15)
             with gr.Row():
                 remove_button_1 = gr.Button("Remove", size="sm")
+
         with gr.Column(scale=8):
             with gr.Row():
                 with gr.Column(scale=0, min_width=50):
-                    lora_image_2 = gr.Image(label="LoRA 2 Image", interactive=False, min_width=50, width=50, show_label=False, show_share_button=False, show_download_button=False, show_fullscreen_button=False, height=50)
+                    lora_image_3 = gr.Image(label="LoRA 3 Image", interactive=False, min_width=50, width=50, show_label=False, show_share_button=False, show_download_button=False, show_fullscreen_button=False, height=50)
                 with gr.Column(scale=3, min_width=100):
-                    selected_info_2 = gr.Markdown("Select a LoRA 2")
+                    selected_info_3 = gr.Markdown("Select a LoRA 3")
                 with gr.Column(scale=5, min_width=50):
-                    lora_scale_2 = gr.Slider(label="LoRA 2 Scale", minimum=0, maximum=3, step=0.01, value=1.15)
+                    lora_scale_3 = gr.Slider(label="LoRA 3 Scale", minimum=0, maximum=3, step=0.01, value=1.15)
             with gr.Row():
-                remove_button_2 = gr.Button("Remove", size="sm")
+                remove_button_3 = gr.Button("Remove", size="sm")
+            
     
     with gr.Row():
         with gr.Column():
@@ -776,7 +791,7 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, delete_cache=(60, 3600)) as a
     gr.on(
         triggers=[generate_button.click, prompt.submit],
         fn=run_lora,
-        inputs=[prompt, input_image, image_strength, cfg_scale, steps, selected_indices, lora_scale_1, lora_scale_2, randomize_seed, seed, width, height, loras_state],
+        inputs=[prompt, input_image, image_strength, cfg_scale, steps, selected_indices, lora_scale_1, lora_scale_2, lora_scale_3, randomize_seed, seed, width, height, loras_state],
         outputs=[result, seed, progress_bar]
     ).then(
         fn=lambda x, history: update_history(x, history) if x is not None else history,
@@ -784,6 +799,8 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, delete_cache=(60, 3600)) as a
         outputs=history_gallery,
     )
 
+
+
     upscale_input.upload(
         lambda x: gr.update(interactive=x is not None),
         inputs=[upscale_input],