Upload 105 files

chain_injectors/krea2_controlnet_injector.py

@@ -0,0 +1,78 @@
+def inject(assembler, chain_definition, chain_items):
+    if not chain_items:
+        return
+
+    ksampler_name = chain_definition.get('ksampler_node', 'ksampler')
+    if ksampler_name not in assembler.node_map:
+        print(f"Warning: Target node '{ksampler_name}' for Krea2 ControlNet chain not found. Skipping.")
+        return
+        
+    ksampler_id = assembler.node_map[ksampler_name]
+
+    if 'model' not in assembler.workflow[ksampler_id]['inputs']:
+        print(f"Warning: KSampler node '{ksampler_name}' is missing 'model' input. Skipping.")
+        return
+
+    vae_source_str = chain_definition.get('vae_source')
+    vae_connection = None
+    if vae_source_str:
+        vae_node_name, vae_idx_str = vae_source_str.split(':')
+        if vae_node_name in assembler.node_map:
+            vae_connection = [assembler.node_map[vae_node_name], int(vae_idx_str)]
+            
+    latent_connection = assembler.workflow[ksampler_id]['inputs'].get('latent_image')
+    if not latent_connection:
+        print(f"Warning: KSampler node '{ksampler_name}' is missing 'latent_image' input. Krea2 ControlNet requires it. Skipping.")
+        return
+
+    current_model_connection = assembler.workflow[ksampler_id]['inputs']['model']
+    
+    for item_data in chain_items:
+        image_loader_id = assembler._get_unique_id()
+        image_loader_node = assembler._get_node_template("LoadImage")
+        image_loader_node['inputs']['image'] = item_data['image']
+        assembler.workflow[image_loader_id] = image_loader_node
+
+        image_scaler_id = assembler._get_unique_id()
+        image_scaler_node = assembler._get_node_template("ImageScaleToTotalPixels")
+        image_scaler_node['inputs']['image'] = [image_loader_id, 0]
+        image_scaler_node['inputs']['upscale_method'] = 'nearest-exact'
+        image_scaler_node['inputs']['megapixels'] = 1.0
+        image_scaler_node['inputs']['resolution_steps'] = 1
+        assembler.workflow[image_scaler_id] = image_scaler_node
+
+        lora_loader_id = assembler._get_unique_id()
+        lora_loader_node = assembler._get_node_template("Krea2ControlLoRALoader")
+        lora_loader_node['inputs']['lora_name'] = item_data['control_net_name']
+        lora_loader_node['inputs']['strength'] = item_data.get('strength', 1.0)
+        lora_loader_node['inputs']['model'] = current_model_connection
+        assembler.workflow[lora_loader_id] = lora_loader_node
+
+        img_encode_id = assembler._get_unique_id()
+        img_encode_node = assembler._get_node_template("Krea2ControlImageEncode")
+        img_encode_node['inputs']['resize'] = "match_latent_size"
+        img_encode_node['inputs']['upscale_method'] = "lanczos"
+        img_encode_node['inputs']['crop'] = "center"
+        img_encode_node['inputs']['channel_mode'] = "rgb"
+        img_encode_node['inputs']['normalize'] = "none"
+        img_encode_node['inputs']['invert'] = False
+        img_encode_node['inputs']['batch_mode'] = "independent_images"
+        img_encode_node['inputs']['control_image'] = [image_scaler_id, 0]
+        if vae_connection:
+            img_encode_node['inputs']['vae'] = vae_connection
+        if latent_connection:
+            img_encode_node['inputs']['latent'] = latent_connection
+        assembler.workflow[img_encode_id] = img_encode_node
+
+        apply_cn_id = assembler._get_unique_id()
+        apply_cn_node = assembler._get_node_template("Krea2ControlApply")
+        apply_cn_node['inputs']['model'] = [lora_loader_id, 0]
+        apply_cn_node['inputs']['control_latent'] = [img_encode_id, 0]
+        
+        assembler.workflow[apply_cn_id] = apply_cn_node
+
+        current_model_connection = [apply_cn_id, 0]
+
+    assembler.workflow[ksampler_id]['inputs']['model'] = current_model_connection
+    
+    print(f"Krea2 ControlNet injector applied. KSampler model input redirected through {len(chain_items)} Krea2 ControlNet nodes.")

comfy_integration/setup.py

@@ -65,6 +65,12 @@ def initialize_comfyui():
         print("✅ ComfyUI-IPAdapter-Flux extension cloned.")
     else:
         print("✅ ComfyUI-IPAdapter-Flux extension already exists.")
+    try:
+        print("--- [Setup] Applying PR #108 compatibility patch for ComfyUI-IPAdapter-Flux ---")
+        os.system(f"git -C {ipadapter_flux_path} fetch origin pull/108/head && git -C {ipadapter_flux_path} checkout -f FETCH_HEAD")
+        print("✅ Successfully applied PR #108 compatibility patch.")
+    except Exception as e:
+        print(f"⚠️ Warning: Could not apply PR #108 compatibility patch for ComfyUI-IPAdapter-Flux: {e}")
 
     # 4. ComfyUI-Newbie-Nodes
     newbie_nodes_path = os.path.join(APP_DIR, "custom_nodes", "ComfyUI-Newbie-Nodes")
@@ -82,6 +88,14 @@ def initialize_comfyui():
     else:
         print("✅ ComfyUI-Anima-LLLite extension already exists.")
 
+    # 6. comfyui-krea2-controlnet
+    krea2_controlnet_nodes_path = os.path.join(APP_DIR, "custom_nodes", "comfyui-krea2-controlnet")
+    if not os.path.exists(krea2_controlnet_nodes_path):
+        os.system(f"git clone https://github.com/facok/comfyui-krea2-controlnet.git {krea2_controlnet_nodes_path}")
+        print("✅ comfyui-krea2-controlnet extension cloned.")
+    else:
+        print("✅ comfyui-krea2-controlnet extension already exists.")
+
     print(f"✅ Current working directory is: {os.getcwd()}")
     
     import comfy.model_management

core/pipelines/pipeline_input_processor.py

@@ -199,6 +199,22 @@ def process_pipeline_inputs(ui_inputs: Dict[str, Any], progress: gr.Progress, wo
                     "image": os.path.basename(cn_temp_path), "strength": cn_strengths[i],
                     "control_net_name": cn_filepaths[i]
                 })
+                
+    krea2_controlnet_data = ui_inputs.get('krea2_controlnet_data', [])
+    active_krea2_controlnets = []
+    if krea2_controlnet_data:
+        (cn_images, _, _, cn_strengths, cn_filepaths) = [krea2_controlnet_data[i::5] for i in range(5)]
+        for i in range(len(cn_images)):
+            if cn_images[i] and cn_strengths[i] > 0 and cn_filepaths[i] and cn_filepaths[i] != "None":
+                ensure_controlnet_model_downloaded(cn_filepaths[i], progress)
+                if not os.path.exists(INPUT_DIR): os.makedirs(INPUT_DIR)
+                cn_temp_path = os.path.join(INPUT_DIR, f"temp_krea2_cn_{i}_{random.randint(1000, 9999)}.png")
+                cn_images[i].save(cn_temp_path, "PNG")
+                temp_files_to_clean.append(cn_temp_path)
+                active_krea2_controlnets.append({
+                    "image": os.path.basename(cn_temp_path), "strength": cn_strengths[i],
+                    "control_net_name": cn_filepaths[i]
+                })
     
     ipadapter_data = ui_inputs.get('ipadapter_data', [])
     active_ipadapters = []
@@ -288,9 +304,13 @@ def process_pipeline_inputs(ui_inputs: Dict[str, Any], progress: gr.Progress, wo
         num_units = len(style_data) // 2
         st_images = style_data[0*num_units : 1*num_units]
         st_strengths = style_data[1*num_units : 2*num_units]
+        style_models_downloaded = False
         for i in range(len(st_images)):
             if st_images[i] and st_strengths[i] > 0:
-                _ensure_model_downloaded("sigclip_vision_patch14_384.safetensors", progress)
+                if not style_models_downloaded:
+                    _ensure_model_downloaded("sigclip_vision_patch14_384.safetensors", progress)
+                    _ensure_model_downloaded("flux1-redux-dev.safetensors", progress)
+                    style_models_downloaded = True
                 temp_path = os.path.join(INPUT_DIR, f"temp_style_{i}_{random.randint(1000, 9999)}.png")
                 st_images[i].save(temp_path, "PNG")
                 temp_files_to_clean.append(temp_path)
@@ -351,6 +371,7 @@ def process_pipeline_inputs(ui_inputs: Dict[str, Any], progress: gr.Progress, wo
         "active_controlnets": active_controlnets,
         "active_anima_controlnets": active_anima_controlnets,
         "active_diffsynth_controlnets": active_diffsynth_controlnets,
+        "active_krea2_controlnets": active_krea2_controlnets,
         "active_ipadapters": active_ipadapters,
         "active_flux1_ipadapters": active_flux1_ipadapters,
         "active_sd3_ipadapters": active_sd3_ipadapters,

core/pipelines/sd_image_pipeline.py

@@ -109,6 +109,7 @@ class SdImagePipeline(BasePipeline):
             active_controlnets = processed["active_controlnets"]
             active_anima_controlnets = processed["active_anima_controlnets"]
             active_diffsynth_controlnets = processed["active_diffsynth_controlnets"]
+            active_krea2_controlnets = processed.get("active_krea2_controlnets", [])
             active_ipadapters = processed["active_ipadapters"]
             active_flux1_ipadapters = processed["active_flux1_ipadapters"]
             active_sd3_ipadapters = processed["active_sd3_ipadapters"]
@@ -163,6 +164,7 @@ class SdImagePipeline(BasePipeline):
                 "controlnet_chain": active_controlnets if not active_anima_controlnets else [],
                 "anima_controlnet_lllite_chain": active_anima_controlnets,
                 "diffsynth_controlnet_chain": active_diffsynth_controlnets,
+                "krea2_controlnet_chain": active_krea2_controlnets,
                 "ipadapter_chain": active_ipadapters,
                 "flux1_ipadapter_chain": active_flux1_ipadapters,
                 "sd3_ipadapter_chain": active_sd3_ipadapters,

core/pipelines/workflow_recipes/_partials/conditioning/krea-2.yaml

@@ -43,6 +43,11 @@ dynamic_lora_chains:
       "model": ["ksampler:model"]
       "clip": ["pos_prompt:clip", "neg_prompt:clip"]
 
+dynamic_krea2_controlnet_chains:
+  krea2_controlnet_chain:
+    ksampler_node: "ksampler"
+    vae_source: "vae_loader:0"
+
 dynamic_conditioning_chains:
   conditioning_chain:
     ksampler_node: "ksampler"

requirements.txt

@@ -1,6 +1,6 @@
 comfyui-frontend-package==1.45.20
-comfyui-workflow-templates==0.11.1
-comfyui-embedded-docs==0.5.6
+comfyui-workflow-templates==0.11.9
+comfyui-embedded-docs==0.5.8
 torch
 torchsde
 torchvision
@@ -22,7 +22,7 @@ alembic
 SQLAlchemy>=2.0.0
 filelock
 av>=16.0.0
-comfy-kitchen==0.2.16
+comfy-kitchen==0.2.19
 comfy-aimdo==0.4.10
 requests
 simpleeval>=1.0.0

ui/events/__init__.py

@@ -6,5 +6,7 @@ from .config_loaders import (
     get_anima_cn_defaults,
     load_diffsynth_controlnet_config,
     get_diffsynth_cn_defaults,
+    load_krea2_controlnet_config,
+    get_krea2_cn_defaults,
     load_ipadapter_config
 )

ui/events/chain_handlers.py

@@ -12,6 +12,7 @@ from .config_loaders import (
     load_controlnet_config,
     load_anima_controlnet_lllite_config,
     load_diffsynth_controlnet_config,
+    load_krea2_controlnet_config,
     load_ipadapter_config
 )
 
@@ -191,6 +192,99 @@ def create_controlnet_event_handlers(prefix, ui_components):
     )
 
 
+def create_krea2_controlnet_event_handlers(prefix, ui_components):
+    cn_rows = ui_components.get(f'krea2_controlnet_rows_{prefix}')
+    if not cn_rows: return
+    cn_types = ui_components[f'krea2_controlnet_types_{prefix}']
+    cn_series = ui_components[f'krea2_controlnet_series_{prefix}']
+    cn_filepaths = ui_components[f'krea2_controlnet_filepaths_{prefix}']
+    cn_images = ui_components[f'krea2_controlnet_images_{prefix}']
+    cn_strengths = ui_components[f'krea2_controlnet_strengths_{prefix}']
+    
+    count_state = ui_components[f'krea2_controlnet_count_state_{prefix}']
+    add_button = ui_components[f'add_krea2_controlnet_button_{prefix}']
+    del_button = ui_components[f'delete_krea2_controlnet_button_{prefix}']
+    accordion = ui_components[f'krea2_controlnet_accordion_{prefix}']
+
+    def add_cn_row(c):
+        c += 1
+        updates = {
+            count_state: c,
+            cn_rows[c-1]: gr.update(visible=True),
+            add_button: gr.update(visible=c < MAX_CONTROLNETS),
+            del_button: gr.update(visible=True)
+        }
+        return updates
+
+    def del_cn_row(c):
+        c -= 1
+        updates = {
+            count_state: c,
+            cn_rows[c]: gr.update(visible=False),
+            cn_images[c]: None,
+            cn_strengths[c]: 1.0,
+            add_button: gr.update(visible=True),
+            del_button: gr.update(visible=c > 0)
+        }
+        return updates
+        
+    add_outputs = [count_state, add_button, del_button] + cn_rows
+    del_outputs = [count_state, add_button, del_button] + cn_rows + cn_images + cn_strengths
+    add_button.click(fn=add_cn_row, inputs=[count_state], outputs=add_outputs, show_progress=False)
+    del_button.click(fn=del_cn_row, inputs=[count_state], outputs=del_outputs, show_progress=False)
+    
+    def on_cn_type_change(selected_type):
+        cn_config = load_krea2_controlnet_config()
+        series_choices = []
+        if selected_type:
+            series_choices = sorted(list(set(
+                model.get("Series", "Default") for model in cn_config
+                if selected_type in model.get("Type", [])
+            )))
+        default_series = series_choices[0] if series_choices else None
+        filepath = "None"
+        if default_series:
+            for model in cn_config:
+                if model.get("Series") == default_series and selected_type in model.get("Type", []):
+                    filepath = model.get("Filepath")
+                    break
+        return gr.update(choices=series_choices, value=default_series), filepath
+
+    def on_cn_series_change(selected_series, selected_type):
+        cn_config = load_krea2_controlnet_config()
+        filepath = "None"
+        if selected_series and selected_type:
+            for model in cn_config:
+                if model.get("Series") == selected_series and selected_type in model.get("Type", []):
+                    filepath = model.get("Filepath")
+                    break
+        return filepath
+        
+    for i in range(MAX_CONTROLNETS):
+        cn_types[i].change(
+            fn=on_cn_type_change,
+            inputs=[cn_types[i]],
+            outputs=[cn_series[i], cn_filepaths[i]],
+            show_progress=False
+        )
+        cn_series[i].change(
+            fn=on_cn_series_change,
+            inputs=[cn_series[i], cn_types[i]],
+            outputs=[cn_filepaths[i]],
+            show_progress=False
+        )
+
+    def on_accordion_expand(*images):
+        return [gr.update() for _ in images]
+    
+    accordion.expand(
+        fn=on_accordion_expand,
+        inputs=cn_images,
+        outputs=cn_images,
+        show_progress=False
+    )
+
+
 def create_anima_controlnet_lllite_event_handlers(prefix, ui_components):
     cn_rows = ui_components.get(f'anima_controlnet_lllite_rows_{prefix}')
     if not cn_rows: return
@@ -677,4 +771,4 @@ def create_conditioning_event_handlers(prefix, ui_components):
     add_outputs = [count_state, add_button, del_button] + rows
     del_outputs = [count_state, add_button, del_button] + rows + prompts
     add_button.click(fn=add_row, inputs=[count_state], outputs=add_outputs, show_progress=False)
-    del_button.click(fn=del_row, inputs=[count_state], outputs=del_outputs, show_progress=False)
+    del_button.click(fn=del_row, inputs=[count_state], outputs=del_outputs, show_progress=False)

ui/events/change_handlers.py

@@ -13,10 +13,11 @@ from .config_loaders import (
     get_cn_defaults,
     get_anima_cn_defaults,
     get_diffsynth_cn_defaults,
+    get_krea2_cn_defaults,
     load_ipadapter_config
 )
 
-def make_update_fn(m_comp, cat_comp, cs_comp, ar_comp, width_comp, height_comp, cn_types, cn_series, cn_filepaths, anima_cn_types, anima_cn_series, anima_cn_filepaths, diffsynth_cn_types, diffsynth_cn_series, diffsynth_cn_filepaths, ipa_preset, lora_acc, cn_acc, anima_cn_acc, diffsynth_cn_acc, ipa_acc, sd3_ipa_acc, flux1_ipa_acc, style_acc, embed_acc, cond_acc, ref_latent_acc, hidream_o1_ref_acc, guidance_comp, prompt_comp, neg_prompt_comp, steps_comp, cfg_comp, sampler_comp, scheduler_comp, pid_acc=None, vae_acc=None):
+def make_update_fn(m_comp, cat_comp, cs_comp, ar_comp, width_comp, height_comp, cn_types, cn_series, cn_filepaths, anima_cn_types, anima_cn_series, anima_cn_filepaths, diffsynth_cn_types, diffsynth_cn_series, diffsynth_cn_filepaths, krea2_cn_types, krea2_cn_series, krea2_cn_filepaths, ipa_preset, lora_acc, cn_acc, anima_cn_acc, diffsynth_cn_acc, krea2_cn_acc, ipa_acc, sd3_ipa_acc, flux1_ipa_acc, style_acc, embed_acc, cond_acc, ref_latent_acc, hidream_o1_ref_acc, guidance_comp, prompt_comp, neg_prompt_comp, steps_comp, cfg_comp, sampler_comp, scheduler_comp, pid_acc=None, vae_acc=None):
     def update_fn(*args):
         arch = args[0]
         category = args[1]
@@ -58,6 +59,7 @@ def make_update_fn(m_comp, cat_comp, cs_comp, ar_comp, width_comp, height_comp,
         if cn_acc: updates[cn_acc] = gr.update(visible=('controlnet' in enabled_chains))
         if anima_cn_acc: updates[anima_cn_acc] = gr.update(visible=('anima_controlnet_lllite' in enabled_chains))
         if diffsynth_cn_acc: updates[diffsynth_cn_acc] = gr.update(visible=('controlnet_model_patch' in enabled_chains))
+        if krea2_cn_acc: updates[krea2_cn_acc] = gr.update(visible=('krea2_controlnet' in enabled_chains))
         if ipa_acc: updates[ipa_acc] = gr.update(visible=('ipadapter' in enabled_chains))
         if flux1_ipa_acc: updates[flux1_ipa_acc] = gr.update(visible=('flux1_ipadapter' in enabled_chains))
         if sd3_ipa_acc: updates[sd3_ipa_acc] = gr.update(visible=('sd3_ipadapter' in enabled_chains))
@@ -110,6 +112,14 @@ def make_update_fn(m_comp, cat_comp, cs_comp, ar_comp, width_comp, height_comp,
             updates[s_comp] = gr.update(choices=diffsynth_series_choices, value=diffsynth_default_series)
         for f_comp in diffsynth_cn_filepaths:
             updates[f_comp] = diffsynth_filepath
+
+        krea2_all_types, krea2_default_type, krea2_series_choices, krea2_default_series, krea2_filepath = get_krea2_cn_defaults()
+        for t_comp in krea2_cn_types:
+            updates[t_comp] = gr.update(choices=krea2_all_types, value=krea2_default_type)
+        for s_comp in krea2_cn_series:
+            updates[s_comp] = gr.update(choices=krea2_series_choices, value=krea2_default_series)
+        for f_comp in krea2_cn_filepaths:
+            updates[f_comp] = krea2_filepath
         
         if ipa_preset and (arch_model_type in ["sdxl", "sd15", "sd35"]):
             config = load_ipadapter_config()
@@ -132,7 +142,7 @@ def make_update_fn(m_comp, cat_comp, cs_comp, ar_comp, width_comp, height_comp,
     return update_fn
 
 
-def make_model_change_fn(cat_comp_ref, cs_comp, ar_comp, width_comp, height_comp, cn_types, cn_series, cn_filepaths, anima_cn_types, anima_cn_series, anima_cn_filepaths, diffsynth_cn_types, diffsynth_cn_series, diffsynth_cn_filepaths, arch_comp_ref, ipa_preset, lora_acc, cn_acc, anima_cn_acc, diffsynth_cn_acc, ipa_acc, sd3_ipa_acc, flux1_ipa_acc, style_acc, embed_acc, cond_acc, ref_latent_acc, hidream_o1_ref_acc, guidance_comp, prompt_comp, neg_prompt_comp, steps_comp, cfg_comp, sampler_comp, scheduler_comp, pid_acc=None, vae_acc=None):
+def make_model_change_fn(cat_comp_ref, cs_comp, ar_comp, width_comp, height_comp, cn_types, cn_series, cn_filepaths, anima_cn_types, anima_cn_series, anima_cn_filepaths, diffsynth_cn_types, diffsynth_cn_series, diffsynth_cn_filepaths, krea2_cn_types, krea2_cn_series, krea2_cn_filepaths, arch_comp_ref, ipa_preset, lora_acc, cn_acc, anima_cn_acc, diffsynth_cn_acc, krea2_cn_acc, ipa_acc, sd3_ipa_acc, flux1_ipa_acc, style_acc, embed_acc, cond_acc, ref_latent_acc, hidream_o1_ref_acc, guidance_comp, prompt_comp, neg_prompt_comp, steps_comp, cfg_comp, sampler_comp, scheduler_comp, pid_acc=None, vae_acc=None):
     def change_fn(*args):
         model_name = args[0]
         idx = 1
@@ -179,6 +189,7 @@ def make_model_change_fn(cat_comp_ref, cs_comp, ar_comp, width_comp, height_comp
         if cn_acc: updates[cn_acc] = gr.update(visible=('controlnet' in enabled_chains))
         if anima_cn_acc: updates[anima_cn_acc] = gr.update(visible=('anima_controlnet_lllite' in enabled_chains))
         if diffsynth_cn_acc: updates[diffsynth_cn_acc] = gr.update(visible=('controlnet_model_patch' in enabled_chains))
+        if krea2_cn_acc: updates[krea2_cn_acc] = gr.update(visible=('krea2_controlnet' in enabled_chains))
         if ipa_acc: updates[ipa_acc] = gr.update(visible=('ipadapter' in enabled_chains))
         if flux1_ipa_acc: updates[flux1_ipa_acc] = gr.update(visible=('flux1_ipadapter' in enabled_chains))
         if sd3_ipa_acc: updates[sd3_ipa_acc] = gr.update(visible=('sd3_ipadapter' in enabled_chains))
@@ -231,6 +242,14 @@ def make_model_change_fn(cat_comp_ref, cs_comp, ar_comp, width_comp, height_comp
             updates[s_comp] = gr.update(choices=diffsynth_series_choices, value=diffsynth_default_series)
         for f_comp in diffsynth_cn_filepaths:
             updates[f_comp] = diffsynth_filepath
+
+        krea2_all_types, krea2_default_type, krea2_series_choices, krea2_default_series, krea2_filepath = get_krea2_cn_defaults()
+        for t_comp in krea2_cn_types:
+            updates[t_comp] = gr.update(choices=krea2_all_types, value=krea2_default_type)
+        for s_comp in krea2_cn_series:
+            updates[s_comp] = gr.update(choices=krea2_series_choices, value=krea2_default_series)
+        for f_comp in krea2_cn_filepaths:
+            updates[f_comp] = krea2_filepath
         
         if ipa_preset and (arch_model_type in ["sdxl", "sd15", "sd35"]):
             config = load_ipadapter_config()
@@ -262,6 +281,7 @@ def initialize_all_cn_dropdowns(ui_components):
     all_types, default_type, series_choices, default_series, filepath = get_cn_defaults(controlnet_key)
     anima_all_types, anima_default_type, anima_series_choices, anima_default_series, anima_filepath = get_anima_cn_defaults()
     diffsynth_all_types, diffsynth_default_type, diffsynth_series_choices, diffsynth_default_series, diffsynth_filepath = get_diffsynth_cn_defaults(controlnet_key)
+    krea2_all_types, krea2_default_type, krea2_series_choices, krea2_default_series, krea2_filepath = get_krea2_cn_defaults()
     
     updates = {}
     for prefix in ["txt2img", "img2img", "inpaint", "outpaint", "hires_fix"]:
@@ -288,6 +308,14 @@ def initialize_all_cn_dropdowns(ui_components):
                 updates[series_dd] = gr.update(choices=diffsynth_series_choices, value=default_series)
             for filepath_state in ui_components[f'diffsynth_controlnet_filepaths_{prefix}']:
                 updates[filepath_state] = diffsynth_filepath
+
+        if f'krea2_controlnet_types_{prefix}' in ui_components:
+            for type_dd in ui_components[f'krea2_controlnet_types_{prefix}']:
+                updates[type_dd] = gr.update(choices=krea2_all_types, value=krea2_default_type)
+            for series_dd in ui_components[f'krea2_controlnet_series_{prefix}']:
+                updates[series_dd] = gr.update(choices=krea2_series_choices, value=krea2_default_series)
+            for filepath_state in ui_components[f'krea2_controlnet_filepaths_{prefix}']:
+                updates[filepath_state] = krea2_filepath
                 
     return updates
 

ui/events/config_loaders.py

@@ -116,6 +116,43 @@ def get_diffsynth_cn_defaults(arch_val):
     return all_types, default_type, series_choices, default_series, filepath
 
 
+@lru_cache(maxsize=1)
+def load_krea2_controlnet_config():
+    _PROJECT_ROOT = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+    _CN_MODEL_LIST_PATH = os.path.join(_PROJECT_ROOT, 'yaml', 'krea2_controlnet_models.yaml')
+    try:
+        print("--- Loading krea2_controlnet_models.yaml ---")
+        with open(_CN_MODEL_LIST_PATH, 'r', encoding='utf-8') as f:
+            config = yaml.safe_load(f)
+        print("--- ✅ krea2_controlnet_models.yaml loaded successfully ---")
+        return config.get("Krea2_ControlNet", [])
+    except Exception as e:
+        print(f"Error loading krea2_controlnet_models.yaml: {e}")
+        return []
+
+def get_krea2_cn_defaults():
+    cn_config = load_krea2_controlnet_config()
+    if not cn_config:
+        return [], None, [], None, "None"
+        
+    all_types = sorted(list(set(t for model in cn_config for t in model.get("Type", []))))
+    default_type = all_types[0] if all_types else None
+    
+    series_choices = []
+    if default_type:
+        series_choices = sorted(list(set(model.get("Series", "Default") for model in cn_config if default_type in model.get("Type", []))))
+    default_series = series_choices[0] if series_choices else None
+    
+    filepath = "None"
+    if default_series and default_type:
+        for model in cn_config:
+            if model.get("Series") == default_series and default_type in model.get("Type", []):
+                filepath = model.get("Filepath")
+                break
+                
+    return all_types, default_type, series_choices, default_series, filepath
+
+
 @lru_cache(maxsize=1)
 def load_ipadapter_config():
     _PROJECT_ROOT = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
@@ -128,4 +165,4 @@ def load_ipadapter_config():
         return config
     except Exception as e:
         print(f"Error loading ipadapter.yaml: {e}")
-        return {}
+        return {}

ui/events/main.py

@@ -4,6 +4,7 @@ from .chain_handlers import (
     create_controlnet_event_handlers,
     create_anima_controlnet_lllite_event_handlers,
     create_diffsynth_controlnet_event_handlers,
+    create_krea2_controlnet_event_handlers,
     create_ipadapter_event_handlers,
     create_embedding_event_handlers,
     create_conditioning_event_handlers,
@@ -47,11 +48,16 @@ def attach_event_handlers(ui_components, demo):
         diffsynth_cn_types_list = ui_components.get(f'diffsynth_controlnet_types_{prefix}', [])
         diffsynth_cn_series_list = ui_components.get(f'diffsynth_controlnet_series_{prefix}', [])
         diffsynth_cn_filepaths_list = ui_components.get(f'diffsynth_controlnet_filepaths_{prefix}', [])
+
+        krea2_cn_types_list = ui_components.get(f'krea2_controlnet_types_{prefix}', [])
+        krea2_cn_series_list = ui_components.get(f'krea2_controlnet_series_{prefix}', [])
+        krea2_cn_filepaths_list = ui_components.get(f'krea2_controlnet_filepaths_{prefix}', [])
         
         lora_accordion = ui_components.get(f'lora_accordion_{prefix}')
         cn_accordion = ui_components.get(f'controlnet_accordion_{prefix}')
         anima_cn_accordion = ui_components.get(f'anima_controlnet_lllite_accordion_{prefix}')
         diffsynth_cn_accordion = ui_components.get(f'diffsynth_controlnet_accordion_{prefix}')
+        krea2_cn_accordion = ui_components.get(f'krea2_controlnet_accordion_{prefix}')
         ipa_accordion = ui_components.get(f'ipadapter_accordion_{prefix}')
         sd3_ipa_accordion = ui_components.get(f'sd3_ipadapter_accordion_{prefix}')
         flux1_ipa_accordion = ui_components.get(f'flux1_ipadapter_accordion_{prefix}')
@@ -83,10 +89,12 @@ def attach_event_handlers(ui_components, demo):
             outputs.extend(cn_types_list + cn_series_list + cn_filepaths_list)
             outputs.extend(anima_cn_types_list + anima_cn_series_list + anima_cn_filepaths_list)
             outputs.extend(diffsynth_cn_types_list + diffsynth_cn_series_list + diffsynth_cn_filepaths_list)
+            outputs.extend(krea2_cn_types_list + krea2_cn_series_list + krea2_cn_filepaths_list)
             if lora_accordion: outputs.append(lora_accordion)
             if cn_accordion: outputs.append(cn_accordion)
             if anima_cn_accordion: outputs.append(anima_cn_accordion)
             if diffsynth_cn_accordion: outputs.append(diffsynth_cn_accordion)
+            if krea2_cn_accordion: outputs.append(krea2_cn_accordion)
             if ipa_accordion: outputs.append(ipa_accordion)
             if sd3_ipa_accordion: outputs.append(sd3_ipa_accordion)
             if flux1_ipa_accordion: outputs.append(flux1_ipa_accordion)
@@ -106,7 +114,8 @@ def attach_event_handlers(ui_components, demo):
                 cn_types_list, cn_series_list, cn_filepaths_list, 
                 anima_cn_types_list, anima_cn_series_list, anima_cn_filepaths_list,
                 diffsynth_cn_types_list, diffsynth_cn_series_list, diffsynth_cn_filepaths_list, 
-                ipa_preset_list, lora_accordion, cn_accordion, anima_cn_accordion, diffsynth_cn_accordion, ipa_accordion, sd3_ipa_accordion, flux1_ipa_accordion, style_accordion, embedding_accordion, conditioning_accordion,
+                krea2_cn_types_list, krea2_cn_series_list, krea2_cn_filepaths_list,
+                ipa_preset_list, lora_accordion, cn_accordion, anima_cn_accordion, diffsynth_cn_accordion, krea2_cn_accordion, ipa_accordion, sd3_ipa_accordion, flux1_ipa_accordion, style_accordion, embedding_accordion, conditioning_accordion,
                 ref_latent_accordion, hidream_o1_ref_accordion, guidance_comp, prompt_comp, neg_prompt_comp, steps_comp, cfg_comp, sampler_comp, scheduler_comp,
                 pid_acc=pid_accordion, vae_acc=vae_accordion
             )
@@ -126,10 +135,12 @@ def attach_event_handlers(ui_components, demo):
             outputs2.extend(cn_types_list + cn_series_list + cn_filepaths_list)
             outputs2.extend(anima_cn_types_list + anima_cn_series_list + anima_cn_filepaths_list)
             outputs2.extend(diffsynth_cn_types_list + diffsynth_cn_series_list + diffsynth_cn_filepaths_list)
+            outputs2.extend(krea2_cn_types_list + krea2_cn_series_list + krea2_cn_filepaths_list)
             if lora_accordion: outputs2.append(lora_accordion)
             if cn_accordion: outputs2.append(cn_accordion)
             if anima_cn_accordion: outputs2.append(anima_cn_accordion)
             if diffsynth_cn_accordion: outputs2.append(diffsynth_cn_accordion)
+            if krea2_cn_accordion: outputs2.append(krea2_cn_accordion)
             if ipa_accordion: outputs2.append(ipa_accordion)
             if sd3_ipa_accordion: outputs2.append(sd3_ipa_accordion)
             if flux1_ipa_accordion: outputs2.append(flux1_ipa_accordion)
@@ -154,7 +165,8 @@ def attach_event_handlers(ui_components, demo):
                     cn_types_list, cn_series_list, cn_filepaths_list, 
                     anima_cn_types_list, anima_cn_series_list, anima_cn_filepaths_list,
                     diffsynth_cn_types_list, diffsynth_cn_series_list, diffsynth_cn_filepaths_list, 
-                    arch_comp, ipa_preset_list, lora_accordion, cn_accordion, anima_cn_accordion, diffsynth_cn_accordion, ipa_accordion, sd3_ipa_accordion, flux1_ipa_accordion, style_accordion, embedding_accordion, conditioning_accordion,
+                    krea2_cn_types_list, krea2_cn_series_list, krea2_cn_filepaths_list,
+                    arch_comp, ipa_preset_list, lora_accordion, cn_accordion, anima_cn_accordion, diffsynth_cn_accordion, krea2_cn_accordion, ipa_accordion, sd3_ipa_accordion, flux1_ipa_accordion, style_accordion, embedding_accordion, conditioning_accordion,
                     ref_latent_accordion, hidream_o1_ref_accordion, guidance_comp, prompt_comp, neg_prompt_comp, steps_comp, cfg_comp, sampler_comp, scheduler_comp,
                     pid_acc=pid_accordion, vae_acc=vae_accordion
                 )
@@ -164,6 +176,7 @@ def attach_event_handlers(ui_components, demo):
         create_controlnet_event_handlers(prefix, ui_components)
         create_anima_controlnet_lllite_event_handlers(prefix, ui_components)
         create_diffsynth_controlnet_event_handlers(prefix, ui_components)
+        create_krea2_controlnet_event_handlers(prefix, ui_components)
         create_ipadapter_event_handlers(prefix, ui_components)
         create_embedding_event_handlers(prefix, ui_components)
         create_conditioning_event_handlers(prefix, ui_components)
@@ -228,6 +241,10 @@ def attach_event_handlers(ui_components, demo):
             all_load_outputs.extend(ui_components[f'diffsynth_controlnet_types_{prefix}'])
             all_load_outputs.extend(ui_components[f'diffsynth_controlnet_series_{prefix}'])
             all_load_outputs.extend(ui_components[f'diffsynth_controlnet_filepaths_{prefix}'])
+        if f'krea2_controlnet_types_{prefix}' in ui_components:
+            all_load_outputs.extend(ui_components[f'krea2_controlnet_types_{prefix}'])
+            all_load_outputs.extend(ui_components[f'krea2_controlnet_series_{prefix}'])
+            all_load_outputs.extend(ui_components[f'krea2_controlnet_filepaths_{prefix}'])
         if f'ipadapter_final_preset_{prefix}' in ui_components:
             all_load_outputs.extend(ui_components[f'ipadapter_lora_strengths_{prefix}'])
             all_load_outputs.append(ui_components[f'ipadapter_final_preset_{prefix}'])

ui/events/run_handlers.py

@@ -43,6 +43,7 @@ def create_run_event(prefix: str, task_type: str, ui_components: dict):
     controlnet_data_components = ui_components.get(f'all_controlnet_components_flat_{prefix}', [])
     anima_controlnet_lllite_data_components = ui_components.get(f'all_anima_controlnet_lllite_components_flat_{prefix}', [])
     diffsynth_controlnet_data_components = ui_components.get(f'all_diffsynth_controlnet_components_flat_{prefix}', [])
+    krea2_controlnet_data_components = ui_components.get(f'all_krea2_controlnet_components_flat_{prefix}', [])
     ipadapter_data_components = ui_components.get(f'all_ipadapter_components_flat_{prefix}', [])
     sd3_ipadapter_data_components = ui_components.get(f'all_sd3_ipadapter_components_flat_{prefix}', [])
     flux1_ipadapter_data_components = ui_components.get(f'all_flux1_ipadapter_components_flat_{prefix}', [])
@@ -59,7 +60,7 @@ def create_run_event(prefix: str, task_type: str, ui_components: dict):
     input_keys = list(run_inputs_map.keys())
     input_list_flat = [v for v in run_inputs_map.values() if v is not None]
     all_chains = [
-        lora_data_components, controlnet_data_components, anima_controlnet_lllite_data_components, diffsynth_controlnet_data_components, ipadapter_data_components,
+        lora_data_components, controlnet_data_components, anima_controlnet_lllite_data_components, diffsynth_controlnet_data_components, krea2_controlnet_data_components, ipadapter_data_components,
         sd3_ipadapter_data_components, flux1_ipadapter_data_components, style_data_components,
         embedding_data_components, conditioning_data_components, reference_latent_data_components, hidream_o1_reference_data_components
     ]
@@ -82,6 +83,7 @@ def create_run_event(prefix: str, task_type: str, ui_components: dict):
         assign_chain_data('controlnet_data', controlnet_data_components)
         assign_chain_data('anima_controlnet_lllite_data', anima_controlnet_lllite_data_components)
         assign_chain_data('diffsynth_controlnet_data', diffsynth_controlnet_data_components)
+        assign_chain_data('krea2_controlnet_data', krea2_controlnet_data_components)
         assign_chain_data('ipadapter_data', ipadapter_data_components)
         assign_chain_data('sd3_ipadapter_chain', sd3_ipadapter_data_components)
         assign_chain_data('flux1_ipadapter_data', flux1_ipadapter_data_components)

ui/shared/hires_fix_ui.py

@@ -3,7 +3,7 @@ from core.settings import MODEL_MAP_CHECKPOINT, MODEL_DEFAULTS_CONFIG
 from comfy_integration.nodes import SAMPLER_CHOICES, SCHEDULER_CHOICES
 from .ui_components import (
     create_lora_settings_ui,
-    create_controlnet_ui, create_anima_controlnet_lllite_ui, create_ipadapter_ui, create_embedding_ui,
+    create_controlnet_ui, create_anima_controlnet_lllite_ui, create_diffsynth_controlnet_ui, create_krea2_controlnet_ui, create_ipadapter_ui, create_embedding_ui,
     create_conditioning_ui, create_vae_override_ui,
     create_model_architecture_filter_ui, create_category_filter_ui,
     create_sd3_ipadapter_ui, create_flux1_ipadapter_ui, create_style_ui,
@@ -91,6 +91,8 @@ def create_ui():
         components.update(create_lora_settings_ui(prefix))
         components.update(create_controlnet_ui(prefix))
         components.update(create_anima_controlnet_lllite_ui(prefix))
+        components.update(create_diffsynth_controlnet_ui(prefix))
+        components.update(create_krea2_controlnet_ui(prefix))
         components.update(create_ipadapter_ui(prefix))
         components.update(create_flux1_ipadapter_ui(prefix))
         components.update(create_sd3_ipadapter_ui(prefix))

ui/shared/img2img_ui.py

@@ -3,7 +3,7 @@ from core.settings import MODEL_MAP_CHECKPOINT, MODEL_DEFAULTS_CONFIG
 from comfy_integration.nodes import SAMPLER_CHOICES, SCHEDULER_CHOICES
 from .ui_components import (
     create_lora_settings_ui,
-    create_controlnet_ui, create_anima_controlnet_lllite_ui, create_diffsynth_controlnet_ui, create_ipadapter_ui, create_embedding_ui,
+    create_controlnet_ui, create_anima_controlnet_lllite_ui, create_diffsynth_controlnet_ui, create_krea2_controlnet_ui, create_ipadapter_ui, create_embedding_ui,
     create_conditioning_ui, create_vae_override_ui,
     create_model_architecture_filter_ui, create_category_filter_ui,
     create_sd3_ipadapter_ui, create_flux1_ipadapter_ui, create_style_ui,
@@ -73,6 +73,7 @@ def create_ui():
         components.update(create_controlnet_ui(prefix))
         components.update(create_anima_controlnet_lllite_ui(prefix))
         components.update(create_diffsynth_controlnet_ui(prefix))
+        components.update(create_krea2_controlnet_ui(prefix))
         components.update(create_ipadapter_ui(prefix))
         components.update(create_flux1_ipadapter_ui(prefix))
         components.update(create_sd3_ipadapter_ui(prefix))

ui/shared/inpaint_ui.py

@@ -2,7 +2,7 @@ import gradio as gr
 from core.settings import MODEL_MAP_CHECKPOINT, MODEL_DEFAULTS_CONFIG
 from .ui_components import (
     create_base_parameter_ui, create_lora_settings_ui,
-    create_controlnet_ui, create_anima_controlnet_lllite_ui, create_diffsynth_controlnet_ui, create_ipadapter_ui, create_embedding_ui,
+    create_controlnet_ui, create_anima_controlnet_lllite_ui, create_diffsynth_controlnet_ui, create_krea2_controlnet_ui, create_ipadapter_ui, create_embedding_ui,
     create_conditioning_ui, create_vae_override_ui,
     create_model_architecture_filter_ui, create_category_filter_ui,
     create_sd3_ipadapter_ui, create_flux1_ipadapter_ui, create_style_ui,
@@ -105,6 +105,7 @@ def create_ui():
             components.update(create_controlnet_ui(prefix))
             components.update(create_anima_controlnet_lllite_ui(prefix))
             components.update(create_diffsynth_controlnet_ui(prefix))
+            components.update(create_krea2_controlnet_ui(prefix))
             components.update(create_ipadapter_ui(prefix))
             components.update(create_flux1_ipadapter_ui(prefix))
             components.update(create_sd3_ipadapter_ui(prefix))

ui/shared/outpaint_ui.py

@@ -3,7 +3,7 @@ from core.settings import MODEL_MAP_CHECKPOINT, MODEL_DEFAULTS_CONFIG
 from comfy_integration.nodes import SAMPLER_CHOICES, SCHEDULER_CHOICES
 from .ui_components import (
     create_lora_settings_ui,
-    create_controlnet_ui, create_anima_controlnet_lllite_ui, create_diffsynth_controlnet_ui, create_ipadapter_ui, create_embedding_ui,
+    create_controlnet_ui, create_anima_controlnet_lllite_ui, create_diffsynth_controlnet_ui, create_krea2_controlnet_ui, create_ipadapter_ui, create_embedding_ui,
     create_conditioning_ui, create_vae_override_ui,
     create_model_architecture_filter_ui, create_category_filter_ui,
     create_sd3_ipadapter_ui, create_flux1_ipadapter_ui, create_style_ui,
@@ -88,6 +88,7 @@ def create_ui():
         components.update(create_controlnet_ui(prefix))
         components.update(create_anima_controlnet_lllite_ui(prefix))
         components.update(create_diffsynth_controlnet_ui(prefix))
+        components.update(create_krea2_controlnet_ui(prefix))
         components.update(create_ipadapter_ui(prefix))
         components.update(create_flux1_ipadapter_ui(prefix))
         components.update(create_sd3_ipadapter_ui(prefix))

ui/shared/txt2img_ui.py

@@ -2,7 +2,7 @@ import gradio as gr
 from core.settings import MODEL_MAP_CHECKPOINT
 from .ui_components import (
     create_base_parameter_ui, create_lora_settings_ui,
-    create_controlnet_ui, create_anima_controlnet_lllite_ui, create_diffsynth_controlnet_ui, create_ipadapter_ui, create_embedding_ui,
+    create_controlnet_ui, create_anima_controlnet_lllite_ui, create_diffsynth_controlnet_ui, create_krea2_controlnet_ui, create_ipadapter_ui, create_embedding_ui,
     create_conditioning_ui, create_vae_override_ui,
     create_model_architecture_filter_ui, create_category_filter_ui,
     create_sd3_ipadapter_ui, create_flux1_ipadapter_ui, create_style_ui,
@@ -45,6 +45,7 @@ def create_ui():
         components.update(create_controlnet_ui(prefix))
         components.update(create_anima_controlnet_lllite_ui(prefix))
         components.update(create_diffsynth_controlnet_ui(prefix))
+        components.update(create_krea2_controlnet_ui(prefix))
         components.update(create_ipadapter_ui(prefix))
         components.update(create_flux1_ipadapter_ui(prefix))
         components.update(create_sd3_ipadapter_ui(prefix))

ui/shared/ui_components.py

@@ -203,6 +203,49 @@ def create_controlnet_ui(prefix: str, max_units=MAX_CONTROLNETS):
 
     return components
 
+def create_krea2_controlnet_ui(prefix: str, max_units=MAX_CONTROLNETS):
+    components = {}
+    key = lambda name: f"{name}_{prefix}"
+
+    with gr.Accordion("Krea2 ControlNet Settings", open=False, visible=('krea2_controlnet' in default_enabled_chains)) as accordion:
+        components[key('krea2_controlnet_accordion')] = accordion
+        gr.Markdown("💡 **Tip:** Processed using the [facok/comfyui-krea2-controlnet](https://github.com/facok/comfyui-krea2-controlnet) node.")       
+        
+        cn_rows, images, series, types, strengths, filepaths = [], [], [], [], [], []
+        components.update({
+            key('krea2_controlnet_rows'): cn_rows,
+            key('krea2_controlnet_images'): images,
+            key('krea2_controlnet_series'): series,
+            key('krea2_controlnet_types'): types,
+            key('krea2_controlnet_strengths'): strengths,
+            key('krea2_controlnet_filepaths'): filepaths
+        })
+        
+        for i in range(max_units):
+            with gr.Row(visible=(i < 1)) as row:
+                with gr.Column(scale=1):
+                    images.append(gr.Image(label=f"Control Image {i+1}", type="pil", sources=["upload"], height=256))
+                with gr.Column(scale=2):
+                    types.append(gr.Dropdown(label="Type", choices=[], interactive=True, allow_custom_value=True))
+                    series.append(gr.Dropdown(label="Series", choices=[], interactive=True, allow_custom_value=True))
+                    strengths.append(gr.Slider(label="Strength", minimum=0.0, maximum=2.0, step=0.05, value=1.0, interactive=True))
+                    filepaths.append(gr.State(None))
+                cn_rows.append(row)
+
+        with gr.Row():
+            components[key('add_krea2_controlnet_button')] = gr.Button("✚ Add Krea2 ControlNet")
+            components[key('delete_krea2_controlnet_button')] = gr.Button("➖ Delete Krea2 ControlNet", visible=False)
+        components[key('krea2_controlnet_count_state')] = gr.State(1)
+        
+        all_cn_components_flat = []
+        for i in range(max_units):
+            all_cn_components_flat.extend([
+                images[i], types[i], series[i], strengths[i], filepaths[i]
+            ])
+        components[key('all_krea2_controlnet_components_flat')] = all_cn_components_flat
+
+    return components
+
 def create_anima_controlnet_lllite_ui(prefix: str, max_units=MAX_CONTROLNETS):
     components = {}
     key = lambda name: f"{name}_{prefix}"

yaml/file_list.yaml

@@ -718,6 +718,11 @@ file:
       repo_id: "black-forest-labs/FLUX.1-Redux-dev"
       repository_file_path: "flux1-redux-dev.safetensors"
   loras: 
+    # Krea2 ControlNet
+    - filename: "depth-control-lora.safetensors"
+      source: "hf"
+      repo_id: "Patil/Krea-2-depth-controlnet"
+      repository_file_path: "depth-control-lora.safetensors"
     # SD1.5 FaceID
     - filename: "ip-adapter-faceid_sd15_lora.safetensors"
       source: "hf"

yaml/image_gen_features.yaml

@@ -11,6 +11,7 @@ default:
 krea-2:
   enabled_chains:
     - lora
+    - krea2_controlnet
     - conditioning
     - pid
 

yaml/injectors.yaml

@@ -7,6 +7,8 @@ injector_definitions:
     module: "chain_injectors.newbie_lora_injector"
   dynamic_controlnet_chains: 
     module: "chain_injectors.controlnet_injector"
+  dynamic_krea2_controlnet_chains:
+    module: "chain_injectors.krea2_controlnet_injector"
   dynamic_anima_controlnet_lllite_chains:
     module: "chain_injectors.anima_controlnet_lllite_injector"
   dynamic_diffsynth_controlnet_chains:
@@ -42,6 +44,7 @@ injector_order:
   - dynamic_conditioning_chains
   - dynamic_style_chains
   - dynamic_controlnet_chains
+  - dynamic_krea2_controlnet_chains
   - dynamic_anima_controlnet_lllite_chains
   - dynamic_hidream_o1_smoothing_chains
   - dynamic_hidream_o1_reference_chains

yaml/krea2_controlnet_models.yaml

@@ -0,0 +1,4 @@
+Krea2_ControlNet:
+  - Filepath: "depth-control-lora.safetensors"
+    Series: "Patil"
+    Type: ["Depth"]