Add Boogu Edit Injector.

README.md

@@ -137,6 +137,8 @@ models:
   - black-forest-labs/FLUX.1-schnell
   - Boogu/Boogu-Image-0.1-Turbo
   - Boogu/Boogu-Image-0.1-Base
+  - Boogu/Boogu-Image-0.1-Edit
+  - Boogu/Boogu-Image-0.1-Edit-Turbo
   - HiDream-ai/HiDream-I1-Dev
   - HiDream-ai/HiDream-I1-Fast
   - HiDream-ai/HiDream-I1-Full

chain_injectors/boogu_edit_injector.py

@@ -0,0 +1,128 @@
+import os
+
+def create_node(assembler, class_type, title):
+    try:
+        node = assembler._get_node_template(class_type)
+    except Exception:
+        node = {
+            "inputs": {},
+            "class_type": class_type,
+            "_meta": {"title": title}
+        }
+    node['_meta']['title'] = title
+    return node
+
+def inject(assembler, chain_definition, chain_items):
+    if not chain_items:
+        return
+
+    valid_images = []
+    for item in chain_items:
+        if not item:
+            continue
+        img_path = item
+        if isinstance(item, dict):
+            img_path = item.get('image') or item.get('filename') or item.get('path')
+        if img_path:
+            valid_images.append(img_path)
+
+    if not valid_images:
+        return
+
+    valid_images = valid_images[:10]
+
+    ksampler_name = chain_definition.get('ksampler_node', 'ksampler')
+    pos_prompt_name = chain_definition.get('pos_prompt_node', 'pos_prompt')
+    neg_prompt_name = chain_definition.get('neg_prompt_node', 'neg_prompt')
+    clip_loader_name = chain_definition.get('clip_loader_node', 'clip_loader')
+    vae_loader_name = chain_definition.get('vae_loader_node', 'vae_loader')
+
+    if ksampler_name not in assembler.node_map:
+        print(f"Warning: Target node '{ksampler_name}' for Boogu Edit 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_connection = None
+    if vae_loader_name in assembler.node_map:
+        vae_connection = [assembler.node_map[vae_loader_name], 0]
+    else:
+        for node_id, node in assembler.workflow.items():
+            if isinstance(node, dict) and node.get('class_type') == 'VAELoader':
+                vae_connection = [node_id, 0]
+                break
+
+    clip_connection = None
+    if clip_loader_name in assembler.node_map:
+        clip_connection = [assembler.node_map[clip_loader_name], 0]
+    elif pos_prompt_name in assembler.node_map:
+        pos_id = assembler.node_map[pos_prompt_name]
+        clip_connection = assembler.workflow[pos_id]['inputs'].get('clip')
+
+    pos_prompt_id = assembler.node_map.get(pos_prompt_name)
+    neg_prompt_id = assembler.node_map.get(neg_prompt_name)
+
+    pos_text = ""
+    if pos_prompt_id and pos_prompt_id in assembler.workflow:
+        pos_text = assembler.workflow[pos_prompt_id]['inputs'].get('text', '')
+    elif hasattr(assembler, 'ui_values') and isinstance(assembler.ui_values, dict):
+        pos_text = assembler.ui_values.get('positive_prompt') or assembler.ui_values.get('prompt') or ''
+
+    if not pos_text:
+        for node_id, node in assembler.workflow.items():
+            if isinstance(node, dict):
+                cls = node.get('class_type', '')
+                if cls in ['TextEncodeBooguEdit', 'Krea2EditGroundedEncode', 'TextEncodeQwenImageEditPlus', 'CLIPTextEncode']:
+                    t = node.get('inputs', {}).get('prompt') or node.get('inputs', {}).get('text')
+                    if t:
+                        pos_text = t
+                        break
+
+    neg_text = ""
+    if neg_prompt_id and neg_prompt_id in assembler.workflow:
+        neg_text = assembler.workflow[neg_prompt_id]['inputs'].get('text', '')
+    elif hasattr(assembler, 'ui_values') and isinstance(assembler.ui_values, dict):
+        neg_text = assembler.ui_values.get('negative_prompt') or assembler.ui_values.get('neg_prompt') or ''
+
+    scaled_image_ids = []
+    for i, img_filename in enumerate(valid_images):
+        load_id = assembler._get_unique_id()
+        load_node = create_node(assembler, "LoadImage", f"Load Reference Image {i+1}")
+        load_node['inputs']['image'] = img_filename
+        assembler.workflow[load_id] = load_node
+
+        scale_id = assembler._get_unique_id()
+        scale_node = create_node(assembler, "ImageScaleToTotalPixels", f"Scale Reference {i+1}")
+        scale_node['inputs']['upscale_method'] = "nearest-exact"
+        scale_node['inputs']['megapixels'] = 1
+        scale_node['inputs']['resolution_steps'] = 1
+        scale_node['inputs']['image'] = [load_id, 0]
+        assembler.workflow[scale_id] = scale_node
+        scaled_image_ids.append(scale_id)
+
+    boogu_encode_id = assembler._get_unique_id()
+    boogu_encode_node = create_node(assembler, "TextEncodeBooguEdit", "TextEncodeBooguEdit")
+    boogu_encode_node['inputs']['prompt'] = pos_text
+    boogu_encode_node['inputs']['negative_prompt'] = neg_text
+    if clip_connection:
+        boogu_encode_node['inputs']['clip'] = clip_connection
+    if vae_connection:
+        boogu_encode_node['inputs']['vae'] = vae_connection
+    for idx, s_id in enumerate(scaled_image_ids):
+        boogu_encode_node['inputs'][f"images.image_{idx+1}"] = [s_id, 0]
+    assembler.workflow[boogu_encode_id] = boogu_encode_node
+
+    assembler.workflow[ksampler_id]['inputs']['positive'] = [boogu_encode_id, 0]
+    assembler.workflow[ksampler_id]['inputs']['negative'] = [boogu_encode_id, 1]
+
+    if pos_prompt_id and pos_prompt_id in assembler.workflow:
+        del assembler.workflow[pos_prompt_id]
+
+    if neg_prompt_id and neg_prompt_id in assembler.workflow:
+        del assembler.workflow[neg_prompt_id]
+
+    print(f"Boogu Edit injector applied with {len(valid_images)} reference image(s). Original CLIPTextEncode nodes replaced.")

core/pipelines/pipeline_input_processor.py

@@ -390,6 +390,17 @@ def process_pipeline_inputs(ui_inputs: Dict[str, Any], progress: gr.Progress, wo
                 temp_files_to_clean.append(temp_path)
                 active_qwen_image_edit.append(os.path.basename(temp_path))
 
+    boogu_edit_data = ui_inputs.get('boogu_edit_data', [])
+    active_boogu_edit = []
+    if boogu_edit_data:
+        for img in boogu_edit_data:
+            if img:
+                if not os.path.exists(INPUT_DIR): os.makedirs(INPUT_DIR)
+                temp_path = os.path.join(INPUT_DIR, f"temp_boogu_edit_ref_{random.randint(1000, 9999)}.png")
+                img.save(temp_path, "PNG")
+                temp_files_to_clean.append(temp_path)
+                active_boogu_edit.append(os.path.basename(temp_path))
+
     reference_image_data = ui_inputs.get('reference_image_data', [])
     active_reference_images = []
     if reference_image_data:
@@ -446,6 +457,7 @@ def process_pipeline_inputs(ui_inputs: Dict[str, Any], progress: gr.Progress, wo
         "active_krea2_identity_edit": active_krea2_identity_edit,
         "active_krea2_reference_edit": active_krea2_reference_edit,
         "active_qwen_image_edit": active_qwen_image_edit,
+        "active_boogu_edit": active_boogu_edit,
         "active_reference_images": active_reference_images,
         "active_conditioning": active_conditioning,
         "temp_files_to_clean": temp_files_to_clean

core/pipelines/sd_image_pipeline.py

@@ -121,6 +121,7 @@ class SdImagePipeline(BasePipeline):
             active_krea2_identity_edit = processed.get("active_krea2_identity_edit", [])
             active_krea2_reference_edit = processed.get("active_krea2_reference_edit", [])
             active_qwen_image_edit = processed.get("active_qwen_image_edit", [])
+            active_boogu_edit = processed.get("active_boogu_edit", [])
             active_reference_images = processed.get("active_reference_images", [])
             active_conditioning = processed["active_conditioning"]
 
@@ -183,6 +184,7 @@ class SdImagePipeline(BasePipeline):
                 "krea2_identity_edit_chain": active_krea2_identity_edit,
                 "krea2_reference_edit_chain": active_krea2_reference_edit,
                 "qwen_image_edit_chain": active_qwen_image_edit,
+                "boogu_edit_chain": active_boogu_edit,
                 "reference_image_chain": active_reference_images,
                 "vae_chain": [ui_inputs.get('vae_name')] if ui_inputs.get('vae_name') else [],
                 "hidream_o1_smoothing_chain": hidream_o1_smoothing_data,

core/pipelines/workflow_recipes/_partials/conditioning/boogu-image.yaml

@@ -54,6 +54,14 @@ dynamic_conditioning_chains:
     ksampler_node: "ksampler"
     clip_source: "clip_loader:0"
 
+dynamic_boogu_edit_chains:
+  boogu_edit_chain:
+    ksampler_node: "ksampler"
+    pos_prompt_node: "pos_prompt"
+    neg_prompt_node: "neg_prompt"
+    clip_loader_node: "clip_loader"
+    vae_loader_node: "vae_loader"
+
 dynamic_pid_chains:
   pid_chain:
     ksampler_node: "ksampler"

requirements.txt

@@ -1,5 +1,5 @@
 comfyui-frontend-package==1.47.11
-comfyui-workflow-templates==0.11.20
+comfyui-workflow-templates==0.11.23
 comfyui-embedded-docs==0.5.9
 torch
 torchsde
@@ -22,7 +22,7 @@ alembic
 SQLAlchemy>=2.0.0
 filelock
 av>=16.0.0
-comfy-kitchen==0.2.25
+comfy-kitchen==0.2.26
 comfy-aimdo==0.4.10
 requests
 simpleeval>=1.0.0

ui/events/chain_handlers.py

@@ -952,6 +952,39 @@ def create_krea2_reference_edit_event_handlers(prefix, ui_components):
             del_button: gr.update(visible=c > 0),
         }
 
+    add_outputs = [count_state, add_button, del_button] + ref_rows
+    del_outputs = [count_state, add_button, del_button] + ref_rows + images
+    add_button.click(fn=add_ref_row, inputs=[count_state], outputs=add_outputs, show_progress=False)
+    del_button.click(fn=del_ref_row, inputs=[count_state], outputs=del_outputs, show_progress=False)
+
+
+def create_boogu_edit_event_handlers(prefix, ui_components, max_units=10):
+    ref_rows = ui_components.get(f'boogu_edit_rows_{prefix}')
+    if not ref_rows: return
+    count_state = ui_components[f'boogu_edit_count_state_{prefix}']
+    add_button = ui_components[f'add_boogu_edit_button_{prefix}']
+    del_button = ui_components[f'delete_boogu_edit_button_{prefix}']
+    images = ui_components[f'boogu_edit_images_{prefix}']
+
+    def add_ref_row(c):
+        c += 1
+        return {
+            count_state: c,
+            ref_rows[c - 1]: gr.update(visible=True),
+            add_button: gr.update(visible=c < max_units),
+            del_button: gr.update(visible=True),
+        }
+
+    def del_ref_row(c):
+        c -= 1
+        return {
+            count_state: c,
+            ref_rows[c]: gr.update(visible=False),
+            images[c]: None,
+            add_button: gr.update(visible=True),
+            del_button: gr.update(visible=c > 1),
+        }
+
     add_outputs = [count_state, add_button, del_button] + ref_rows
     del_outputs = [count_state, add_button, del_button] + ref_rows + images
     add_button.click(fn=add_ref_row, inputs=[count_state], outputs=add_outputs, show_progress=False)

ui/events/change_handlers.py

@@ -17,7 +17,7 @@ from .config_loaders import (
     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, 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, joyai_ref_acc=None, krea2_identity_edit_acc=None, krea2_reference_edit_acc=None, qwen_image_edit_acc=None, ref_img_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, joyai_ref_acc=None, krea2_identity_edit_acc=None, krea2_reference_edit_acc=None, qwen_image_edit_acc=None, boogu_edit_acc=None, ref_img_acc=None):
     def update_fn(*args):
         arch = args[0]
         category = args[1]
@@ -72,6 +72,7 @@ def make_update_fn(m_comp, cat_comp, cs_comp, ar_comp, width_comp, height_comp,
         if krea2_identity_edit_acc: updates[krea2_identity_edit_acc] = gr.update(visible=('krea2_identity_edit' in enabled_chains))
         if krea2_reference_edit_acc: updates[krea2_reference_edit_acc] = gr.update(visible=('krea2_reference_edit' in enabled_chains))
         if qwen_image_edit_acc: updates[qwen_image_edit_acc] = gr.update(visible=('qwen_image_edit' in enabled_chains))
+        if boogu_edit_acc: updates[boogu_edit_acc] = gr.update(visible=('boogu_edit' in enabled_chains))
         if ref_img_acc: updates[ref_img_acc] = gr.update(visible=('reference_image' in enabled_chains))
         if pid_acc: updates[pid_acc] = gr.update(visible=('pid' in enabled_chains))
         if vae_acc: updates[vae_acc] = gr.update(visible=('vae' in enabled_chains))
@@ -147,7 +148,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, 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, joyai_ref_acc=None, krea2_identity_edit_acc=None, krea2_reference_edit_acc=None, qwen_image_edit_acc=None, ref_img_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, joyai_ref_acc=None, krea2_identity_edit_acc=None, krea2_reference_edit_acc=None, qwen_image_edit_acc=None, boogu_edit_acc=None, ref_img_acc=None):
     def change_fn(*args):
         model_name = args[0]
         idx = 1
@@ -207,6 +208,7 @@ def make_model_change_fn(cat_comp_ref, cs_comp, ar_comp, width_comp, height_comp
         if krea2_identity_edit_acc: updates[krea2_identity_edit_acc] = gr.update(visible=('krea2_identity_edit' in enabled_chains))
         if krea2_reference_edit_acc: updates[krea2_reference_edit_acc] = gr.update(visible=('krea2_reference_edit' in enabled_chains))
         if qwen_image_edit_acc: updates[qwen_image_edit_acc] = gr.update(visible=('qwen_image_edit' in enabled_chains))
+        if boogu_edit_acc: updates[boogu_edit_acc] = gr.update(visible=('boogu_edit' in enabled_chains))
         if ref_img_acc: updates[ref_img_acc] = gr.update(visible=('reference_image' in enabled_chains))
         if pid_acc: updates[pid_acc] = gr.update(visible=('pid' in enabled_chains))
         if vae_acc: updates[vae_acc] = gr.update(visible=('vae' in enabled_chains))

ui/events/main.py

@@ -17,6 +17,7 @@ from .chain_handlers import (
     create_krea2_identity_edit_event_handlers,
     create_krea2_reference_edit_event_handlers,
     create_qwen_image_edit_event_handlers,
+    create_boogu_edit_event_handlers,
     create_reference_image_event_handlers
 )
 from .change_handlers import (
@@ -75,6 +76,7 @@ def attach_event_handlers(ui_components, demo):
         krea2_identity_edit_accordion = ui_components.get(f'krea2_identity_edit_accordion_{prefix}')
         krea2_reference_edit_accordion = ui_components.get(f'krea2_reference_edit_accordion_{prefix}')
         qwen_image_edit_accordion = ui_components.get(f'qwen_image_edit_accordion_{prefix}')
+        boogu_edit_accordion = ui_components.get(f'boogu_edit_accordion_{prefix}')
         ref_img_accordion = ui_components.get(f'reference_image_accordion_{prefix}')
         pid_accordion = ui_components.get(f'pid_accordion_{prefix}')
         vae_accordion = ui_components.get(f'vae_accordion_{prefix}')
@@ -117,6 +119,7 @@ def attach_event_handlers(ui_components, demo):
             if krea2_identity_edit_accordion: outputs.append(krea2_identity_edit_accordion)
             if krea2_reference_edit_accordion: outputs.append(krea2_reference_edit_accordion)
             if qwen_image_edit_accordion: outputs.append(qwen_image_edit_accordion)
+            if boogu_edit_accordion: outputs.append(boogu_edit_accordion)
             if ref_img_accordion: outputs.append(ref_img_accordion)
             if pid_accordion: outputs.append(pid_accordion)
             if vae_accordion: outputs.append(vae_accordion)
@@ -132,7 +135,7 @@ def attach_event_handlers(ui_components, demo):
                 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, joyai_ref_acc=joyai_ref_accordion, krea2_identity_edit_acc=krea2_identity_edit_accordion, krea2_reference_edit_acc=krea2_reference_edit_accordion, qwen_image_edit_acc=qwen_image_edit_accordion, ref_img_acc=ref_img_accordion
+                pid_acc=pid_accordion, vae_acc=vae_accordion, joyai_ref_acc=joyai_ref_accordion, krea2_identity_edit_acc=krea2_identity_edit_accordion, krea2_reference_edit_acc=krea2_reference_edit_accordion, qwen_image_edit_acc=qwen_image_edit_accordion, boogu_edit_acc=boogu_edit_accordion, ref_img_acc=ref_img_accordion
             )
             inputs = [arch_comp, cat_comp]
             if aspect_ratio_comp:
@@ -168,6 +171,7 @@ def attach_event_handlers(ui_components, demo):
             if krea2_identity_edit_accordion: outputs2.append(krea2_identity_edit_accordion)
             if krea2_reference_edit_accordion: outputs2.append(krea2_reference_edit_accordion)
             if qwen_image_edit_accordion: outputs2.append(qwen_image_edit_accordion)
+            if boogu_edit_accordion: outputs2.append(boogu_edit_accordion)
             if ref_img_accordion: outputs2.append(ref_img_accordion)
             if pid_accordion: outputs2.append(pid_accordion)
             if vae_accordion: outputs2.append(vae_accordion)
@@ -188,7 +192,7 @@ def attach_event_handlers(ui_components, demo):
                     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, joyai_ref_acc=joyai_ref_accordion, krea2_identity_edit_acc=krea2_identity_edit_accordion, krea2_reference_edit_acc=krea2_reference_edit_accordion, qwen_image_edit_acc=qwen_image_edit_accordion, ref_img_acc=ref_img_accordion
+                    pid_acc=pid_accordion, vae_acc=vae_accordion, joyai_ref_acc=joyai_ref_accordion, krea2_identity_edit_acc=krea2_identity_edit_accordion, krea2_reference_edit_acc=krea2_reference_edit_accordion, qwen_image_edit_acc=qwen_image_edit_accordion, boogu_edit_acc=boogu_edit_accordion, ref_img_acc=ref_img_accordion
                 )
                 model_comp.change(fn=change_fn, inputs=inputs2, outputs=outputs2)
 
@@ -209,6 +213,7 @@ def attach_event_handlers(ui_components, demo):
         create_krea2_identity_edit_event_handlers(prefix, ui_components)
         create_krea2_reference_edit_event_handlers(prefix, ui_components)
         create_qwen_image_edit_event_handlers(prefix, ui_components)
+        create_boogu_edit_event_handlers(prefix, ui_components)
         create_reference_image_event_handlers(prefix, ui_components)
         create_run_event(prefix, task_type, ui_components)
 

ui/events/run_handlers.py

@@ -56,6 +56,7 @@ def create_run_event(prefix: str, task_type: str, ui_components: dict):
     krea2_identity_edit_data_components = ui_components.get(f'all_krea2_identity_edit_components_flat_{prefix}', [])
     krea2_reference_edit_data_components = ui_components.get(f'all_krea2_reference_edit_components_flat_{prefix}', [])
     qwen_image_edit_data_components = ui_components.get(f'all_qwen_image_edit_components_flat_{prefix}', [])
+    boogu_edit_data_components = ui_components.get(f'all_boogu_edit_components_flat_{prefix}', [])
     reference_image_data_components = ui_components.get(f'all_reference_image_components_flat_{prefix}', [])
     
     run_inputs_map['vae_source'] = ui_components.get(f'vae_source_{prefix}')
@@ -67,7 +68,7 @@ def create_run_event(prefix: str, task_type: str, ui_components: dict):
     all_chains = [
         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, joyai_reference_data_components, krea2_identity_edit_data_components, krea2_reference_edit_data_components, qwen_image_edit_data_components, reference_image_data_components
+        embedding_data_components, conditioning_data_components, reference_latent_data_components, hidream_o1_reference_data_components, joyai_reference_data_components, krea2_identity_edit_data_components, krea2_reference_edit_data_components, qwen_image_edit_data_components, boogu_edit_data_components, reference_image_data_components
     ]
     for chain in all_chains:
         if chain:
@@ -101,6 +102,7 @@ def create_run_event(prefix: str, task_type: str, ui_components: dict):
         assign_chain_data('krea2_identity_edit_data', krea2_identity_edit_data_components)
         assign_chain_data('krea2_reference_edit_data', krea2_reference_edit_data_components)
         assign_chain_data('qwen_image_edit_data', qwen_image_edit_data_components)
+        assign_chain_data('boogu_edit_data', boogu_edit_data_components)
         assign_chain_data('reference_image_data', reference_image_data_components)
 
         return ui_dict

ui/shared/hires_fix_ui.py

@@ -7,7 +7,7 @@ from .ui_components import (
     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,
-    create_reference_latent_ui, create_hidream_o1_reference_ui, create_joyai_reference_ui, create_reference_image_ui, create_krea2_identity_edit_ui, create_krea2_reference_edit_ui, create_qwen_image_edit_ui
+    create_reference_latent_ui, create_hidream_o1_reference_ui, create_joyai_reference_ui, create_reference_image_ui, create_krea2_identity_edit_ui, create_krea2_reference_edit_ui, create_qwen_image_edit_ui, create_boogu_edit_ui
 )
 
 default_vals = MODEL_DEFAULTS_CONFIG.get('Default', {})
@@ -105,6 +105,7 @@ def create_ui():
         components.update(create_krea2_identity_edit_ui(prefix))
         components.update(create_krea2_reference_edit_ui(prefix))
         components.update(create_qwen_image_edit_ui(prefix))
+        components.update(create_boogu_edit_ui(prefix))
         components.update(create_reference_image_ui(prefix))
         components.update(create_vae_override_ui(prefix))
             

ui/shared/img2img_ui.py

@@ -7,7 +7,7 @@ from .ui_components import (
     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,
-    create_reference_latent_ui, create_hidream_o1_reference_ui, create_joyai_reference_ui, create_reference_image_ui, create_krea2_identity_edit_ui, create_krea2_reference_edit_ui, create_qwen_image_edit_ui
+    create_reference_latent_ui, create_hidream_o1_reference_ui, create_joyai_reference_ui, create_reference_image_ui, create_krea2_identity_edit_ui, create_krea2_reference_edit_ui, create_qwen_image_edit_ui, create_boogu_edit_ui
 )
 
 default_vals = MODEL_DEFAULTS_CONFIG.get('Default', {})
@@ -86,6 +86,7 @@ def create_ui():
         components.update(create_krea2_identity_edit_ui(prefix))
         components.update(create_krea2_reference_edit_ui(prefix))
         components.update(create_qwen_image_edit_ui(prefix))
+        components.update(create_boogu_edit_ui(prefix))
         components.update(create_reference_image_ui(prefix))
         components.update(create_vae_override_ui(prefix))
             

ui/shared/inpaint_ui.py

@@ -6,7 +6,7 @@ from .ui_components import (
     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,
-    create_reference_latent_ui, create_hidream_o1_reference_ui, create_joyai_reference_ui, create_reference_image_ui, create_krea2_identity_edit_ui, create_krea2_reference_edit_ui, create_qwen_image_edit_ui
+    create_reference_latent_ui, create_hidream_o1_reference_ui, create_joyai_reference_ui, create_reference_image_ui, create_krea2_identity_edit_ui, create_krea2_reference_edit_ui, create_qwen_image_edit_ui, create_boogu_edit_ui
 )
 
 default_vals = MODEL_DEFAULTS_CONFIG.get('Default', {})
@@ -118,6 +118,7 @@ def create_ui():
             components.update(create_krea2_identity_edit_ui(prefix))
             components.update(create_krea2_reference_edit_ui(prefix))
             components.update(create_qwen_image_edit_ui(prefix))
+            components.update(create_boogu_edit_ui(prefix))
             components.update(create_reference_image_ui(prefix))
             components.update(create_vae_override_ui(prefix))
         components[f'accordion_wrapper_{prefix}'] = accordion_wrapper

ui/shared/outpaint_ui.py

@@ -7,7 +7,7 @@ from .ui_components import (
     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,
-    create_reference_latent_ui, create_hidream_o1_reference_ui, create_joyai_reference_ui, create_reference_image_ui, create_krea2_identity_edit_ui, create_krea2_reference_edit_ui, create_qwen_image_edit_ui
+    create_reference_latent_ui, create_hidream_o1_reference_ui, create_joyai_reference_ui, create_reference_image_ui, create_krea2_identity_edit_ui, create_krea2_reference_edit_ui, create_qwen_image_edit_ui, create_boogu_edit_ui
 )
 
 default_vals = MODEL_DEFAULTS_CONFIG.get('Default', {})
@@ -101,6 +101,7 @@ def create_ui():
         components.update(create_krea2_identity_edit_ui(prefix))
         components.update(create_krea2_reference_edit_ui(prefix))
         components.update(create_qwen_image_edit_ui(prefix))
+        components.update(create_boogu_edit_ui(prefix))
         components.update(create_reference_image_ui(prefix))
         components.update(create_vae_override_ui(prefix))
             

ui/shared/txt2img_ui.py

@@ -6,7 +6,7 @@ from .ui_components import (
     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,
-    create_reference_latent_ui, create_hidream_o1_reference_ui, create_joyai_reference_ui, create_reference_image_ui, create_krea2_identity_edit_ui, create_krea2_reference_edit_ui, create_qwen_image_edit_ui,
+    create_reference_latent_ui, create_hidream_o1_reference_ui, create_joyai_reference_ui, create_reference_image_ui, create_krea2_identity_edit_ui, create_krea2_reference_edit_ui, create_qwen_image_edit_ui, create_boogu_edit_ui,
     create_pid_ui
 )
 
@@ -58,6 +58,7 @@ def create_ui():
         components.update(create_krea2_identity_edit_ui(prefix))
         components.update(create_krea2_reference_edit_ui(prefix))
         components.update(create_qwen_image_edit_ui(prefix))
+        components.update(create_boogu_edit_ui(prefix))
         components.update(create_reference_image_ui(prefix))
         components.update(create_vae_override_ui(prefix))
         components.update(create_pid_ui(prefix))

ui/shared/ui_components.py

@@ -856,4 +856,33 @@ def create_krea2_reference_edit_ui(prefix: str, max_units=3):
         
         components[key('all_krea2_reference_edit_components_flat')] = ref_image_inputs
 
+    return components
+
+def create_boogu_edit_ui(prefix: str, max_units=10):
+    components = {}
+    key = lambda name: f"{name}_{prefix}"
+
+    with gr.Accordion("Boogu-Image Edit Settings", open=False, visible=('boogu_edit' in default_enabled_chains)) as ref_accordion:
+        components[key('boogu_edit_accordion')] = ref_accordion
+        gr.Markdown("💡 **Tip:** (Boogu-Image-Edit-Turbo/Boogu-Image-Edit recommended, Boogu-Image-Edit need set ZeroGPU Duration (s) to 120 ) In txt2img mode, adding a single reference image performs an **Image Edit**, while adding multiple images performs an **Image Combine**.")
+        
+        ref_image_groups = []
+        ref_image_inputs = []
+        with gr.Row():
+            for i in range(max_units):
+                with gr.Column(visible=(i < 1), min_width=160) as img_col:
+                    img_comp = gr.Image(type="pil", label=f"Ref. {i+1}", sources=["upload"], height=150)
+                    ref_image_groups.append(img_col)
+                    ref_image_inputs.append(img_comp)
+        
+        components[key('boogu_edit_rows')] = ref_image_groups
+        components[key('boogu_edit_images')] = ref_image_inputs
+
+        with gr.Row():
+            components[key('add_boogu_edit_button')] = gr.Button("✚ Add Reference Image")
+            components[key('delete_boogu_edit_button')] = gr.Button("➖ Delete Reference Image", visible=False)
+        components[key('boogu_edit_count_state')] = gr.State(1)
+        
+        components[key('all_boogu_edit_components_flat')] = ref_image_inputs
+
     return components

yaml/file_list.yaml

@@ -448,6 +448,14 @@ file:
       source: "hf"
       repo_id: "Comfy-Org/Boogu-Image"
       repository_file_path: "diffusion_models/boogu_image_turbo_hotfix_nvfp4.safetensors"
+    - filename: "boogu_image_edit_turbo_hotfix_1k_20260708_int8_convrot.safetensors"
+      source: "hf"
+      repo_id: "Comfy-Org/Boogu-Image"
+      repository_file_path: "diffusion_models/boogu_image_edit_turbo_hotfix_1k_20260708_int8_convrot.safetensors"
+    - filename: "boogu_image_edit_nvfp4.safetensors"
+      source: "hf"
+      repo_id: "Comfy-Org/Boogu-Image"
+      repository_file_path: "diffusion_models/boogu_image_edit_nvfp4.safetensors"
     # Ideogram-4
     - filename: "ideogram4_nvfp4_mixed.safetensors"
       source: "hf"

yaml/image_gen_features.yaml

@@ -27,7 +27,7 @@ krea-2:
 boogu-image:
   enabled_chains:
   - lora
-  - conditioning
+  - boogu_edit
   - pid
 
 pixeldit:

yaml/injectors.yaml

@@ -37,6 +37,8 @@ injector_definitions:
     module: "chain_injectors.krea2_reference_edit_injector"
   dynamic_qwen_image_edit_chains:
     module: "chain_injectors.qwen_image_edit_injector"
+  dynamic_boogu_edit_chains:
+    module: "chain_injectors.boogu_edit_injector"
   dynamic_reference_image_chains:
     module: "chain_injectors.reference_image_injector"
   dynamic_pid_chains:
@@ -58,6 +60,7 @@ injector_order:
   - dynamic_krea2_identity_edit_chains
   - dynamic_krea2_reference_edit_chains
   - dynamic_qwen_image_edit_chains
+  - dynamic_boogu_edit_chains
   - dynamic_anima_controlnet_lllite_chains
   - dynamic_hidream_o1_smoothing_chains
   - dynamic_hidream_o1_reference_chains

yaml/model_defaults.yaml

@@ -52,6 +52,11 @@ Boogu-Image:
     cfg: 1.0
     sampler_name: "lcm"
     scheduler: "sgm_uniform"
+  "Boogu-Image-Edit-Turbo":
+    steps: 4
+    cfg: 1.0
+    sampler_name: "lcm"
+    scheduler: "sgm_uniform"
 
 PixelDiT:
   _defaults:

yaml/model_list.yaml

@@ -71,6 +71,16 @@ Checkpoint:
           unet: "boogu_image_base_nvfp4.safetensors"
           clip: "qwen3vl_8b_nvfp4.safetensors"
           vae: "ae.safetensors"
+      - display_name: "Boogu-Image-Edit-Turbo"
+        components:
+          unet: "boogu_image_edit_turbo_hotfix_1k_20260708_int8_convrot.safetensors"
+          clip: "qwen3vl_8b_nvfp4.safetensors"
+          vae: "ae.safetensors"
+      - display_name: "Boogu-Image-Edit"
+        components:
+          unet: "boogu_image_edit_nvfp4.safetensors"
+          clip: "qwen3vl_8b_nvfp4.safetensors"
+          vae: "ae.safetensors"
   PixelDiT:
     latent_type: chroma_radiance_latent
     models: