Merge pull request #46 from Bria-AI/WAI-4807

WAI-4807

Readme.md

@@ -8,10 +8,7 @@ This repository provides custom nodes for ComfyUI, enabling direct access to **B
 
 BRIA's APIs and models are built for commercial use and trained on 100% licensed data and does not contain copyrighted materials, such as fictional characters, logos, trademarks, public figures, harmful content, or privacy-infringing content.
 
-An API token is required to use the nodes in your workflows. Get started quickly here
-<a href="https://bria.ai/api/" style="text-decoration:none; vertical-align:middle;">
-  <img src="https://img.shields.io/badge/GET%20YOUR%20TOKEN-1000%20Free%20Calls-blue?style=flat-square" alt="Get Your Token" height="20">
-</a>.
+An API token is required to use the nodes in your workflows. Get yours at the [BRIA Platform](https://platform.bria.ai/organization-management/api-keys).
 
 for direct API endpoint use, you can find our APIs through partners like [**fal.ai**](https://fal.ai/models?keywords=bria).  
 For source code and weigths access, go to our [**Hugging Face**](https://huggingface.co/briaai) space.
@@ -107,13 +104,30 @@ These nodes create high-quality product images for eCommerce workflows.
 | **ShotByText**         | Modifies an image's background by providing a text prompt. Powered by BRIA's ControlNet Background-Generation. |
 | **ShotByImage**        | Modifies an image's background by providing a reference image. Uses BRIA's ControlNet Background-Generation and Image-Prompt. |
 
+## Video Editing Nodes
+
+These nodes perform high-quality edits for a given video.
+
+| Node | Description |
+|------|-------------|
+| **Bria Video Remove Background** | Remove the background from a video. |
+| **Bria Video Green Screen** | Replace the background of a video with a Chroma-green color. |
+| **Bria Video Replace Background** | Replaces the background of a video with a user-provided image or video |
+| **Bria SolidColor Background Video** | Replace the background of a video with a solid color. |
+| **Bria Video Increase Resolution** | Upscales video resolution |
+| **Bria Video Erase Elements** | Erases selected elements from the video using a mask |
+| **Bria Video Mask By Prompt** | Generates a mask video using a text prompt describing what to mask. |
+| **Bria Video Mask By Key Points** | Generates a mask video using key-points guidance |
+
+  Check out the example workflow in the workflows/ folder to see how the nodes should be wired together for loading and previewing a video end-to-end.
+
 ## Attribution Node
 
 | Node                          | Description |
 |-------------------------------|-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
 | **Attribution By Image Node** | This node shares generated images via API for Bria to pay attribution to the data owners who contributed to the generation. Once the images are shared with Bria, Bria calculates the attribution, completes the payment on behalf of the user, and erases the images immediately. This node should be included in any workflow using nodes of Bria’s Models (not necessary for Bria’s API nodes). You can also refer to the [**API documentation**]( https://docs.bria.ai/bria-attribution-service/other/postattributionbyimage) |
 
-
+An example workflow in the [workflows](workflows) folder is **`Video_Editig_Workflow.json`**, which wires several of these nodes together. Video API details are covered in the [**BRIA API documentation**](https://docs.bria.ai/).
 
 
 # Installation

__init__.py

@@ -2,6 +2,7 @@ from .nodes import (
     EraserNode,
     GenFillNode,
     ImageExpansionNode,
+    ImageEnhanceNode,
     ReplaceBgNode,
     RmbgNode,
     RemoveForegroundNode,
@@ -15,7 +16,11 @@ from .nodes import (
     TailoredPortraitNode,
     ReimagineNode,
     GenerateImageNodeV2,
+    GenerateImageLiteNodeV2,
     RefineImageNodeV2,
+    RefineImageLiteNodeV2,
+    GenerateStructuredPromptNodeV2,
+    GenerateStructuredPromptLiteNodeV2,
     ShotByTextAutomaticNode,
     ShotByImageManualPaddingNode,
     ShotByImageAutomaticAspectRatioNode,
@@ -26,7 +31,21 @@ from .nodes import (
     ShotByTextManualPlacementNode,
     ShotByTextManualPaddingNode,
     ShotByTextCustomCoordinatesNode,
-    AttributionByImageNode
+    AttributionByImageNode,
+    RemoveVideoBackgroundNode,
+    GreenScreenVideoNode,
+    ReplaceVideoBackgroundNode,
+    VideoSolidColorBackgroundNode,
+    VideoMaskByPromptNode,
+    VideoMaskByKeyPointsNode,
+    VideoIncreaseResolutionNode,
+    VideoEraseElementsNode,
+    LoadVideoFramesNode,
+    PreviewVideoURLNode,
+    FIBOEditNode,
+    FIBOEditStructuredInstructionNode,
+    BriaMultiImageSelect,
+    ProductIntegrateNode
 )
 
 # Map the node class to a name used internally by ComfyUI
@@ -34,6 +53,7 @@ NODE_CLASS_MAPPINGS = {
     "BriaEraser": EraserNode,  # Return the class, not an instance
     "BriaGenFill": GenFillNode,
     "ImageExpansionNode": ImageExpansionNode,
+    "ImageEnhanceNode": ImageEnhanceNode,
     "ReplaceBgNode": ReplaceBgNode,
     "RmbgNode": RmbgNode,
     "RemoveForegroundNode": RemoveForegroundNode,
@@ -58,13 +78,33 @@ NODE_CLASS_MAPPINGS = {
     "ReimagineNode": ReimagineNode,
     "AttributionByImageNode": AttributionByImageNode,
     "GenerateImageNodeV2": GenerateImageNodeV2,
+    "GenerateImageLiteNodeV2": GenerateImageLiteNodeV2,
     "RefineImageNodeV2": RefineImageNodeV2,
+    "RefineImageLiteNodeV2": RefineImageLiteNodeV2,
+    "GenerateStructuredPromptNodeV2": GenerateStructuredPromptNodeV2,
+    "GenerateStructuredPromptLiteNodeV2": GenerateStructuredPromptLiteNodeV2,
+    "RemoveVideoBackgroundNode":RemoveVideoBackgroundNode,
+    "GreenScreenVideoNode": GreenScreenVideoNode,
+    "ReplaceVideoBackgroundNode": ReplaceVideoBackgroundNode,
+    "VideoSolidColorBackgroundNode":VideoSolidColorBackgroundNode,
+    "VideoMaskByPromptNode":VideoMaskByPromptNode,
+    "VideoMaskByKeyPointsNode":VideoMaskByKeyPointsNode,
+    "VideoIncreaseResolutionNode":VideoIncreaseResolutionNode,
+    "VideoEraseElementsNode":VideoEraseElementsNode,
+    "LoadVideoFramesNode":LoadVideoFramesNode,
+    "PreviewVideoURLNode":PreviewVideoURLNode,
+    "FIBOEditNode": FIBOEditNode,
+    "FIBOEditStructuredInstructionNode": FIBOEditStructuredInstructionNode,
+    "BriaMultiImageSelect":BriaMultiImageSelect,
+    "ProductIntegrateNode": ProductIntegrateNode
+
 }
 # Map the node display name to the one shown in the ComfyUI node interface
 NODE_DISPLAY_NAME_MAPPINGS = {
     "BriaEraser": "Bria Eraser",
     "BriaGenFill": "Bria GenFill",
     "ImageExpansionNode": "Bria Image Expansion",
+    "ImageEnhanceNode": "Bria Image Enhance",
     "ReplaceBgNode": "Bria Replace Background",
     "RmbgNode": "Bria RMBG",
     "RemoveForegroundNode": "Bria Remove Foreground",
@@ -88,6 +128,27 @@ NODE_DISPLAY_NAME_MAPPINGS = {
     "Text2ImageHDNode": "Bria Text2Image HD",
     "ReimagineNode": "Bria Reimagine",
     "AttributionByImageNode": "Attribution By Image Node",
-    "GenerateImageNodeV2": "Generate Image",
-    "RefineImageNodeV2": "Refine and Regenerate Image",
+    "GenerateImageNodeV2": "FIBO - Generate Image",
+    "GenerateImageLiteNodeV2": "FIBO - Generate Image - Lite",
+    "RefineImageNodeV2": "FIBO - Refine and Regenerate Image",
+    "RefineImageLiteNodeV2": "FIBO - Refine Image - Lite",
+    "GenerateStructuredPromptNodeV2": "FIBO - Generate Structured Prompt",
+    "GenerateStructuredPromptLiteNodeV2": "FIBO - Generate Structured Prompt - Lite",
+    "RemoveVideoBackgroundNode": "Bria Video Remove Background",
+    "GreenScreenVideoNode": "Bria Video Green Screen",
+    "ReplaceVideoBackgroundNode": "Bria Video Replace Background",
+    "VideoSolidColorBackgroundNode":"Bria SolidColor Background Video",
+    "VideoMaskByPromptNode":"Bria Video Mask By Prompt",
+    "VideoMaskByKeyPointsNode":"Bria Video Mask By Key Points",
+    "VideoIncreaseResolutionNode":"Bria Video Increase Resolution",
+    "VideoEraseElementsNode":"Bria Video Erase Elements",
+    "LoadVideoFramesNode":"Bria Load Video",
+    "PreviewVideoURLNode":"Bria Preview Video",
+    "FIBOEditNode": "FIBO - Edit",
+    "FIBOEditStructuredInstructionNode": "FIBO - Edit - Structured Instruction",
+    "BriaMultiImageSelect":"Bria Multi Image Select",
+    "ProductIntegrateNode": "Bria Product Integrate"
+
 }
+
+WEB_DIRECTORY = "./web"

nodes/__init__.py

@@ -1,6 +1,7 @@
 from .eraser_node import EraserNode
 from .generative_fill_node import GenFillNode
 from .image_expansion_node import ImageExpansionNode
+from .image_enhance_node import ImageEnhanceNode
 from .replace_bg_node import ReplaceBgNode
 from .rmbg_node import RmbgNode
 from .remove_foreground_node import RemoveForegroundNode
@@ -12,7 +13,11 @@ from .text_2_image_fast_node import Text2ImageFastNode
 from .text_2_image_hd_node import Text2ImageHDNode
 from .reimagine_node import ReimagineNode
 from .generate_image_node_v2 import GenerateImageNodeV2
+from .generate_image_lite_node_v2 import GenerateImageLiteNodeV2
 from .refine_image_node_v2 import RefineImageNodeV2
+from .refine_image_lite_node_v2 import RefineImageLiteNodeV2
+from .generate_structured_prompt_node_v2 import GenerateStructuredPromptNodeV2
+from .generate_structured_prompt_lite_node_v2 import GenerateStructuredPromptLiteNodeV2
 from .shot_by_text_node import ShotByTextOriginalNode
 from .shot_by_text_automatic_aspect_ratio_node import ShotByTextAutomaticAspectRatioNode
 from .shot_by_text_automatic_node import ShotByTextAutomaticNode
@@ -28,3 +33,17 @@ from .shot_by_image_node import ShotByImageOriginalNode
 from .shot_by_image_manual_placement_node import ShotByImageManualPlacementNode
 from .shot_by_image_manual_padding_node import ShotByImageManualPaddingNode
 from .attribution_by_image_node import AttributionByImageNode
+from .video_nodes.remove_video_background_node import RemoveVideoBackgroundNode
+from .video_nodes.green_screen_video_node import GreenScreenVideoNode
+from .video_nodes.replace_video_background_node import ReplaceVideoBackgroundNode
+from .video_nodes.video_increase_resolution_node import VideoIncreaseResolutionNode
+from .video_nodes.video_solid_color_background_node import VideoSolidColorBackgroundNode
+from .video_nodes.video_erase_elements_node import VideoEraseElementsNode
+from .video_nodes.video_mask_by_prompt_node import VideoMaskByPromptNode
+from .video_nodes.video_mask_by_key_points_node import VideoMaskByKeyPointsNode
+from .video_nodes.load_video import LoadVideoFramesNode
+from .video_nodes.preview_video_node_from_url import PreviewVideoURLNode
+from .fibo_edit_node import FIBOEditNode
+from .fibo_edit_structured_instruction_node import FIBOEditStructuredInstructionNode
+from .multi_image_select import BriaMultiImageSelect
+from .product_integrate_node import ProductIntegrateNode

nodes/attribution_by_image_node.py

@@ -1,17 +1,22 @@
 import requests
-import torch
 
-from .common import deserialize_and_get_comfy_key, preprocess_image, image_to_base64, poll_status_until_completed
+from .common import (
+    bria_json_headers,
+    image_to_base64,
+    normalize_images_input,
+    poll_status_until_completed,
+    to_pil_safe,
+)
 
 class AttributionByImageNode():
     @classmethod
     def INPUT_TYPES(self):
         return {
             "required": {
-                "image": ("IMAGE",),
-                "model_version": (["2.3", "3.0","3.2"], {"default": "2.3"}),
+                "images": ("IMAGE",),
+                "model_version": (["2.3", "3.0", "3.2"], {"default": "2.3"}),
                 "api_key": ("STRING", {"default": "BRIA_API_TOKEN"}),
-            },              
+            },
         }
 
     RETURN_TYPES = ("STRING",)
@@ -22,46 +27,43 @@ class AttributionByImageNode():
     def __init__(self):
         self.api_url = "https://engine.prod.bria-api.com/v2/image/attribution/by_image"
 
-    # Define the execute method as expected by ComfyUI
-    def execute(self, image, model_version, api_key):
+    def execute(self, images, model_version, api_key):
         if api_key.strip() == "" or api_key.strip() == "BRIA_API_TOKEN":
             raise Exception("Please insert a valid API key.")
-        api_key = deserialize_and_get_comfy_key(api_key)
+        images = normalize_images_input(images)
 
-        # Check if image is tensor, if so, convert to NumPy array
-        if isinstance(image, torch.Tensor):
-            image = preprocess_image(image)
+        batch_results = []
 
-        # Convert image to base64 for the new API format
-        image_base64 = image_to_base64(image)
-        payload = {
-            "image": image_base64,
-            "model_version": model_version,
-        }
+        for idx, pil_image in enumerate(images):
+            try:
+                image_base64 = image_to_base64(pil_image)
 
-        headers = {
-            "Content-Type": "application/json",
-            "api_token": f"{api_key}"
-        }
+                payload = {
+                    "image": image_base64,
+                    "model_version": model_version,
+                }
 
-        try:
-            response = requests.post(self.api_url, json=payload, headers=headers)
-            
-            if response.status_code == 200 or response.status_code == 202:
-                print('Initial Attribution via Images API request successful, polling for completion...')
-                response_dict = response.json()
+                headers = bria_json_headers(api_key)
 
-                status_url = response_dict.get('status_url')
-                request_id = response_dict.get('request_id')
+                response = requests.post(self.api_url, json=payload, headers=headers)
                 
+                if response.status_code not in (200, 202):
+                    raise Exception(f"API request failed with status {response.status_code}: {response.text}")
+
+                # Poll until completion
+                response_dict = response.json()
+                status_url = response_dict.get('status_url')
                 if not status_url:
                     raise Exception("No status_url returned from API")
-                
-                print(f"Request ID: {request_id}, Status URL: {status_url}")
-                
+
                 final_response = poll_status_until_completed(status_url, api_key)
-                return (str(final_response.get("result",{}).get("content")),)
-            else:
-                raise Exception(f"Error: API request failed with status code {response.status_code} {response.text}")
-        except Exception as e:
-            raise Exception(f"{e}")
+                content = str(final_response.get("result", {}).get("content", ""))
+                batch_results.append(content)
+
+            except Exception as e:
+                print(f"[AttributionByImageNode] Skipping image {idx} due to error: {e}")
+                batch_results.append("")
+
+        # Join all responses with a delimiter
+        combined_response = "\n---\n".join(batch_results)
+        return (combined_response,)

nodes/common.py

@@ -6,16 +6,18 @@ import base64
 from torchvision.transforms import ToPILImage
 import requests
 import time
-import json
+import os
+import uuid
 
+BRIA_COMFYUI_USER_AGENT = "bria/ComfyUI"
 
-COMFY_KEY_ERROR = (
-    "Invalid Token Type\n\n"
-    "The API token you’ve entered is not a ComfyUI token.\n"
-    "Please use the valid token from your BRIA Account API Keys page:\n"
-    "https://platform.bria.ai/console/account/api-keys"
-)
-
+def bria_json_headers(api_token: str) -> dict:
+    """Headers for JSON POST requests to Bria API."""
+    return {
+        "Content-Type": "application/json",
+        "api_token": api_token,
+        "User-Agent": BRIA_COMFYUI_USER_AGENT,
+    }
 def postprocess_image(image):
     result_image = Image.open(io.BytesIO(image))
     result_image = result_image.convert("RGB")
@@ -41,6 +43,35 @@ def preprocess_image(image):
             print("Unexpected image dimensions. Expected 4D tensor.")
     return image
 
+def to_pil_safe(image):
+    """
+    Converts a single image tensor or numpy array (H,W,C) to PIL Image.
+    Handles float32 in 0-1 and uint8.
+    """
+    if isinstance(image, torch.Tensor):
+        image = image.detach().cpu().numpy()
+
+    # If image is empty, replace with 1x1 black
+    if image.size == 0:
+        image = np.zeros((1,1,3), dtype=np.uint8)
+
+    # Ensure float images are scaled 0-255
+    if image.dtype in [np.float32, np.float64]:
+        if image.max() <= 1.0:
+            image = (image * 255).astype(np.uint8)
+        else:
+            image = image.astype(np.uint8)
+
+    # Handle grayscale images
+    if image.ndim == 2:
+        return Image.fromarray(image, mode="L")
+    elif image.shape[2] == 3:
+        return Image.fromarray(image, mode="RGB")
+    elif image.shape[2] == 4:
+        return Image.fromarray(image, mode="RGBA")
+    else:
+        raise ValueError(f"Cannot convert image with shape {image.shape} to PIL")
+
 
 def preprocess_mask(mask):
     if isinstance(mask, torch.Tensor):
@@ -57,7 +88,6 @@ def preprocess_mask(mask):
 def process_request(api_url, image, mask, api_key, visual_input_content_moderation, visual_output_content_moderation):
     if api_key.strip() == "" or api_key.strip() == "BRIA_API_TOKEN":
         raise Exception("Please insert a valid API key.")
-    api_key = deserialize_and_get_comfy_key(api_key)
 
     # Check if image and mask are tensors, if so, convert to NumPy arrays
     if isinstance(image, torch.Tensor):
@@ -77,10 +107,7 @@ def process_request(api_url, image, mask, api_key, visual_input_content_moderati
         "visual_output_content_moderation":visual_output_content_moderation
     }
 
-    headers = {
-        "Content-Type": "application/json",
-        "api_token": f"{api_key}"
-    }
+    headers = bria_json_headers(api_key)
 
     try:
         response = requests.post(api_url, json=payload, headers=headers) 
@@ -132,7 +159,7 @@ def poll_status_until_completed(status_url, api_key, timeout=360, check_interval
         Exception: If timeout is reached or API request fails
     """
     start_time = time.time()
-    headers = {"api_token": api_key}
+    headers = bria_json_headers(api_key)
     
     while time.time() - start_time < timeout:
         try:
@@ -156,20 +183,108 @@ def poll_status_until_completed(status_url, api_key, timeout=360, check_interval
     
     raise Exception(f"Timeout reached after {timeout} seconds")
 
-def deserialize_and_get_comfy_key(encoded: str) -> str:
+
+def normalize_images_input(images):
     """
-    Decodes a base64-encoded JSON token and returns the ComfyUI API key.
+    Converts various image inputs into a list of PIL images:
+    - PIL.Image → [PIL.Image]
+    - list of PIL.Image → unchanged
+    - torch.Tensor (H,W,C) → [PIL.Image]
+    - torch.Tensor (B,H,W,C) → list of PIL.Images
     """
-    try:
-        decoded = base64.b64decode(encoded).decode("utf-8")
-        payload = json.loads(decoded)
 
-        if payload.get("type") != "comfy":
-            raise Exception(COMFY_KEY_ERROR)
 
-        return payload.get("apiKey")
-
-    except Exception as e:
-        raise Exception(COMFY_KEY_ERROR)
+    if isinstance(images, Image.Image):
+        return [images]
+    elif isinstance(images, list):
+        return [to_pil_safe(img) if isinstance(img, torch.Tensor) else img for img in images]
+    elif isinstance(images, torch.Tensor):
+        if images.ndim == 3:  # (H,W,C)
+            return [to_pil_safe(images)]
+        elif images.ndim == 4:  # (B,H,W,C)
+            return [to_pil_safe(img) for img in images]
+        else:
+            raise ValueError(f"Unsupported tensor shape: {images.shape}")
+    else:
+        raise ValueError(f"Unsupported input type: {type(images)}")
 
    
+
+_EXT_TO_PIL_AND_MIME = {
+    ".png": ("PNG", "image/png"),
+    ".jpg": ("JPEG", "image/jpeg"),
+    ".jpeg": ("JPEG", "image/jpeg"),
+    ".webp": ("WEBP", "image/webp"),
+    ".gif": ("GIF", "image/gif"),
+    ".bmp": ("BMP", "image/bmp"),
+    ".tif": ("TIFF", "image/tiff"),
+    ".tiff": ("TIFF", "image/tiff"),
+}
+
+
+def _pil_format_and_mime_for_filename(file_name):
+    """Return (pil_format, content_type, file_name_with_ext). Uses .png only when there is no extension."""
+    base = file_name.strip() if file_name else ""
+    if not base:
+        base = f"{uuid.uuid4()}_background"
+    root, ext = os.path.splitext(base)
+    ext = ext.lower()
+    if not ext:
+        ext = ".png"
+        base = f"{root}{ext}"
+    elif ext not in _EXT_TO_PIL_AND_MIME:
+        ext = ".png"
+        base = f"{root}{ext}"
+    pil_format, mime = _EXT_TO_PIL_AND_MIME[ext]
+    return pil_format, mime, base
+
+
+def upload_pil_image_to_temp(pil_image, api_token, file_name=None):
+    """
+    Request an anonymous presigned PUT URL, upload the image bytes, return the public temp URL.
+
+    ``file_name`` keeps its extension for format and Content-Type; if it has no extension, ``.png``
+    is appended. Matches platform POST /upload-image/anonymous/presigned-url (same pattern as video).
+    """
+    api_url = "https://platform.prod.bria-api.com/upload-image/anonymous/presigned-url"
+    headers = {"Content-Type": "application/json"}
+    if api_token:
+        headers["api_token"] = api_token
+
+    pil_format, content_type, file_name = _pil_format_and_mime_for_filename(file_name or "")
+
+    payload = {
+        "file_name": file_name,
+        "content_type": content_type,
+    }
+
+    buf = io.BytesIO()
+    to_save = pil_image
+    if pil_format == "JPEG" and to_save.mode in ("RGBA", "P"):
+        to_save = to_save.convert("RGB")
+    save_kwargs = {}
+    if pil_format == "JPEG":
+        save_kwargs["quality"] = 95
+    to_save.save(buf, format=pil_format, **save_kwargs)
+    buf.seek(0)
+    image_bytes = buf.read()
+    response = requests.post(api_url, json=payload, headers=headers)
+    if response.status_code != 200:
+        raise Exception(f"Failed to get image presigned URL: {response.status_code} {response.text}")
+
+    response_data = response.json()
+    image_url = response_data.get("image_url")
+    upload_url = response_data.get("upload_url")
+
+    if not image_url or not upload_url:
+        raise Exception(f"Invalid response from image presigned URL API: {response_data}")
+
+    upload_response = requests.put(
+        upload_url,
+        data=image_bytes,
+        headers={"Content-Type": content_type},
+    )
+    if upload_response.status_code not in (200, 204):
+        raise Exception(f"Failed to upload image to S3: {upload_response.status_code}")
+
+    return image_url

nodes/fibo_edit_node.py

@@ -0,0 +1,162 @@
+import requests
+import torch
+
+from .common import (
+    bria_json_headers,
+    image_to_base64,
+    poll_status_until_completed,
+    preprocess_image,
+    preprocess_mask,
+    postprocess_image,
+)
+
+
+class FIBOEditNode:
+    """FIBO Edit Node - Edit images with instructions"""
+
+    api_url = "https://engine.prod.bria-api.com/v2/image/edit"
+
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "api_token": ("STRING", {"default": "BRIA_API_TOKEN"}),
+                "images": ("IMAGE",),
+            },
+            "optional": {
+                "instruction": ("STRING",),
+                "mask": ("MASK",),
+                "structured_instruction": ("STRING",),
+                "negative_prompt": ("STRING",),
+                "steps_num": (
+                    "INT",
+                    {
+                        "default": 30,
+                        "min": 1,
+                        "max": 100,
+                    },
+                ),
+                "guidance_scale": (
+                    "INT",
+                    {
+                        "default": 5,
+                        "min": 1,
+                        "max": 20,
+                    },
+                ),
+                "seed": ("INT", {"default": 123456}),
+            },
+        }
+
+    RETURN_TYPES = ("IMAGE", "STRING", "INT")
+    RETURN_NAMES = ("IMAGE", "structured_instruction", "seed")
+    CATEGORY = "API Nodes"
+    FUNCTION = "execute"
+
+    def _validate_token(self, api_token: str):
+        if api_token.strip() == "" or api_token.strip() == "BRIA_API_TOKEN":
+            raise Exception("Please insert a valid API token.")
+
+    def _build_payload(
+        self,
+        instruction,
+        images,
+        mask=None,
+        structured_instruction=None,
+        negative_prompt=None,
+        steps_num=50,
+        guidance_scale=5,
+        seed=123456,
+    ):
+        # Process images
+        if isinstance(images, torch.Tensor):
+            processed_images = preprocess_image(images)
+        else:
+            processed_images = images
+
+        payload = {
+            "images": [image_to_base64(processed_images)],
+            "steps_num": steps_num,
+            "guidance_scale": guidance_scale,
+            "seed": seed,
+        }
+
+        # Add optional mask
+        if mask is not None:
+            if isinstance(mask, torch.Tensor):
+                processed_mask = preprocess_mask(mask)
+            else:
+                processed_mask = mask
+            payload["mask"] = image_to_base64(processed_mask)
+
+        # Add optional structured_instruction
+        if structured_instruction:
+            payload["structured_instruction"] = structured_instruction
+         # Add optional structured_instruction
+        if instruction:
+            payload["instruction"] = instruction
+        # Add optional negative_prompt
+        if negative_prompt:
+            payload["negative_prompt"] = negative_prompt
+
+        return payload
+
+    def execute(
+        self,
+        api_token,
+        instruction,
+        images,
+        mask=None,
+        structured_instruction=None,
+        negative_prompt=None,
+        steps_num=50,
+        guidance_scale=5,
+        seed=123456,
+    ):
+        self._validate_token(api_token)
+        payload = self._build_payload(
+            instruction,
+            images,
+            mask,
+            structured_instruction,
+            negative_prompt,
+            steps_num,
+            guidance_scale,
+            seed,
+        )
+        headers = bria_json_headers(api_token)
+
+        try:
+            response = requests.post(self.api_url, json=payload, headers=headers)
+
+            if response.status_code in (200, 202):
+                print(
+                    f"Initial request successful to {self.api_url}, polling for completion..."
+                )
+                response_dict = response.json()
+                status_url = response_dict.get("status_url")
+                request_id = response_dict.get("request_id")
+
+                if not status_url:
+                    raise Exception("No status_url returned from API")
+
+                print(f"Request ID: {request_id}, Status URL: {status_url}")
+
+                final_response = poll_status_until_completed(status_url, api_token)
+
+                result = final_response.get("result", {})
+                result_image_url = result.get("image_url")
+                structured_prompt = result.get("structured_prompt", "")
+                used_seed = result.get("seed")
+
+                image_response = requests.get(result_image_url)
+                result_image = postprocess_image(image_response.content)
+
+                return (result_image, structured_prompt, used_seed)
+
+            raise Exception(
+                f"Error: API request failed with status code {response.status_code} {response.text}"
+            )
+
+        except Exception as e:
+            raise Exception(f"{e}")

nodes/fibo_edit_structured_instruction_node.py

@@ -0,0 +1,74 @@
+import requests
+from .common import (
+    bria_json_headers,
+    image_to_base64,
+    normalize_images_input,
+    poll_status_until_completed,
+)
+
+
+class FIBOEditStructuredInstructionNode:
+    """FIBO Edit Structured Instruction Node - Generate structured instructions for image editing"""
+
+    api_url = "https://engine.prod.bria-api.com/v2/structured_instruction/generate"
+
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "api_token": ("STRING", {"default": "BRIA_API_TOKEN"}),
+                "images": ("IMAGE",),
+                "instruction": ("STRING",),
+            },
+        }
+
+    RETURN_TYPES = ("STRING",)
+    RETURN_NAMES = ("structured_instruction",)
+    CATEGORY = "API Nodes"
+    FUNCTION = "execute"
+
+    def _validate_token(self, api_token: str):
+        if api_token.strip() == "" or api_token.strip() == "BRIA_API_TOKEN":
+            raise Exception("Please insert a valid API token.")
+
+    def _build_payload(self, processed_image, instruction):
+        payload = {
+            "instruction": instruction,
+            "images": [image_to_base64(processed_image)],
+        }
+        return payload
+
+    def execute(self, api_token, images, instruction):
+        self._validate_token(api_token)
+        # Normalize input to list of PIL images
+        images = normalize_images_input(images)
+
+        batch_results = []
+
+        for idx, pil_image in enumerate(images):
+            try:
+                payload = self._build_payload(pil_image, instruction)
+                headers = bria_json_headers(api_token)
+
+                response = requests.post(self.api_url, json=payload, headers=headers)
+
+                if response.status_code not in (200, 202):
+                    raise Exception(f"API request failed with status {response.status_code}: {response.text}")
+
+                print(f"Initial request successful for image {idx}, polling for completion...")
+                response_dict = response.json()
+                status_url = response_dict.get("status_url")
+                if not status_url:
+                    raise Exception("No status_url returned from API")
+
+                final_response = poll_status_until_completed(status_url, api_token)
+                result = final_response.get("result", {})
+                structured_instruction = result.get("structured_instruction", "")
+                batch_results.append(structured_instruction)
+
+            except Exception as e:
+                print(f"[FIBOEditStructuredInstructionNode] Skipping image {idx} due to error: {e}")
+                batch_results.append("")
+
+        combined_instructions = "\n---\n".join(batch_results)
+        return (combined_instructions,)

nodes/generate_image_lite_node_v2.py

@@ -0,0 +1,157 @@
+import requests
+import torch
+from .common import (
+    bria_json_headers,
+    image_to_base64,
+    normalize_images_input,
+    poll_status_until_completed,
+    postprocess_image,
+)
+class GenerateImageLiteNodeV2:
+    """Lite Image Generation Node (multi-image compatible)"""
+
+    api_url = "https://engine.prod.bria-api.com/v2/image/generate/lite"
+
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "api_token": ("STRING", {"default": "BRIA_API_TOKEN"}),
+                "prompt": ("STRING",),
+            },
+            "optional": {
+                "model_version": (["FIBO"], {"default": "FIBO"}),
+                "structured_prompt": ("STRING", {"default": ""}),
+                "images": ("IMAGE",),
+                "aspect_ratio": (
+                    ["1:1", "2:3", "3:2", "3:4", "4:3", "4:5", "5:4", "9:16", "16:9"],
+                    {"default": "1:1"},
+                ),
+                "steps_num": ("INT", {"default": 8, "min": 8, "max": 30}),
+                "guidance_scale": ("INT", {"default": 5, "min": 3, "max": 5}),
+                "seed": ("STRING", {"default": "123456"}),
+            },
+        }
+
+    RETURN_TYPES = ("IMAGE", "STRING", "STRING")
+    RETURN_NAMES = ("image", "structured_prompt", "seed")
+    CATEGORY = "API Nodes"
+    FUNCTION = "execute"
+
+    def _validate_token(self, api_token: str):
+        if api_token.strip() == "" or api_token.strip() == "BRIA_API_TOKEN":
+            raise Exception("Please insert a valid API token.")
+
+    def _build_payload(
+        self,
+        prompt,
+        model_version,
+        structured_prompt,
+        aspect_ratio,
+        steps_num,
+        guidance_scale,
+        seed,
+        processed_image=None,
+    ):
+        payload = {
+            "prompt": prompt,
+            "model_version": model_version,
+            "aspect_ratio": aspect_ratio,
+            "steps_num": steps_num,
+            "guidance_scale": guidance_scale,
+            "seed": int(seed),
+        }
+        if structured_prompt:
+            payload["structured_prompt"] = structured_prompt
+        if processed_image is not None:
+            payload["images"] = [image_to_base64(processed_image)]
+        return payload
+
+    def execute(
+        self,
+        api_token,
+        prompt,
+        model_version,
+        structured_prompt,
+        aspect_ratio,
+        steps_num,
+        guidance_scale,
+        seed,
+        images=None,
+    ):
+        self._validate_token(api_token)
+        images_list = normalize_images_input(images) if images is not None else [None]
+
+        # Structured prompts per image
+        if isinstance(structured_prompt, str):
+            structured_prompts_list = structured_prompt.split("\n---\n")
+        elif isinstance(structured_prompt, list):
+            structured_prompts_list = structured_prompt
+        else:
+            structured_prompts_list = [""] * len(images_list)
+        if len(structured_prompts_list) < len(images_list):
+            structured_prompts_list += [structured_prompts_list[-1]] * (len(images_list) - len(structured_prompts_list))
+
+        # Seeds per image
+        if isinstance(seed, str):
+            seed_values = [int(s.strip()) for s in seed.split(",")]
+        else:
+            seed_values = [int(seed)]
+        if len(seed_values) < len(images_list):
+            seed_values += [seed_values[-1]] * (len(images_list) - len(seed_values))
+
+        batch_results = []
+        batch_structured_prompts = []
+        batch_seeds = []
+
+        for idx, ref_image in enumerate(images_list):
+            try:
+
+                payload = self._build_payload(
+                    prompt,
+                    model_version,
+                    structured_prompts_list[idx],
+                    aspect_ratio,
+                    steps_num,
+                    guidance_scale,
+                    seed_values[idx],
+                    ref_image,
+                )
+
+                headers = bria_json_headers(api_token)
+                response = requests.post(self.api_url, json=payload, headers=headers)
+                if response.status_code not in (200, 202):
+                    raise Exception(
+                        f"API request failed with status code {response.status_code}: {response.text}"
+                    )
+
+                print(f"GenerateImageLiteNodeV2 - Initial request successful for image {idx}, polling for completion...")
+                response_dict = response.json()
+                status_url = response_dict.get("status_url")
+                if not status_url:
+                    raise Exception("No status_url returned from API")
+
+                final_response = poll_status_until_completed(status_url, api_token)
+                result = final_response.get("result", {})
+                result_image_url = result.get("image_url")
+                structured_prompt_result = result.get("structured_prompt", "")
+                used_seed = result.get("seed", seed_values[idx])
+
+                image_response = requests.get(result_image_url)
+                result_image = postprocess_image(image_response.content)
+
+                batch_results.append(result_image)
+                batch_structured_prompts.append(structured_prompt_result)
+                batch_seeds.append(str(used_seed))
+
+            except Exception as e:
+                print(f"[GenerateImageLiteNodeV2] Skipping iteration {idx} due to error: {e}")
+                batch_results.append(torch.zeros((1, 512, 512, 3), dtype=torch.float32))
+                batch_structured_prompts.append("")
+                batch_seeds.append(str(seed_values[idx]))
+
+        output_batch = torch.cat(batch_results, dim=0)
+        combined_structured_prompts = "\n---\n".join(batch_structured_prompts)
+        combined_seeds = ",".join(batch_seeds)
+
+        return output_batch, combined_structured_prompts, combined_seeds

nodes/generate_image_node_v2.py

@@ -2,18 +2,18 @@ import requests
 import torch
 
 from .common import (
-    deserialize_and_get_comfy_key,
-    postprocess_image,
-    preprocess_image,
+    bria_json_headers,
     image_to_base64,
+    normalize_images_input,
     poll_status_until_completed,
+    postprocess_image,
 )
 
 
-class _BaseGenerateImageNodeV2:
-    """Base class for image generation nodes (standard & pro)."""
+class GenerateImageNodeV2:
+    """Standard Image Generation Node (multi-image compatible)"""
 
-    api_url = None  # Each subclass must define its API endpoint
+    api_url = "https://engine.prod.bria-api.com/v2/image/generate"
 
     @classmethod
     def INPUT_TYPES(cls):
@@ -24,19 +24,20 @@ class _BaseGenerateImageNodeV2:
             },
             "optional": {
                 "model_version": (["FIBO"], {"default": "FIBO"}),
-                "negative_prompt": ("STRING", {"default": ""}),
+                "structured_prompt": ("STRING", {"default": ""}),
+                "negative_prompt": ("STRING",),
                 "images": ("IMAGE",),
                 "aspect_ratio": (
                     ["1:1", "2:3", "3:2", "3:4", "4:3", "4:5", "5:4", "9:16", "16:9"],
                     {"default": "1:1"},
                 ),
-                "steps_num": ("INT", {"default": 50, "min": 20, "max": 50}),
+                "steps_num": ("INT", {"default": 50, "min": 35, "max": 50}),
                 "guidance_scale": ("INT", {"default": 5, "min": 3, "max": 5}),
-                "seed": ("INT", {"default": 123456}),
+                "seed": ("STRING", {"default": "123456"}),  # Accept string to match previous node
             },
         }
 
-    RETURN_TYPES = ("IMAGE", "STRING", "INT")
+    RETURN_TYPES = ("IMAGE", "STRING", "STRING")  # images, structured_prompts, seeds
     RETURN_NAMES = ("image", "structured_prompt", "seed")
     CATEGORY = "API Nodes"
     FUNCTION = "execute"
@@ -49,29 +50,28 @@ class _BaseGenerateImageNodeV2:
         self,
         prompt,
         model_version,
-        negative_prompt,
+        structured_prompt,
         aspect_ratio,
         steps_num,
         guidance_scale,
         seed,
-        images=None,
+        negative_prompt=None,
+        processed_image=None,
     ):
         payload = {
             "prompt": prompt,
             "model_version": model_version,
-            "negative_prompt": negative_prompt,
             "aspect_ratio": aspect_ratio,
             "steps_num": steps_num,
             "guidance_scale": guidance_scale,
-            "seed": seed,
+            "seed": int(seed),
         }
-
-        if images is not None:
-            if isinstance(images, torch.Tensor):
-                preprocess_images = preprocess_image(images)
-            payload["images"] = [image_to_base64(preprocess_images)]
-
-
+        if structured_prompt:
+            payload["structured_prompt"] = structured_prompt
+        if negative_prompt:
+            payload["negative_prompt"] = negative_prompt
+        if processed_image is not None:
+            payload["images"] = [image_to_base64(processed_image)]
         return payload
 
     def execute(
@@ -79,65 +79,88 @@ class _BaseGenerateImageNodeV2:
         api_token,
         prompt,
         model_version,
-        negative_prompt,
+        structured_prompt,
         aspect_ratio,
         steps_num,
         guidance_scale,
         seed,
+        negative_prompt=None,
         images=None,
     ):
         self._validate_token(api_token)
-        payload = self._build_payload(
-            prompt,
-            model_version,
-            negative_prompt,
-            aspect_ratio,
-            steps_num,
-            guidance_scale,
-            seed,
-            images,
-        )
-        api_token = deserialize_and_get_comfy_key(api_token)
+        images_list = normalize_images_input(images) if images is not None else [None]
 
-        headers = {"Content-Type": "application/json", "api_token": api_token}
+        # Structured prompts per image
+        if isinstance(structured_prompt, str):
+            structured_prompts_list = structured_prompt.split("\n---\n")
+        elif isinstance(structured_prompt, list):
+            structured_prompts_list = structured_prompt
+        else:
+            structured_prompts_list = [""] * len(images_list)
+        if len(structured_prompts_list) < len(images_list):
+            structured_prompts_list += [structured_prompts_list[-1]] * (len(images_list) - len(structured_prompts_list))
 
-        try:
-            response = requests.post(self.api_url, json=payload, headers=headers)
+        # Seeds per image
+        if isinstance(seed, str):
+            seed_values = [int(s.strip()) for s in seed.split(",")]
+        else:
+            seed_values = [int(seed)]
+        if len(seed_values) < len(images_list):
+            seed_values += [seed_values[-1]] * (len(images_list) - len(seed_values))
 
-            if response.status_code in (200, 202):
-                print(
-                    f"Initial request successful to {self.api_url}, polling for completion..."
+        batch_results = []
+        batch_structured_prompts = []
+        batch_seeds = []
+
+        for idx, ref_image in enumerate(images_list):
+            try:
+                payload = self._build_payload(
+                    prompt,
+                    model_version,
+                    structured_prompts_list[idx],
+                    aspect_ratio,
+                    steps_num,
+                    guidance_scale,
+                    seed_values[idx],
+                    negative_prompt,
+                    ref_image,
                 )
+
+                headers = bria_json_headers(api_token)
+                response = requests.post(self.api_url, json=payload, headers=headers)
+
+                if response.status_code not in (200, 202):
+                    raise Exception(
+                        f"API request failed with status code {response.status_code}: {response.text}"
+                    )
+                print(f"GenerateImageNodeV2 - Initial request successful for image {idx}, polling for completion...")
                 response_dict = response.json()
                 status_url = response_dict.get("status_url")
-                request_id = response_dict.get("request_id")
-
                 if not status_url:
                     raise Exception("No status_url returned from API")
 
-                print(f"Request ID: {request_id}, Status URL: {status_url}")
-
                 final_response = poll_status_until_completed(status_url, api_token)
-
                 result = final_response.get("result", {})
                 result_image_url = result.get("image_url")
-                structured_prompt = result.get("structured_prompt", "")
-                used_seed = result.get("seed")
+                structured_prompt_result = result.get("structured_prompt", "")
+                used_seed = result.get("seed", seed_values[idx])
 
                 image_response = requests.get(result_image_url)
                 result_image = postprocess_image(image_response.content)
 
-                return (result_image, structured_prompt, used_seed)
+                batch_results.append(result_image)
+                batch_structured_prompts.append(structured_prompt_result)
+                batch_seeds.append(str(used_seed))
 
-            raise Exception(
-                f"Error: API request failed with status code {response.status_code} {response.text}"
-            )
+            except Exception as e:
+                print(f"[GenerateImageNodeV2] Skipping iteration {idx} due to error: {e}")
+                batch_results.append(torch.zeros((1, 512, 512, 3), dtype=torch.float32))
+                batch_structured_prompts.append("")
+                batch_seeds.append(str(seed_values[idx]))
 
-        except Exception as e:
-            raise Exception(f"{e}")
+        # Return all as strings for proper chaining
+        output_batch = torch.cat(batch_results, dim=0)
+        combined_structured_prompts = "\n---\n".join(batch_structured_prompts)
+        combined_seeds = ",".join(batch_seeds)
 
-
-class GenerateImageNodeV2(_BaseGenerateImageNodeV2):
-    """Standard Image Generation Node"""
-    def __init__(self):
-        self.api_url = "https://engine.prod.bria-api.com/v2/image/generate"
+        return output_batch, combined_structured_prompts, combined_seeds

nodes/generate_structured_prompt_lite_node_v2.py

@@ -0,0 +1,110 @@
+import requests
+from .common import (
+    bria_json_headers,
+    image_to_base64,
+    normalize_images_input,
+    poll_status_until_completed,
+)
+
+
+class GenerateStructuredPromptLiteNodeV2:
+    """Lite Structured Prompt Generation Node (multi-image compatible)"""
+
+    api_url = "https://engine.prod.bria-api.com/v2/structured_prompt/generate/lite"
+
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "api_token": ("STRING", {"default": "BRIA_API_TOKEN"}),
+                "prompt": ("STRING",),
+            },
+            "optional": {
+                "structured_prompt": ("STRING",),
+                "images": ("IMAGE",),
+                "seed": ("STRING", {"default": "123456"}),
+            },
+        }
+
+    RETURN_TYPES = ("STRING", "STRING")  # structured_prompts, seeds as comma-separated string
+    RETURN_NAMES = ("structured_prompt", "seed")
+    CATEGORY = "API Nodes"
+    FUNCTION = "execute"
+
+    def _validate_token(self, api_token: str):
+        if api_token.strip() == "" or api_token.strip() == "BRIA_API_TOKEN":
+            raise Exception("Please insert a valid API token.")
+
+    def _build_payload(self, prompt, seed, structured_prompt, processed_image=None):
+        payload = {"prompt": prompt, "seed": int(seed)}
+        if structured_prompt:
+            payload["structured_prompt"] = structured_prompt
+        if processed_image is not None:
+            payload["images"] = [image_to_base64(processed_image)]
+        return payload
+
+    def execute(self, api_token, prompt, seed, structured_prompt, images=None):
+        self._validate_token(api_token)
+        images_list = normalize_images_input(images) if images is not None else [None]
+
+        # Seeds per image
+        if isinstance(seed, str):
+            seed_values = [int(s.strip()) for s in seed.split(",")]
+        else:
+            seed_values = [seed]
+        if len(seed_values) < len(images_list):
+            seed_values += [seed_values[-1]] * (len(images_list) - len(seed_values))
+
+        # Structured prompts per image
+        if isinstance(structured_prompt, str):
+            structured_prompts_list = structured_prompt.split("\n---\n")
+        elif isinstance(structured_prompt, list):
+            structured_prompts_list = structured_prompt
+        else:
+            structured_prompts_list = [""] * len(images_list)
+        if len(structured_prompts_list) < len(images_list):
+            structured_prompts_list += [structured_prompts_list[-1]] * (len(images_list) - len(structured_prompts_list))
+
+        batch_structured_prompts = []
+        batch_seeds = []
+
+        for idx, image in enumerate(images_list):
+            try:
+
+                payload = self._build_payload(
+                    prompt,
+                    seed_values[idx],
+                    structured_prompts_list[idx],
+                    image,
+                )
+
+                headers = bria_json_headers(api_token)
+                response = requests.post(self.api_url, json=payload, headers=headers)
+                if response.status_code not in (200, 202):
+                    raise Exception(
+                        f"API request failed with status code {response.status_code}: {response.text}"
+                    )
+
+                response_dict = response.json()
+                print(f"GenerateStructuredPromptLiteNodeV2 - Initial request successful for image {idx}, polling for completion...")
+
+                status_url = response_dict.get("status_url")
+                if not status_url:
+                    raise Exception("No status_url returned from API")
+
+                final_response = poll_status_until_completed(status_url, api_token)
+                result = final_response.get("result", {})
+                structured_prompt_result = result.get("structured_prompt", "")
+                used_seed = result.get("seed", seed_values[idx])
+
+                batch_structured_prompts.append(structured_prompt_result)
+                batch_seeds.append(str(used_seed))
+
+            except Exception as e:
+                print(f"[GenerateStructuredPromptLiteNodeV2] Skipping iteration {idx} due to error: {e}")
+                batch_structured_prompts.append("")
+                batch_seeds.append(str(seed_values[idx]))
+
+        combined_prompts = "\n---\n".join(batch_structured_prompts)
+        combined_seeds = ",".join(batch_seeds)
+        return combined_prompts, combined_seeds

nodes/generate_structured_prompt_node_v2.py

@@ -0,0 +1,113 @@
+import requests
+
+from .common import (
+    bria_json_headers,
+    image_to_base64,
+    normalize_images_input,
+    poll_status_until_completed,
+)
+
+
+class GenerateStructuredPromptNodeV2:
+    """Structured Prompt Generation Node (multi-image compatible)"""
+
+    api_url = "https://engine.prod.bria-api.com/v2/structured_prompt/generate"
+
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "api_token": ("STRING", {"default": "BRIA_API_TOKEN"}),
+                "prompt": ("STRING",),
+            },
+            "optional": {
+                "structured_prompt": ("STRING",),
+                "images": ("IMAGE",),
+                "seed": ("STRING", {"default": "123456"}),
+            },
+        }
+
+    RETURN_TYPES = ("STRING", "STRING")  # structured_prompts, seeds as comma-separated string
+    RETURN_NAMES = ("structured_prompt", "seed")
+    CATEGORY = "API Nodes"
+    FUNCTION = "execute"
+
+    def _validate_token(self, api_token: str):
+        if api_token.strip() == "" or api_token.strip() == "BRIA_API_TOKEN":
+            raise Exception("Please insert a valid API token.")
+
+    def _build_payload(self, prompt, seed, structured_prompt, processed_image=None):
+        payload = {"prompt": prompt, "seed": int(seed)}
+        if structured_prompt:
+            payload["structured_prompt"] = structured_prompt
+        if processed_image is not None:
+            payload["images"] = [image_to_base64(processed_image)]
+        return payload
+
+    def execute(self, api_token, prompt, seed, structured_prompt, images=None):
+        self._validate_token(api_token)
+        images_list = normalize_images_input(images) if images is not None else [None]
+
+        # Seeds per image
+        if isinstance(seed, str):
+            seed_values = [int(s.strip()) for s in seed.split(",")]
+        else:
+            seed_values = [seed]
+        if len(seed_values) < len(images_list):
+            seed_values += [seed_values[-1]] * (len(images_list) - len(seed_values))
+
+        # Structured prompts per image
+        if isinstance(structured_prompt, str):
+            structured_prompts_list = structured_prompt.split("\n---\n")
+        elif isinstance(structured_prompt, list):
+            structured_prompts_list = structured_prompt
+        else:
+            structured_prompts_list = [""] * len(images_list)
+        if len(structured_prompts_list) < len(images_list):
+            structured_prompts_list += [structured_prompts_list[-1]] * (len(images_list) - len(structured_prompts_list))
+
+        batch_structured_prompts = []
+        batch_seeds = []
+
+        for idx, image in enumerate(images_list):
+            try:
+
+                payload = self._build_payload(
+                    prompt,
+                    seed_values[idx],
+                    structured_prompts_list[idx],
+                    image
+                )
+
+                headers = bria_json_headers(api_token)
+                response = requests.post(self.api_url, json=payload, headers=headers)
+                if response.status_code not in (200, 202):
+                    raise Exception(
+                        f"API request failed with status code {response.status_code}: {response.text}"
+                    )
+
+                response_dict = response.json()
+                print(f"GenerateStructuredPromptNodeV2 - Initial request successful for image {idx}, polling for completion...")
+
+                status_url = response_dict.get("status_url")
+                if not status_url:
+                    raise Exception("No status_url returned from API")
+
+                final_response = poll_status_until_completed(status_url, api_token)
+
+                result = final_response.get("result", {})
+                structured_prompt_result = result.get("structured_prompt", "")
+                used_seed = result.get("seed", seed_values[idx])
+
+                batch_structured_prompts.append(structured_prompt_result)
+                batch_seeds.append(str(used_seed))  # Keep as string for passing between nodes
+
+            except Exception as e:
+                print(f"[GenerateStructuredPromptNodeV2] Skipping iteration {idx} due to error: {e}")
+                batch_structured_prompts.append("")
+                batch_seeds.append(str(seed_values[idx]))
+
+        # Return combined structured prompts and seeds as strings
+        combined_prompts = "\n---\n".join(batch_structured_prompts)
+        combined_seeds = ",".join(batch_seeds)
+        return combined_prompts, combined_seeds

nodes/generative_fill_node.py

@@ -4,7 +4,13 @@ from PIL import Image
 import io
 import torch
 
-from .common import deserialize_and_get_comfy_key, preprocess_image, preprocess_mask, image_to_base64, poll_status_until_completed
+from .common import (
+    bria_json_headers,
+    image_to_base64,
+    poll_status_until_completed,
+    preprocess_image,
+    preprocess_mask,
+)
 
 
 class GenFillNode():
@@ -39,8 +45,6 @@ class GenFillNode():
     def execute(self, image, mask, prompt, api_key, seed, prompt_content_moderation, visual_input_content_moderation, visual_output_content_moderation):
         if api_key.strip() == "" or api_key.strip() == "BRIA_API_TOKEN":
             raise Exception("Please insert a valid API key.")
-        api_key = deserialize_and_get_comfy_key(api_key)
-
         # Check if image and mask are tensors, if so, convert to NumPy arrays
         if isinstance(image, torch.Tensor):
             image = preprocess_image(image)
@@ -64,10 +68,7 @@ class GenFillNode():
             "version": 2
         }
 
-        headers = {
-            "Content-Type": "application/json",
-            "api_token": f"{api_key}"
-        }
+        headers = bria_json_headers(api_key)
 
         try:
             # Send initial request to get status URL

nodes/image_enhance_node.py

@@ -0,0 +1,110 @@
+import io
+import requests
+import numpy as np
+from PIL import Image
+import torch
+
+from .common import (
+    bria_json_headers,
+    image_to_base64,
+    normalize_images_input,
+    poll_status_until_completed,
+)
+
+class ImageEnhanceNode():
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "images": ("IMAGE",),
+                "api_key": ("STRING", {"default": "BRIA_API_TOKEN"}),
+            },
+            "optional": {
+                "steps_num": ("INT", {"default": 20, "min": 10, "max": 50}),
+                "resolution": (["1MP", "2MP", "4MP"], {"default": "1MP"}),
+                "visual_input_content_moderation": ("BOOLEAN", {"default": False}),
+                "visual_output_content_moderation": ("BOOLEAN", {"default": False}),
+                "seed": ("INT", {"default": 681794}),
+                "preserve_alpha": ("BOOLEAN", {"default": True}),
+            }
+        }
+
+    RETURN_TYPES = ("IMAGE", "STRING")
+    RETURN_NAMES = ("output_images", "seeds",)
+    CATEGORY = "API Nodes"
+    FUNCTION = "execute"
+
+    def __init__(self):
+        self.api_url = "https://engine.prod.bria-api.com/v2/image/edit/enhance"
+
+    def execute(
+        self,
+        images,
+        api_key,
+        visual_input_content_moderation,
+        visual_output_content_moderation,
+        seed,
+        steps_num,
+        resolution,
+        preserve_alpha
+    ):
+        # Validate API key
+        if api_key.strip() == "" or api_key.strip() == "BRIA_API_TOKEN":
+            raise Exception("Please insert a valid API key.")
+        # Normalize input to list of PIL images
+        images = normalize_images_input(images)
+
+        batch_results = []
+        batch_seeds = []
+
+        for idx, pil_image in enumerate(images):
+            try:
+                image_base64 = image_to_base64(pil_image)
+
+                payload = {
+                    "image": image_base64,
+                    "visual_input_content_moderation": visual_input_content_moderation,
+                    "visual_output_content_moderation": visual_output_content_moderation,
+                    "seed": seed,
+                    "steps_num": steps_num,
+                    "resolution": resolution,
+                    "preserve_alpha": preserve_alpha
+                }
+
+                headers = bria_json_headers(api_key)
+
+                # Send request
+                response = requests.post(self.api_url, json=payload, headers=headers)
+                if response.status_code not in (200, 202):
+                    raise Exception(f"API request failed with status {response.status_code}: {response.text}")
+
+                # Poll until completion
+                response_dict = response.json()
+                status_url = response_dict.get("status_url")
+                if not status_url:
+                    raise Exception("No status_url returned from API")
+
+                print(f"ImageEnhanceNode - Initial request successful for image {idx}, polling for completion...")
+                final_response = poll_status_until_completed(status_url, api_key)
+                result_image_url = final_response["result"]["image_url"]
+                used_seed = final_response["result"].get("seed", seed)
+
+                # Download and process image
+                image_response = requests.get(result_image_url)
+                result_image = Image.open(io.BytesIO(image_response.content)).convert("RGB")
+                result_array = np.array(result_image).astype(np.float32) / 255.0
+                result_tensor = torch.from_numpy(result_array)  # shape: (H,W,C)
+
+                batch_results.append(result_tensor)
+                batch_seeds.append(used_seed)
+
+            except Exception as e:
+                print(f"[ImageEnhanceNode] Skipping image {idx} due to error: {e}")
+                # Append fallback tensor with same size as input
+                fallback_array = np.array(pil_image).astype(np.float32) / 255.0
+                batch_results.append(torch.from_numpy(fallback_array))
+                batch_seeds.append(seed)
+
+        # Return list of tensors (not concatenated) + comma-separated seeds
+        combined_seeds = ",".join(map(str, batch_seeds))
+        return (batch_results, combined_seeds)

nodes/image_expansion_node.py

@@ -1,136 +1,133 @@
-import numpy as np
-import requests
-from PIL import Image
 import io
+import requests
+import numpy as np
+from PIL import Image
 import torch
 
-from .common import deserialize_and_get_comfy_key, image_to_base64, preprocess_image, poll_status_until_completed
-
-
+from .common import (
+    bria_json_headers,
+    image_to_base64,
+    normalize_images_input,
+    poll_status_until_completed,
+)
 class ImageExpansionNode():
     @classmethod
-    def INPUT_TYPES(self):
+    def INPUT_TYPES(cls):
         return {
             "required": {
-                "image": ("IMAGE",),  # Input image from another node
-                "api_key": ("STRING", {"default": "BRIA_API_TOKEN"}), # API Key input with a default value
+                "images": ("IMAGE",),
+                "api_key": ("STRING", {"default": "BRIA_API_TOKEN"}),
             },
-            
             "optional": {
                 "original_image_size": ("STRING",),
                 "original_image_location": ("STRING",),
                 "canvas_size": ("STRING", {"default": "1000, 1000"}),
-                "aspect_ratio": (["1:1", "2:3", "3:2", "3:4", "4:3", "4:5", "5:4", "9:16", "16:9","None"], {"default": "None"}),
+                "aspect_ratio": (["1:1", "2:3", "3:2", "3:4", "4:3", "4:5", "5:4", "9:16", "16:9", "None"], {"default": "None"}),
                 "prompt": ("STRING", {"default": ""}),
-                "seed": ("INT", {"default": 681794}),
+                "seed": ("STRING", {"default": "681794"}),  # <-- accepts seeds from Enhance
                 "negative_prompt": ("STRING", {"default": "Ugly, mutated"}),
                 "prompt_content_moderation": ("BOOLEAN", {"default": False}),
                 "preserve_alpha": ("BOOLEAN", {"default": True}),
                 "visual_input_content_moderation": ("BOOLEAN", {"default": False}),
                 "visual_output_content_moderation": ("BOOLEAN", {"default": False}),
-            } 
+            }
         }
 
     RETURN_TYPES = ("IMAGE",)
-    RETURN_NAMES = ("output_image",)
+    RETURN_NAMES = ("output_images",)
     CATEGORY = "API Nodes"
-    FUNCTION = "execute"  # This is the method that will be executed
+    FUNCTION = "execute"
 
     def __init__(self):
-        self.api_url = "https://engine.prod.bria-api.com/v2/image/edit/expand"  # Image Expansion API URL
+        self.api_url = "https://engine.prod.bria-api.com/v2/image/edit/expand"
 
-    # Define the execute method as expected by ComfyUI
-    def execute(self, image, 
-                original_image_size,
-                original_image_location,
-                canvas_size,
-                aspect_ratio,
-                prompt,
-                seed,
-                negative_prompt,
-                prompt_content_moderation,
-                preserve_alpha,
-                visual_input_content_moderation,
-                visual_output_content_moderation,
-                api_key):
-        if api_key.strip() == "" or api_key.strip() == "BRIA_API_TOKEN":
+    def execute(
+        self,
+        images,
+        original_image_size,
+        original_image_location,
+        canvas_size,
+        aspect_ratio,
+        prompt,
+        seed,
+        negative_prompt,
+        prompt_content_moderation,
+        preserve_alpha,
+        visual_input_content_moderation,
+        visual_output_content_moderation,
+        api_key
+    ):
+        if api_key.strip() in ("", "BRIA_API_TOKEN"):
             raise Exception("Please insert a valid API key.")
-        api_key = deserialize_and_get_comfy_key(api_key)
+        images = normalize_images_input(images)
+        canvas_size = [int(x.strip()) for x in canvas_size.split(",")] if canvas_size else ()
         original_image_size = [int(x.strip()) for x in original_image_size.split(",")] if original_image_size else ()
         original_image_location = [int(x.strip()) for x in original_image_location.split(",")] if original_image_location else ()
-        canvas_size = [int(x.strip()) for x in canvas_size.split(",")] if canvas_size else ()
-        
-        if negative_prompt == "":
-            negative_prompt = " " # hack to avoid error in triton which expects non-empty string
 
-        # Check if image and mask are tensors, if so, convert to NumPy arrays
-        if isinstance(image, torch.Tensor):
-            image = preprocess_image(image)
+        # Prepare per-image seeds
+        seed_values = [int(s.strip()) for s in seed.split(",")] if isinstance(seed, str) else [seed]
+        if len(seed_values) < len(images):
+            seed_values += [seed_values[-1]] * (len(images) - len(seed_values))
 
-        # Convert the image directly to Base64 string
-        image_base64 = image_to_base64(image)
-        if aspect_ratio and aspect_ratio != "None":
-            payload = {
-            "image": image_base64,
-            "aspect_ratio": aspect_ratio,
-            "prompt": prompt,
-            "negative_prompt": negative_prompt,
-            "seed": seed,
-            "prompt_content_moderation": prompt_content_moderation,
-            "preserve_alpha": preserve_alpha,
-            "visual_input_content_moderation": visual_input_content_moderation,
-            "visual_output_content_moderation": visual_output_content_moderation
-        }
-        else:
-            payload = {
-                "image": image_base64,
-                "original_image_size": original_image_size,
-                "original_image_location": original_image_location,
-                "canvas_size": canvas_size,
-                "prompt": prompt,
-                "negative_prompt": negative_prompt,
-                "seed": seed,
-                "prompt_content_moderation": prompt_content_moderation,
-                "preserve_alpha": preserve_alpha,
-                "visual_input_content_moderation": visual_input_content_moderation,
-                "visual_output_content_moderation": visual_output_content_moderation
-            }
+        if not negative_prompt:
+            negative_prompt = " "
 
-        headers = {
-            "Content-Type": "application/json",
-            "api_token": f"{api_key}"
-        }
+        batch_results = []
+
+        for idx, pil_image in enumerate(images):
+            try:
+                image_base64 = image_to_base64(pil_image)
+
+                if aspect_ratio and aspect_ratio != "None":
+                    payload = {
+                        "image": image_base64,
+                        "aspect_ratio": aspect_ratio,
+                        "prompt": prompt,
+                        "negative_prompt": negative_prompt,
+                        "seed": seed_values[idx],
+                        "prompt_content_moderation": prompt_content_moderation,
+                        "preserve_alpha": preserve_alpha,
+                        "visual_input_content_moderation": visual_input_content_moderation,
+                        "visual_output_content_moderation": visual_output_content_moderation
+                    }
+                else:
+                    payload = {
+                        "image": image_base64,
+                        "original_image_size": original_image_size,
+                        "original_image_location": original_image_location,
+                        "canvas_size": canvas_size,
+                        "prompt": prompt,
+                        "negative_prompt": negative_prompt,
+                        "seed": seed_values[idx],
+                        "prompt_content_moderation": prompt_content_moderation,
+                        "preserve_alpha": preserve_alpha,
+                        "visual_input_content_moderation": visual_input_content_moderation,
+                        "visual_output_content_moderation": visual_output_content_moderation
+                    }
+
+                headers = bria_json_headers(api_key)
+                response = requests.post(self.api_url, json=payload, headers=headers)
+                if response.status_code not in (200, 202):
+                    raise Exception(f"API request failed with status {response.status_code}: {response.text}")
 
-        try:
-            response = requests.post(self.api_url, json=payload, headers=headers)
-            
-            if response.status_code == 200 or response.status_code == 202:
-                print('Initial image expansion request successful, polling for completion...')
                 response_dict = response.json()
-                status_url = response_dict.get('status_url')
-                request_id = response_dict.get('request_id')
-                
+                status_url = response_dict.get("status_url")
                 if not status_url:
                     raise Exception("No status_url returned from API")
-                
-                print(f"Request ID: {request_id}, Status URL: {status_url}")
-                
-                # Poll status URL until completion
-                final_response = poll_status_until_completed(status_url, api_key)
-                
-                # Get the result image URL
-                result_image_url = final_response['result']['image_url']
-                
-                # Download and process the result image
-                image_response = requests.get(result_image_url)
-                result_image = Image.open(io.BytesIO(image_response.content))
-                result_image = result_image.convert("RGB")
-                result_image = np.array(result_image).astype(np.float32) / 255.0
-                result_image = torch.from_numpy(result_image)[None,]
-                
-                return (result_image,)
-            else:
-                raise Exception(f"Error: API request failed with status code {response.status_code}: {response.text}")
 
-        except Exception as e:
-            raise Exception(f"{e}")
+                print(f"ImageExpansionNode - Initial request successful for image {idx}, polling for completion...")
+                final_response = poll_status_until_completed(status_url, api_key)
+                result_image_url = final_response["result"]["image_url"]
+
+                image_response = requests.get(result_image_url)
+                result_image = Image.open(io.BytesIO(image_response.content)).convert("RGB")
+                result_tensor = torch.from_numpy(np.array(result_image).astype(np.float32) / 255.0)
+
+                batch_results.append(result_tensor)
+
+            except Exception as e:
+                print(f"[ImageExpansionNode] Skipping image {idx} due to error: {e}")
+                fallback_array = np.array(pil_image).astype(np.float32) / 255.0
+                batch_results.append(torch.from_numpy(fallback_array))
+
+        return (batch_results,)

nodes/multi_image_select.py

@@ -0,0 +1,85 @@
+import os
+import json
+from typing import List
+
+import torch
+import torch.nn.functional as F
+import numpy as np
+from PIL import Image, ImageOps
+
+try:
+    from folder_paths import get_input_directory
+except Exception:
+    get_input_directory = None
+
+
+IMG_EXTS = (".png", ".jpg", ".jpeg", ".webp", ".bmp", ".tif", ".tiff")
+
+
+def input_root() -> str:
+    return os.path.abspath(get_input_directory() if get_input_directory else "input")
+
+
+def parse_paths(value: str) -> List[str]:
+    if not value:
+        return []
+    try:
+        data = json.loads(value)
+        if isinstance(data, list):
+            return [str(x) for x in data]
+    except Exception:
+        pass
+    return []
+
+
+
+class BriaMultiImageSelect:
+    """
+    Select multiple images and return them as a list of PIL Images.
+    Images keep their original size.
+    """
+
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "selected_paths": (
+                    "STRING",
+                    {
+                        "multiline": True,
+                        "default": "",
+                        "placeholder": "Filled automatically by Select Images button",
+                    },
+                ),
+            }
+        }
+
+    RETURN_TYPES = ("IMAGE", "STRING")
+    RETURN_NAMES = ("images", "filenames")
+    FUNCTION = "load"
+    CATEGORY = "API Nodes"
+
+    def load(self, selected_paths: str):
+        paths = parse_paths(selected_paths)
+        if not paths:
+            raise RuntimeError("BriaMultiImageSelect: No images selected")
+
+        root = input_root()
+        pil_images: List[Image.Image] = []
+        names: List[str] = []
+
+        for rel in paths:
+            abs_path = os.path.join(root, rel)
+            if not abs_path.lower().endswith(IMG_EXTS):
+                continue
+            if not os.path.isfile(abs_path):
+                continue
+
+            pil_images.append(Image.open(abs_path))
+            names.append(os.path.splitext(os.path.basename(rel))[0])
+
+        if not pil_images:
+            raise RuntimeError("BriaMultiImageSelect: No valid images found")
+
+        filenames = ", ".join(names)
+        return (pil_images, filenames)

nodes/product_integrate_node.py

@@ -0,0 +1,134 @@
+import requests
+import torch
+
+from .common import (
+    bria_json_headers,
+    image_to_base64,
+    poll_status_until_completed,
+    postprocess_image,
+    preprocess_image,
+)
+
+
+class ProductIntegrateNode:
+    """Product Integrate Node - Integrate a single product into a background scene"""
+
+    api_url = "https://engine.prod.bria-api.com/v2/image/edit/product/integrate"
+
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "api_token": ("STRING", {"default": "BRIA_API_TOKEN"}),
+                "scene": ("IMAGE",),
+                "product_image": ("IMAGE",),
+                "x_coordinate": ("INT", {"default": 0, "min": 0, "max": 10000}),
+                "y_coordinate": ("INT", {"default": 0, "min": 0, "max": 10000}),
+                "width": ("INT", {"default": 512, "min": 1, "max": 10000}),
+                "height": ("INT", {"default": 512, "min": 1, "max": 10000}),
+            },
+            "optional": {
+                "seed": ("STRING", {"default": "123456"}),
+            }
+        }
+
+    RETURN_TYPES = ("IMAGE", "INT")
+    RETURN_NAMES = ("IMAGE", "seed")
+    CATEGORY = "API Nodes"
+    FUNCTION = "execute"
+
+    def _validate_token(self, api_token: str):
+        if api_token.strip() == "" or api_token.strip() == "BRIA_API_TOKEN":
+            raise Exception("Please insert a valid API token.")
+
+    def _build_payload(
+        self,
+        scene_image,
+        product_image,
+        x_coordinate,
+        y_coordinate,
+        width,
+        height,
+        seed,
+    ):
+        payload = {
+            "scene": image_to_base64(scene_image),
+            "products": [
+                {
+                    "image": image_to_base64(product_image),
+                    "coordinates": {
+                        "x": x_coordinate,
+                        "y": y_coordinate,
+                        "width": width,
+                        "height": height,
+                    }
+                }
+            ],
+            "seed": int(seed),
+        }
+
+        return payload
+
+    def execute(
+        self,
+        api_token,
+        scene,
+        product_image,
+        x_coordinate,
+        y_coordinate,
+        width,
+        height,
+        seed,
+    ):
+        self._validate_token(api_token)
+        # Process single scene image
+        if isinstance(scene, torch.Tensor):
+            processed_scene = preprocess_image(scene)
+        if isinstance(product_image, torch.Tensor):
+            processed_product = preprocess_image(product_image)
+
+        payload = self._build_payload(
+            processed_scene,
+            processed_product,
+            x_coordinate,
+            y_coordinate,
+            width,
+            height,
+            seed,
+        )
+
+        headers = bria_json_headers(api_token)
+
+        try:
+            response = requests.post(self.api_url, json=payload, headers=headers)
+
+            if response.status_code in (200, 202):
+                print(
+                    f"Initial product integrate request successful, polling for completion..."
+                )
+                response_dict = response.json()
+                status_url = response_dict.get("status_url")
+                request_id = response_dict.get("request_id")
+
+                if not status_url:
+                    raise Exception("No status_url returned from API")
+
+                print(f"Request ID: {request_id}, Status URL: {status_url}")
+
+                final_response = poll_status_until_completed(status_url, api_token)
+
+                result = final_response.get("result", {})
+                result_image_url = result.get("image_url")
+                used_seed = result.get("seed", seed)
+
+                image_response = requests.get(result_image_url)
+                result_image = postprocess_image(image_response.content)
+
+                return (result_image, used_seed)
+            else:
+                raise Exception(
+                    f"API request failed with status code {response.status_code} {response.text}"
+                )
+
+        except Exception as e:
+            raise Exception(f"[ProductIntegrateNode] Error: {e}")

nodes/refine_image_lite_node_v2.py

@@ -0,0 +1,167 @@
+import requests
+
+from .common import bria_json_headers, poll_status_until_completed, postprocess_image
+
+
+
+class RefineImageLiteNodeV2:
+    """Lite Refine Image Node"""
+
+    api_url = "https://engine.prod.bria-api.com/v2/structured_prompt/generate/lite"
+    generate_api_url = "https://engine.prod.bria-api.com/v2/image/generate/lite"
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "api_token": ("STRING", {"default": "BRIA_API_TOKEN"}),
+                "prompt": ("STRING",),
+                "structured_prompt": ("STRING",),
+            },
+            "optional": {
+                "model_version": (["FIBO"], {"default": "FIBO"}),
+                "aspect_ratio": (
+                    ["1:1", "2:3", "3:2", "3:4", "4:3", "4:5", "5:4", "9:16", "16:9"],
+                    {"default": "1:1"},
+                ),
+                "steps_num": (
+                    "INT",
+                    {
+                        "default": 8,
+                        "min": 8,
+                        "max": 30,
+                    },
+                ),
+                "guidance_scale": (
+                    "INT",
+                    {
+                        "default": 5,
+                        "min": 3,
+                        "max": 5,
+                    },
+                ),
+                "seed": ("INT", {"default": 123456}),
+            },
+        }
+
+
+    RETURN_TYPES = ("IMAGE", "STRING", "INT")
+    RETURN_NAMES = ("image", "structured_prompt", "seed")
+    CATEGORY = "API Nodes"
+    FUNCTION = "execute"
+
+    def _validate_token(self, api_token: str):
+        if api_token.strip() == "" or api_token.strip() == "BRIA_API_TOKEN":
+            raise Exception("Please insert a valid API token.")
+
+    def _build_payload(
+        self,
+        prompt,
+        structured_prompt,
+        model_version,
+        aspect_ratio,
+        steps_num,
+        guidance_scale,
+        seed,
+    ):
+        payload = {
+            "prompt": prompt,
+            "model_version": model_version,
+            "aspect_ratio": aspect_ratio,
+            "steps_num": steps_num,
+            "guidance_scale": guidance_scale,
+            "seed": seed,
+        }
+        if structured_prompt:
+            payload["structured_prompt"] = structured_prompt
+
+        return payload
+
+    def execute(
+        self,
+        api_token,
+        prompt,
+        structured_prompt,
+        model_version,
+        aspect_ratio,
+        steps_num,
+        guidance_scale,
+        seed
+    ):
+        self._validate_token(api_token)
+        payload = self._build_payload(
+            prompt,
+            structured_prompt,
+            model_version,
+            aspect_ratio,
+            steps_num,
+            guidance_scale,
+            seed,
+        )
+        headers = bria_json_headers(api_token)
+
+        try:
+            response = requests.post(self.api_url, json=payload, headers=headers)
+
+            if response.status_code in (200, 202):
+                print(f"Initial refine request successful to {self.api_url}, polling for completion...")
+                response_dict = response.json()
+                status_url = response_dict.get("status_url")
+                request_id = response_dict.get("request_id")
+
+                if not status_url:
+                    raise Exception("No status_url returned from API")
+
+                print(f"Request ID: {request_id}, Status URL: {status_url}")
+
+                final_response = poll_status_until_completed(status_url, api_token)
+
+                result = final_response.get("result", {})
+                structured_prompt = result.get("structured_prompt", "")
+                used_seed = result.get("seed", seed)
+
+                # Step 2 to call genearte image
+                payloadForImageGenetrate = {
+                    "prompt": prompt,
+                    "structured_prompt":structured_prompt,
+                    "model_version": model_version,
+                    "aspect_ratio": aspect_ratio,
+                    "steps_num": steps_num,
+                    "guidance_scale": guidance_scale,
+                    "seed": used_seed,
+                }
+                
+                headers = bria_json_headers(api_token)
+
+                response = requests.post(self.generate_api_url, json=payloadForImageGenetrate, headers=headers)
+
+                if response.status_code in (200, 202):
+                    print(
+                        f"Initial request successful to {self.generate_api_url}, polling for completion..."
+                    )
+                    response_dict = response.json()
+                    status_url = response_dict.get("status_url")
+                    request_id = response_dict.get("request_id")
+
+                    if not status_url:
+                        raise Exception("No status_url returned from API")
+
+                    print(f"Request ID: {request_id}, Status URL: {status_url}")
+
+                    final_response = poll_status_until_completed(status_url, api_token)
+
+                    result = final_response.get("result", {})
+                    result_image_url = result.get("image_url")
+                    structured_prompt = result.get("structured_prompt", "")
+                    used_seed = result.get("seed")
+
+                    image_response = requests.get(result_image_url)
+                    result_image = postprocess_image(image_response.content)
+
+                    return (result_image, structured_prompt, used_seed)
+
+            raise Exception(
+                f"Error: API request failed with status code {response.status_code} {response.text}"
+            )
+
+        except Exception as e:
+            raise Exception(f"{e}")

nodes/refine_image_node_v2.py

@@ -1,13 +1,13 @@
 import requests
-from .common import deserialize_and_get_comfy_key, poll_status_until_completed, postprocess_image
+
+from .common import bria_json_headers, poll_status_until_completed, postprocess_image
 
 
-class _BaseRefineImageNodeV2:
-    """Base class for refine image nodes (standard & pro)."""
-
-    api_url = None  # Must be overridden by subclasses
-    generate_api_url = None
+class RefineImageNodeV2:
+    """Standard Refine Image Node"""
 
+    api_url = "https://engine.prod.bria-api.com/v2/structured_prompt/generate"  # Must be overridden by subclasses
+    generate_api_url = "https://engine.prod.bria-api.com/v2/image/generate"
     @classmethod
     def INPUT_TYPES(cls):
         return {
@@ -18,17 +18,32 @@ class _BaseRefineImageNodeV2:
             },
             "optional": {
                 "model_version": (["FIBO"], {"default": "FIBO"}),
-                "negative_prompt": ("STRING", {"default": ""}),
                 "aspect_ratio": (
                     ["1:1", "2:3", "3:2", "3:4", "4:3", "4:5", "5:4", "9:16", "16:9"],
                     {"default": "1:1"},
                 ),
-                "steps_num": ("INT", {"default": 50, "min": 20, "max": 50}),
-                "guidance_scale": ("INT", {"default": 5, "min": 3, "max": 5}),
+                "steps_num": (
+                    "INT",
+                    {
+                        "default": 50,
+                        "min": 35,
+                        "max": 50,
+                    },
+                ),
+                "guidance_scale": (
+                    "INT",
+                    {
+                        "default": 5,
+                        "min": 3,
+                        "max": 5,
+                    },
+                ),
                 "seed": ("INT", {"default": 123456}),
+                "negative_prompt": ("STRING", {"default": ""}),
             },
         }
 
+
     RETURN_TYPES = ("IMAGE", "STRING", "INT")
     RETURN_NAMES = ("image", "structured_prompt", "seed")
     CATEGORY = "API Nodes"
@@ -43,22 +58,23 @@ class _BaseRefineImageNodeV2:
         prompt,
         structured_prompt,
         model_version,
-        negative_prompt,
         aspect_ratio,
         steps_num,
         guidance_scale,
         seed,
     ):
-        return {
+        payload = {
             "prompt": prompt,
             "model_version": model_version,
-            "negative_prompt": negative_prompt,
             "aspect_ratio": aspect_ratio,
             "steps_num": steps_num,
             "guidance_scale": guidance_scale,
             "seed": seed,
-            "structured_prompt": structured_prompt,
         }
+        if structured_prompt:
+            payload["structured_prompt"] = structured_prompt
+        
+        return payload
 
     def execute(
         self,
@@ -66,25 +82,23 @@ class _BaseRefineImageNodeV2:
         prompt,
         structured_prompt,
         model_version,
-        negative_prompt,
         aspect_ratio,
         steps_num,
         guidance_scale,
         seed,
+        negative_prompt=None,
     ):
         self._validate_token(api_token)
         payload = self._build_payload(
             prompt,
             structured_prompt,
             model_version,
-            negative_prompt,
             aspect_ratio,
             steps_num,
             guidance_scale,
             seed,
         )
-        api_token = deserialize_and_get_comfy_key(api_token)
-        headers = {"Content-Type": "application/json", "api_token": api_token}
+        headers = bria_json_headers(api_token)
 
         try:
             response = requests.post(self.api_url, json=payload, headers=headers)
@@ -111,13 +125,14 @@ class _BaseRefineImageNodeV2:
                     "prompt": prompt,
                     "structured_prompt":structured_prompt,
                     "model_version": model_version,
-                    "negative_prompt": negative_prompt,
                     "aspect_ratio": aspect_ratio,
                     "steps_num": steps_num,
                     "guidance_scale": guidance_scale,
                     "seed": used_seed,
+                    "negative_prompt":negative_prompt
                 }
-                headers = {"Content-Type": "application/json", "api_token": api_token}
+                
+                headers = bria_json_headers(api_token)
 
                 response = requests.post(self.generate_api_url, json=payloadForImageGenetrate, headers=headers)
 
@@ -152,10 +167,3 @@ class _BaseRefineImageNodeV2:
 
         except Exception as e:
             raise Exception(f"{e}")
-
-
-class RefineImageNodeV2(_BaseRefineImageNodeV2):
-    """Standard Refine Image Node"""
-    def __init__(self):
-        self.api_url = "https://engine.prod.bria-api.com/v2/structured_prompt/generate"
-        self.generate_api_url = "https://engine.prod.bria-api.com/v2/image/generate"

nodes/reimagine_node.py

@@ -1,6 +1,11 @@
 import requests
 
-from .common import deserialize_and_get_comfy_key, postprocess_image, preprocess_image, image_to_base64
+from .common import (
+    bria_json_headers,
+    image_to_base64,
+    postprocess_image,
+    preprocess_image,
+)
 
 
 class ReimagineNode():
@@ -38,7 +43,6 @@ class ReimagineNode():
             tailored_model_id=None, tailored_model_influence=None, tailored_generation_prefix=None,
             content_moderation=0,
         ):    
-        api_key = deserialize_and_get_comfy_key(api_key)    
         payload = {
             "prompt": tailored_generation_prefix + prompt,
             "num_results": 1,
@@ -59,7 +63,7 @@ class ReimagineNode():
         response = requests.post(
             self.api_url,
             json=payload,
-            headers={"api_token": api_key}
+            headers=bria_json_headers(api_key),
         )
         if response.status_code == 200:
                 response_dict = response.json()

nodes/remove_foreground_node.py

@@ -1,91 +1,93 @@
-import numpy as np
-import requests
-from PIL import Image
 import io
+import requests
+import numpy as np
+from PIL import Image
 import torch
 
-from .common import deserialize_and_get_comfy_key, preprocess_image, image_to_base64, poll_status_until_completed
-
+from .common import (
+    bria_json_headers,
+    image_to_base64,
+    normalize_images_input,
+    poll_status_until_completed,
+)
 
 class RemoveForegroundNode():
     @classmethod
-    def INPUT_TYPES(self):
+    def INPUT_TYPES(cls):
         return {
             "required": {
-                "image": ("IMAGE",),  # Input image from another node
-                "api_key": ("STRING", {"default": "BRIA_API_TOKEN"}), # API Key input with a default value
+                "images": ("IMAGE",),
+                "api_key": ("STRING", {"default": "BRIA_API_TOKEN"}),
             },
             "optional": {
-                "visual_input_content_moderation": ("BOOLEAN", {"default": False}), 
-                "visual_output_content_moderation": ("BOOLEAN", {"default": False}), 
-                "preserve_alpha": ("BOOLEAN", {"default": True}), 
+                "visual_input_content_moderation": ("BOOLEAN", {"default": False}),
+                "visual_output_content_moderation": ("BOOLEAN", {"default": False}),
+                "preserve_alpha": ("BOOLEAN", {"default": True}),
             }
         }
 
     RETURN_TYPES = ("IMAGE",)
-    RETURN_NAMES = ("output_image",)
+    RETURN_NAMES = ("output_images",)
     CATEGORY = "API Nodes"
-    FUNCTION = "execute"  # This is the method that will be executed
+    FUNCTION = "execute"
 
     def __init__(self):
-        self.api_url = "https://engine.prod.bria-api.com/v2/image/edit/erase_foreground"  # remove foreground API URL
+        self.api_url = "https://engine.prod.bria-api.com/v2/image/edit/erase_foreground"
 
-    # Define the execute method as expected by ComfyUI
-    def execute(self, image, visual_input_content_moderation, visual_output_content_moderation, preserve_alpha, api_key):
+    def execute(
+        self,
+        images,
+        visual_input_content_moderation,
+        visual_output_content_moderation,
+        preserve_alpha,
+        api_key
+    ):
         if api_key.strip() == "" or api_key.strip() == "BRIA_API_TOKEN":
             raise Exception("Please insert a valid API key.")
-        api_key = deserialize_and_get_comfy_key(api_key)
+        images = normalize_images_input(images)
+        batch_results = []
 
-        # Check if image is tensor, if so, convert to NumPy array
-        if isinstance(image, torch.Tensor):
-            image = preprocess_image(image)
+        for idx, pil_image in enumerate(images):
+            try:
+                image_base64 = image_to_base64(pil_image)
 
-        # Prepare the API request payload
-        # temporary save the image to /tmp
-        # temp_img_path = "/tmp/temp_img.jpeg"
-        # image.save(temp_img_path, format="JPEG")
-        
-#         files=[('file',('temp_img.jpeg', open(temp_img_path, 'rb'),'image/jpeg'))
-#   ]
-        payload = {
-            "image": image_to_base64(image),
-            "visual_input_content_moderation": visual_input_content_moderation,
-            "visual_output_content_moderation":visual_output_content_moderation,
-            "preserve_alpha": preserve_alpha
-        }
+                payload = {
+                    "image": image_base64,
+                    "visual_input_content_moderation": visual_input_content_moderation,
+                    "visual_output_content_moderation": visual_output_content_moderation,
+                    "preserve_alpha": preserve_alpha
+                }
 
-        headers = {
-            "Content-Type": "application/json",
-            "api_token": f"{api_key}"
-        }
+                headers = bria_json_headers(api_key)
 
-        try:
-            response = requests.post(self.api_url, json=payload, headers=headers)
+                response = requests.post(self.api_url, json=payload, headers=headers)
+                if response.status_code not in (200, 202):
+                    raise Exception(f"API request failed with status {response.status_code}: {response.text}")
 
-            if response.status_code == 200 or response.status_code == 202:
-                print('Initial request successful, polling for completion...')
                 response_dict = response.json()
-                status_url = response_dict.get('status_url')
-                request_id = response_dict.get('request_id')
-                
+                status_url = response_dict.get("status_url")
                 if not status_url:
                     raise Exception("No status_url returned from API")
-                
-                print(f"Request ID: {request_id}, Status URL: {status_url}")
-                
-                # Poll status URL until completion
-                final_response = poll_status_until_completed(status_url, api_key)
-                
-                # Get the result image URL
-                result_image_url = final_response['result']['image_url']
-                
-                image_response = requests.get(result_image_url)
-                result_image = Image.open(io.BytesIO(image_response.content))
-                result_image = np.array(result_image).astype(np.float32) / 255.0
-                result_image = torch.from_numpy(result_image)[None,]
-                return (result_image,)
-            else:
-                raise Exception(f"Error: API request failed with status code {response.status_code} {response.text}")
 
-        except Exception as e:
-            raise Exception(f"{e}")
+                print(f"RemoveForegroundNode - Initial request successful for image {idx}, polling for completion...")
+                final_response = poll_status_until_completed(status_url, api_key)
+                result_image_url = final_response["result"]["image_url"]
+
+                # Download result
+                image_response = requests.get(result_image_url)
+                result_image = Image.open(io.BytesIO(image_response.content)).convert("RGB")
+
+                # Convert to float32 tensor (H, W, C)
+                result_array = np.array(result_image).astype(np.float32) / 255.0
+                result_tensor = torch.from_numpy(result_array)
+
+                batch_results.append(result_tensor)
+
+            except Exception as e:
+                print(f"[RemoveForegroundNode] Skipping image {idx} due to error: {e}")
+                # fallback: use original image as tensor
+                fallback_array = np.array(pil_image).astype(np.float32) / 255.0
+                batch_results.append(torch.from_numpy(fallback_array))
+
+        # Return list of tensors
+        return (batch_results,)

nodes/replace_bg_node.py

@@ -1,124 +1,122 @@
-import numpy as np
-import requests
-from PIL import Image
 import io
+import requests
+import numpy as np
+from PIL import Image
 import torch
 
-from .common import deserialize_and_get_comfy_key, image_to_base64, preprocess_image, preprocess_mask, poll_status_until_completed
-
+from .common import (
+    bria_json_headers,
+    image_to_base64,
+    normalize_images_input,
+    poll_status_until_completed,
+)
 
 class ReplaceBgNode():
     @classmethod
-    def INPUT_TYPES(self):
+    def INPUT_TYPES(cls):
         return {
             "required": {
-                "image": ("IMAGE",),  # Input image from another node
-                "api_key": ("STRING", {"default": "BRIA_API_TOKEN"}), # API Key input with a default value
+                "images": ("IMAGE",),
+                "api_key": ("STRING", {"default": "BRIA_API_TOKEN"}),
             },
             "optional": {
                 "mode": (["base", "fast", "high_control"], {"default": "base"}),
-                "prompt": ("STRING",),
+                "prompt": ("STRING", {"default": ""}),
                 "ref_images": ("IMAGE",),
-                "refine_prompt": ("BOOLEAN", {"default": True}), 
-                "enhance_ref_images": ("BOOLEAN", {"default": True}), 
-                "original_quality": ("BOOLEAN", {"default": False}), 
-                "negative_prompt": ("STRING", {"default": None}), 
-                "seed": ("INT", {"default": 681794}),
-                "visual_output_content_moderation": ("BOOLEAN", {"default": False}), 
+                "refine_prompt": ("BOOLEAN", {"default": True}),
+                "enhance_ref_images": ("BOOLEAN", {"default": True}),
+                "original_quality": ("BOOLEAN", {"default": False}),
+                "negative_prompt": ("STRING", {"default": None}),
+                "seed": ("STRING", {"default": "681794"}),  # Accept comma-separated seeds
+                "visual_output_content_moderation": ("BOOLEAN", {"default": False}),
                 "prompt_content_moderation": ("BOOLEAN", {"default": False}),
                 "force_background_detection": ("BOOLEAN", {"default": False}),
             }
         }
 
     RETURN_TYPES = ("IMAGE",)
-    RETURN_NAMES = ("output_image",)
+    RETURN_NAMES = ("output_images",)
     CATEGORY = "API Nodes"
-    FUNCTION = "execute"  # This is the method that will be executed
+    FUNCTION = "execute"
 
     def __init__(self):
-        self.api_url = "https://engine.prod.bria-api.com/v2/image/edit/replace_background"  # Replace BG API URL
+        self.api_url = "https://engine.prod.bria-api.com/v2/image/edit/replace_background"
 
-    # Define the execute method as expected by ComfyUI
-    def execute(self, image, mode,
-                refine_prompt,
-                original_quality,
-                negative_prompt,
-                seed,
-                api_key,
-                visual_output_content_moderation,
-                prompt_content_moderation,
-                enhance_ref_images,
-                force_background_detection,
-                prompt=None,
-                ref_images=None,):
-        if api_key.strip() == "" or api_key.strip() == "BRIA_API_TOKEN":
+    def execute(
+        self,
+        images,
+        mode,
+        refine_prompt,
+        original_quality,
+        negative_prompt,
+        seed,
+        api_key,
+        visual_output_content_moderation,
+        prompt_content_moderation,
+        enhance_ref_images,
+        force_background_detection,
+        prompt=None,
+        ref_images=None
+    ):
+        if api_key.strip() in ("", "BRIA_API_TOKEN"):
             raise Exception("Please insert a valid API key.")
-        api_key = deserialize_and_get_comfy_key(api_key)
+        images = normalize_images_input(images)
 
-        # Check if image and mask are tensors, if so, convert to NumPy arrays
-        if isinstance(image, torch.Tensor):
-            image = preprocess_image(image)
-
-        # Convert the image to Base64 string
-        image_base64 = image_to_base64(image)
-        
+        # Normalize reference images
+        ref_images_base64 = []
         if ref_images is not None:
-            ref_images = preprocess_image(ref_images)
-            ref_images = [image_to_base64(ref_images)]
-        else:
-            ref_images=[]
+            ref_images_list = normalize_images_input(ref_images)
+            ref_images_base64 = [image_to_base64(img) for img in ref_images_list]
 
-        # Prepare the API request payload for v2 API
-        payload = {
-            "image": image_base64,
-            "mode": mode,
-            "prompt": prompt,
-            "ref_images":ref_images,
-            "refine_prompt": refine_prompt,
-            "original_quality": original_quality,
-            "negative_prompt": negative_prompt,
-            "seed": seed,
-            "prompt_content_moderation": prompt_content_moderation,
-            "visual_output_content_moderation":visual_output_content_moderation,
-            "enhance_ref_images":enhance_ref_images,
-            "force_background_detection": force_background_detection        
-        }   
+        # Prepare per-image seeds
+        seed_values = [int(s.strip()) for s in seed.split(",")] if isinstance(seed, str) else [seed]
+        if len(seed_values) < len(images):
+            seed_values += [seed_values[-1]] * (len(images) - len(seed_values))
 
-        headers = {
-            "Content-Type": "application/json",
-            "api_token": f"{api_key}"
-        }
+        batch_results = []
+
+        for idx, pil_image in enumerate(images):
+            try:
+                image_base64 = image_to_base64(pil_image)
+
+                payload = {
+                    "image": image_base64,
+                    "mode": mode,
+                    "prompt": prompt,
+                    "ref_images": ref_images_base64,
+                    "refine_prompt": refine_prompt,
+                    "original_quality": original_quality,
+                    "negative_prompt": negative_prompt,
+                    "seed": seed_values[idx],
+                    "prompt_content_moderation": prompt_content_moderation,
+                    "visual_output_content_moderation": visual_output_content_moderation,
+                    "enhance_ref_images": enhance_ref_images,
+                    "force_background_detection": force_background_detection
+                }
+
+                headers = bria_json_headers(api_key)
+                response = requests.post(self.api_url, json=payload, headers=headers)
+                if response.status_code not in (200, 202):
+                    raise Exception(f"API request failed with status {response.status_code}: {response.text}")
 
-        try:
-            response = requests.post(self.api_url, json=payload, headers=headers)
-            
-            if response.status_code == 200 or response.status_code == 202:
-                print('Initial replace background request successful, polling for completion...')
                 response_dict = response.json()
-                status_url = response_dict.get('status_url')
-                request_id = response_dict.get('request_id')
-                
+                status_url = response_dict.get("status_url")
                 if not status_url:
                     raise Exception("No status_url returned from API")
-                
-                print(f"Request ID: {request_id}, Status URL: {status_url}")
-                
-                # Poll status URL until completion
-                final_response = poll_status_until_completed(status_url, api_key)
-                
-                # Get the result image URL
-                result_image_url = final_response['result']['image_url']
-                
-                # Download and process the result image
-                image_response = requests.get(result_image_url)
-                result_image = Image.open(io.BytesIO(image_response.content))
-                result_image = result_image.convert("RGB")
-                result_image = np.array(result_image).astype(np.float32) / 255.0
-                result_image = torch.from_numpy(result_image)[None,]
-                
-                return (result_image,)
-            else:
-                raise Exception(f"Error: API request failed with status code {response.status_code}{response.text}")
 
-        except Exception as e:
-            raise Exception(f"{e}")
+                print(f"ReplaceBgNode - Initial request successful for image {idx}, polling for completion...")
+                final_response = poll_status_until_completed(status_url, api_key)
+                result_image_url = final_response["result"]["image_url"]
+
+                image_response = requests.get(result_image_url)
+                result_image = Image.open(io.BytesIO(image_response.content)).convert("RGB")
+                result_tensor = torch.from_numpy(np.array(result_image).astype(np.float32) / 255.0)
+
+                batch_results.append(result_tensor)
+
+            except Exception as e:
+                print(f"[ReplaceBgNode] Skipping image {idx} due to error: {e}")
+                fallback_array = np.array(pil_image).astype(np.float32) / 255.0
+                batch_results.append(torch.from_numpy(fallback_array))
+
+        return (batch_results,)

nodes/rmbg_node.py

@@ -1,87 +1,89 @@
-import numpy as np
-import requests
-from PIL import Image
 import io
+import requests
+import numpy as np
+from PIL import Image
 import torch
 
-from .common import deserialize_and_get_comfy_key, preprocess_image, image_to_base64, poll_status_until_completed
+from .common import (
+    bria_json_headers,
+    image_to_base64,
+    normalize_images_input,
+    poll_status_until_completed,
+    to_pil_safe,
+)
 
 class RmbgNode():
     @classmethod
-    def INPUT_TYPES(self):
+    def INPUT_TYPES(cls):
         return {
             "required": {
-                "image": ("IMAGE",),  # Input image from another node
-                "api_key": ("STRING", {"default": "BRIA_API_TOKEN"}), # API Key input with a default value
+                "images": ("IMAGE",),  # Accepts list of PIL Images or single tensor
+                "api_key": ("STRING", {"default": "BRIA_API_TOKEN"}), 
             },
             "optional": {
                 "visual_input_content_moderation": ("BOOLEAN", {"default": False}), 
                 "visual_output_content_moderation": ("BOOLEAN", {"default": False}), 
                 "preserve_alpha": ("BOOLEAN", {"default": True}), 
-            
             }               
         }
 
     RETURN_TYPES = ("IMAGE",)
-    RETURN_NAMES = ("output_image",)
+    RETURN_NAMES = ("output_images",)
     CATEGORY = "API Nodes"
-    FUNCTION = "execute"  # This is the method that will be executed
+    FUNCTION = "execute"
 
     def __init__(self):
-        self.api_url = "https://engine.prod.bria-api.com/v2/image/edit/remove_background"  # RMBG API URL
+        self.api_url = "https://engine.prod.bria-api.com/v2/image/edit/remove_background"
 
-    # Define the execute method as expected by ComfyUI
-    def execute(self, image, visual_input_content_moderation, visual_output_content_moderation, preserve_alpha, api_key):
+    def execute(self, images, visual_input_content_moderation, visual_output_content_moderation, preserve_alpha, api_key):
         if api_key.strip() == "" or api_key.strip() == "BRIA_API_TOKEN":
             raise Exception("Please insert a valid API key.")
-        api_key = deserialize_and_get_comfy_key(api_key)
-        # Check if image is tensor, if so, convert to NumPy array
-        if isinstance(image, torch.Tensor):
-            image = preprocess_image(image)
+        # Normalize input to list of PIL images
+        images = normalize_images_input(images)
 
-        # Convert image to base64 for the new API format
-        image_base64 = image_to_base64(image)
-        payload = {
-            "image": image_base64,
-            "visual_input_content_moderation": visual_input_content_moderation,
-            "visual_output_content_moderation":visual_output_content_moderation,
-            "preserve_alpha":preserve_alpha
-        }
+        batch_results = []
 
-        headers = {
-            "Content-Type": "application/json",
-            "api_token": f"{api_key}"
-        }
+        for idx, pil_image in enumerate(images):
+            try:
+                image_base64 = image_to_base64(pil_image)
 
-        try:
-            response = requests.post(self.api_url, json=payload, headers=headers)
-            
-            if response.status_code == 200 or response.status_code == 202:
-                print('Initial RMBG request successful, polling for completion...')
+                payload = {
+                    "image": image_base64,
+                    "visual_input_content_moderation": visual_input_content_moderation,
+                    "visual_output_content_moderation": visual_output_content_moderation,
+                    "preserve_alpha": preserve_alpha
+                }
+
+                headers = bria_json_headers(api_key)
+
+                response = requests.post(self.api_url, json=payload, headers=headers)
+                if response.status_code not in (200, 202):
+                    raise Exception(f"API request failed with status {response.status_code}: {response.text}")
+
+                # Poll until completion
                 response_dict = response.json()
-
                 status_url = response_dict.get('status_url')
-                request_id = response_dict.get('request_id')
-                
                 if not status_url:
                     raise Exception("No status_url returned from API")
-                
-                print(f"Request ID: {request_id}, Status URL: {status_url}")
-                
+
+                print(f"RmbgNode - Initial request successful for image {idx}, polling for completion...")
                 final_response = poll_status_until_completed(status_url, api_key)
-                
-                # Get the result image URL
                 result_image_url = final_response['result']['image_url']
-                
-                # Download and process the result image
+
+                # Download result
                 image_response = requests.get(result_image_url)
                 result_image = Image.open(io.BytesIO(image_response.content))
-                result_image = np.array(result_image).astype(np.float32) / 255.0
-                result_image = torch.from_numpy(result_image)[None,]
-                
-                return (result_image,)
-            else:
-                raise Exception(f"Error: API request failed with status code {response.status_code} {response.text}")
 
-        except Exception as e:
-            raise Exception(f"{e}")
+                # Convert to float32 tensor (H, W, C), 0-1
+                result_array = np.array(result_image).astype(np.float32) / 255.0
+                result_tensor = torch.from_numpy(result_array)  # shape: (H,W,4)
+                batch_results.append(result_tensor)
+
+            except Exception as e:
+                print(f"[RmbgNode] Skipping image {idx} due to error: {e}")
+                # Append empty tensor of the same size as original
+                empty_array = np.zeros((pil_image.height, pil_image.width, 4), dtype=np.float32)
+                batch_results.append(torch.from_numpy(empty_array))
+
+        # Return list of tensors (Comfy preview handles this correctly)
+        return (batch_results,)

nodes/tailored_gen_node.py

@@ -1,6 +1,11 @@
 import requests
 
-from .common import deserialize_and_get_comfy_key, postprocess_image, preprocess_image, image_to_base64
+from .common import (
+    bria_json_headers,
+    image_to_base64,
+    postprocess_image,
+    preprocess_image,
+)
 
 
 class TailoredGenNode():
@@ -45,7 +50,6 @@ class TailoredGenNode():
             guidance_method_2=None, guidance_method_2_scale=None, guidance_method_2_image=None,
             content_moderation=0,
         ):
-        api_key = deserialize_and_get_comfy_key(api_key)
         payload = {
             "prompt": generation_prefix + prompt,
             "num_results": 1,
@@ -74,7 +78,7 @@ class TailoredGenNode():
         response = requests.post(
             self.api_url + model_id,
             json=payload,
-            headers={"api_token": api_key}
+            headers=bria_json_headers(api_key),
         )
         if response.status_code == 200:
                 response_dict = response.json()

nodes/tailored_model_info_node.py

@@ -1,5 +1,5 @@
 import requests
-from .common import deserialize_and_get_comfy_key
+from .common import bria_json_headers
 
 class TailoredModelInfoNode():
     @classmethod
@@ -21,10 +21,9 @@ class TailoredModelInfoNode():
 
     # Define the execute method as expected by ComfyUI
     def execute(self, model_id, api_key):
-        api_key = deserialize_and_get_comfy_key(api_key)
         response = requests.get(
             self.api_url + model_id,
-            headers={"api_token": api_key}
+            headers=bria_json_headers(api_key),
         )
         if response.status_code == 200:
             generation_prefix = response.json()["generation_prefix"]

nodes/tailored_portrait_node.py

@@ -1,19 +1,24 @@
-import numpy as np
-import requests
-from PIL import Image
 import io
+import requests
+import numpy as np
+from PIL import Image
 import torch
 
-from .common import deserialize_and_get_comfy_key, image_to_base64, preprocess_image
+from .common import (
+    bria_json_headers,
+    image_to_base64,
+    normalize_images_input,
+    to_pil_safe,
+)
 
 class TailoredPortraitNode():
     @classmethod
-    def INPUT_TYPES(self):
+    def INPUT_TYPES(cls):
         return {
             "required": {
-                "image": ("IMAGE",),  # Input image from another node
-                "tailored_model_id": ("STRING",), # API Key input with a default value
-                "api_key": ("STRING", {"default": "BRIA_API_TOKEN"}), # API Key input with a default value
+                "images": ("IMAGE",),
+                "tailored_model_id": ("STRING",),
+                "api_key": ("STRING", {"default": "BRIA_API_TOKEN"}),
             },
             "optional": {
                 "seed": ("INT", {"default": 123456}),
@@ -23,54 +28,61 @@ class TailoredPortraitNode():
         }
 
     RETURN_TYPES = ("IMAGE",)
-    RETURN_NAMES = ("output_image",)
+    RETURN_NAMES = ("output_images",)
     CATEGORY = "API Nodes"
-    FUNCTION = "execute"  # This is the method that will be executed
+    FUNCTION = "execute"
 
     def __init__(self):
-        self.api_url = "https://engine.prod.bria-api.com/v1/tailored-gen/restyle_portrait"  # Eraser API URL
+        self.api_url = "https://engine.prod.bria-api.com/v1/tailored-gen/restyle_portrait"
 
-    # Define the execute method as expected by ComfyUI
-    def execute(self, image, tailored_model_id, api_key, seed, tailored_model_influence, id_strength):
+    def execute(
+        self,
+        images,
+        tailored_model_id,
+        api_key,
+        seed,
+        tailored_model_influence,
+        id_strength
+    ):
         if api_key.strip() == "" or api_key.strip() == "BRIA_API_TOKEN":
             raise Exception("Please insert a valid API key.")
-        api_key = deserialize_and_get_comfy_key(api_key)
+        # Normalize images to list of PIL images
+        images = normalize_images_input(images)
 
-        # Convert the image and mask directly to  if isinstance(image, torch.Tensor):
-        if isinstance(image, torch.Tensor):
-            image = preprocess_image(image)
-            
-        image_base64 = image_to_base64(image)
+        batch_results = []
 
-        # Prepare the API request payload
-        payload = {
-            "id_image_file": f"{image_base64}",
-            "tailored_model_id": int(tailored_model_id),
-            "tailored_model_influence": tailored_model_influence,
-            "id_strength": id_strength,
-            "seed": seed
-        }
+        for idx, pil_image in enumerate(images):
+            try:
+                image_base64 = image_to_base64(pil_image)
 
-        headers = {
-            "Content-Type": "application/json",
-            "api_token": f"{api_key}"
-        }
+                payload = {
+                    "id_image_file": image_base64,
+                    "tailored_model_id": int(tailored_model_id),
+                    "tailored_model_influence": tailored_model_influence,
+                    "id_strength": id_strength,
+                    "seed": seed
+                }
+
+                headers = bria_json_headers(api_key)
+
+                response = requests.post(self.api_url, json=payload, headers=headers)
+                if response.status_code != 200:
+                    raise Exception(f"API request failed with status {response.status_code}: {response.text}")
 
-        try:
-            response = requests.post(self.api_url, json=payload, headers=headers)
-            # Check for successful response
-            if response.status_code == 200:
-                print('response is 200')
-                # Process the output image from API response
                 response_dict = response.json()
-                image_response = requests.get(response_dict['image_res'])
-                result_image = Image.open(io.BytesIO(image_response.content))
-                result_image = result_image.convert("RGB")
-                result_image = np.array(result_image).astype(np.float32) / 255.0
-                result_image = torch.from_numpy(result_image)[None,]
-                return (result_image,)
-            else:
-                raise Exception(f"Error: API request failed with status code {response.status_code} {response.text}")
+                image_response = requests.get(response_dict["image_res"])
+                result_image = Image.open(io.BytesIO(image_response.content)).convert("RGB")
 
-        except Exception as e:
-            raise Exception(f"{e}")
+                # Convert to float32 tensor (H,W,C), 0-1
+                result_array = np.array(result_image).astype(np.float32) / 255.0
+                result_tensor = torch.from_numpy(result_array)
+
+                batch_results.append(result_tensor)
+
+            except Exception as e:
+                print(f"[TailoredPortraitNode] Skipping image {idx} due to error: {e}")
+                fallback_array = np.array(pil_image).astype(np.float32) / 255.0
+                batch_results.append(torch.from_numpy(fallback_array))
+
+        # Return list of tensors (avoids size/dtype mismatch)
+        return (batch_results,)

nodes/text_2_image_base_node.py

@@ -1,6 +1,11 @@
 import requests
 
-from .common import deserialize_and_get_comfy_key, postprocess_image, preprocess_image, image_to_base64
+from .common import (
+    bria_json_headers,
+    image_to_base64,
+    postprocess_image,
+    preprocess_image,
+)
 
 
 class Text2ImageBaseNode():
@@ -48,7 +53,6 @@ class Text2ImageBaseNode():
             image_prompt_mode=None, image_prompt_image=None, image_prompt_scale=None,
             content_moderation=0,
         ):
-        api_key = deserialize_and_get_comfy_key(api_key)
         payload = {
             "prompt": prompt,
             "num_results": 1,
@@ -84,7 +88,7 @@ class Text2ImageBaseNode():
         response = requests.post(
             self.api_url,
             json=payload,
-            headers={"api_token": api_key}
+            headers=bria_json_headers(api_key),
         )
         if response.status_code == 200:
                 response_dict = response.json()

nodes/text_2_image_fast_node.py

@@ -1,6 +1,11 @@
 import requests
 
-from .common import deserialize_and_get_comfy_key, postprocess_image, preprocess_image, image_to_base64
+from .common import (
+    bria_json_headers,
+    image_to_base64,
+    postprocess_image,
+    preprocess_image,
+)
 
 
 class Text2ImageFastNode():
@@ -45,7 +50,6 @@ class Text2ImageFastNode():
             image_prompt_mode=None, image_prompt_image=None, image_prompt_scale=None,
             content_moderation=0,
         ):
-        api_key = deserialize_and_get_comfy_key(api_key)
         payload = {
             "prompt": prompt,
             "num_results": 1,
@@ -77,7 +81,7 @@ class Text2ImageFastNode():
         response = requests.post(
             self.api_url,
             json=payload,
-            headers={"api_token": api_key}
+            headers=bria_json_headers(api_key),
         )
         if response.status_code == 200:
                 response_dict = response.json()

nodes/text_2_image_hd_node.py

@@ -1,6 +1,6 @@
 import requests
 
-from .common import deserialize_and_get_comfy_key, postprocess_image
+from .common import bria_json_headers, postprocess_image
 
 
 class Text2ImageHDNode():
@@ -35,7 +35,6 @@ class Text2ImageHDNode():
             self, api_key, prompt, aspect_ratio, seed, negative_prompt, 
             steps_num, prompt_enhancement, text_guidance_scale, medium, content_moderation=0,
         ):
-        api_key = deserialize_and_get_comfy_key(api_key)
         payload = {
             "prompt": prompt,
             "num_results": 1,
@@ -53,7 +52,7 @@ class Text2ImageHDNode():
         response = requests.post(
             self.api_url,
             json=payload,
-            headers={"api_token": api_key}
+            headers=bria_json_headers(api_key),
         )
         if response.status_code == 200:
                 response_dict = response.json()

nodes/utils/shot_utils.py

@@ -1,6 +1,11 @@
 import requests
 import torch
-from ..common import deserialize_and_get_comfy_key, postprocess_image, preprocess_image, image_to_base64
+from ..common import (
+    bria_json_headers,
+    image_to_base64,
+    postprocess_image,
+    preprocess_image,
+)
 
 shot_by_text_api_url = (
     "https://engine.prod.bria-api.com/v1/product/lifestyle_shot_by_text"
@@ -130,8 +135,7 @@ def make_api_request(api_url, payload, api_key, Placement_type = None):
 
 
     try:
-        api_key = deserialize_and_get_comfy_key(api_key)
-        headers = {"Content-Type": "application/json", "api_token": f"{api_key}"}
+        headers = bria_json_headers(api_key)
         response = requests.post(api_url, json=payload, headers=headers)
 
         if response.status_code == 200:

nodes/video_nodes/green_screen_video_node.py

@@ -0,0 +1,116 @@
+import os
+import uuid
+import requests
+from ..common import (
+    bria_json_headers,
+    poll_status_until_completed,
+)
+from .video_utils import upload_video_to_s3
+
+
+class GreenScreenVideoNode():
+    """
+    Applies green-screen (chroma key) background removal using the Bria API
+    (POST /v2/video/edit/green_screen). Output is a processed video with a solid-color background.
+    """
+    @classmethod
+    def INPUT_TYPES(self):
+        return {
+            "required": {
+                "api_key": ("STRING", {"default": "BRIA_API_TOKEN"}),
+                "video_url": ("STRING", {
+                    "default": "",
+                    "tooltip": "Local path or publicly accessible URL of the video to process.",
+                }),
+            },
+            "optional": {
+                "green_shade": ([
+                    "broadcast_green",
+                    "chroma_green",
+                    "blue_screen",
+                ], {"default": "broadcast_green"}),
+                "output_container_and_codec": ([
+                    "mp4_h264",
+                    "mp4_h265",
+                    "webm_vp9",
+                    "mov_h265",
+                    "mov_proresks",
+                    "mkv_h264",
+                    "mkv_h265",
+                    "mkv_vp9",
+                    "gif"
+                ], {"default": "mp4_h264"}),
+                "preserve_audio": ("BOOLEAN", {"default": True}),
+            }
+        }
+
+    RETURN_TYPES = ("STRING",)
+    RETURN_NAMES = ("result_video_url",)
+    CATEGORY = "API Nodes"
+    FUNCTION = "execute"
+
+    def __init__(self):
+        self.api_url = "https://engine.prod.bria-api.com/v2/video/edit/green_screen"
+
+    def execute(
+        self,
+        api_key,
+        video_url,
+        green_shade="broadcast_green",
+        output_container_and_codec="mp4_h264",
+        preserve_audio=True,
+    ):
+        if api_key.strip() == "" or api_key.strip() == "BRIA_API_TOKEN":
+            raise Exception("Please insert a valid API key.")
+
+        if not video_url or not str(video_url).strip():
+            raise Exception("video_url is required: provide a local path or a publicly accessible video URL.")
+
+        if os.path.exists(video_url):
+            filename = f"{str(uuid.uuid4())}_{os.path.basename(video_url)}"
+            input_video_url = upload_video_to_s3(video_url, filename, api_key)
+            if not input_video_url or not (
+                input_video_url.startswith("http://") or input_video_url.startswith("https://")
+            ):
+                raise Exception(f"Failed to upload video to S3. Got: {input_video_url}")
+        else:
+            input_video_url = video_url.strip()
+
+        try:
+            print("Calling Bria API for video green screen...")
+            payload = {
+                "video": input_video_url,
+                "green_shade": green_shade,
+                "output_container_and_codec": output_container_and_codec,
+                "preserve_audio": preserve_audio,
+            }
+            headers = bria_json_headers(api_key)
+
+
+            response = requests.post(self.api_url, json=payload, headers=headers)
+
+            if response.status_code == 200 or response.status_code == 202:
+                print("Initial video green-screen request accepted, polling for completion...")
+                response_dict = response.json()
+
+                status_url = response_dict.get("status_url")
+                request_id = response_dict.get("request_id")
+
+                if not status_url:
+                    raise Exception("No status_url returned from API")
+
+                print(f"Request ID: {request_id}, Status URL: {status_url}")
+
+                final_response = poll_status_until_completed(
+                    status_url, api_key, timeout=3600, check_interval=5
+                )
+
+                result_video_url = final_response["result"]["video_url"]
+
+                print(f"Video processing completed. Result URL: {result_video_url}")
+                return (result_video_url,)
+
+            raise Exception(f"Error: API request failed with status code {response.status_code} {response.text}")
+
+        except Exception as e:
+            raise Exception(f"{e}")

nodes/video_nodes/load_video.py

@@ -0,0 +1,52 @@
+import os
+import folder_paths
+
+class LoadVideoFramesNode:
+    """
+    Load a video file from the input folder or upload.
+
+    Parameters:
+        video (str): Selected or uploaded video filename.
+
+    Returns:
+        video_path (STRING): Absolute path to the video file.
+    """
+    
+    @classmethod
+    def INPUT_TYPES(cls):
+        input_dir = folder_paths.get_input_directory()
+        files = [f for f in os.listdir(input_dir) if os.path.isfile(os.path.join(input_dir, f))]
+        files = folder_paths.filter_files_content_types(files, ["video"])
+        
+        return {
+            "required": {
+                "video": (sorted(files), {"video_upload": True}),
+            }
+        }
+    
+    RETURN_TYPES = ("STRING",)
+    RETURN_NAMES = ("video_path",)
+    FUNCTION = "load_video"
+    CATEGORY = "API Nodes"
+
+    def load_video(self, video):
+        video_path = folder_paths.get_annotated_filepath(video)        
+        if not os.path.exists(video_path):
+            raise FileNotFoundError(f"Video file not found: {video_path}")
+    
+        return (video_path,)
+    
+    @classmethod
+    def IS_CHANGED(cls, video, **kwargs):
+        """Force re-execution when video file changes"""
+        video_path = folder_paths.get_annotated_filepath(video)
+        if os.path.exists(video_path):
+            return os.path.getmtime(video_path)
+        return float("nan")
+    
+    @classmethod
+    def VALIDATE_INPUTS(cls, video, **kwargs):
+        """Validate that the video file exists"""
+        if not folder_paths.exists_annotated_filepath(video):
+            return f"Invalid video file: {video}"
+        return True

nodes/video_nodes/preview_video_node_from_url.py

@@ -0,0 +1,135 @@
+import os
+import uuid
+import folder_paths
+import requests
+
+class PreviewVideoURLNode:
+    """
+    Bria Preview Video URL Node
+    
+    This node takes a video URL as a string and downloads it to preview
+    directly in the ComfyUI interface.
+    
+    Parameters:
+    - video_url: URL of the video to preview (http/https)
+    """
+   
+    def __init__(self):
+        self.output_dir = folder_paths.get_temp_directory()
+        self.type = "temp"
+    
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "video_url": ("STRING", {
+                    "default": "",
+                    "multiline": False,
+                    "tooltip": "URL of the video to preview (http/https)"
+                }),
+            },
+            "hidden": {
+                "prompt": "PROMPT",
+                "extra_pnginfo": "EXTRA_PNGINFO"
+            },
+        }
+    
+    RETURN_TYPES = ()
+    FUNCTION = "preview_video_url"
+    OUTPUT_NODE = True
+    CATEGORY = "API Nodes"
+    DESCRIPTION = "Previews a video from URL directly in the ComfyUI interface."
+
+    def preview_video_url(self, video_url, prompt=None, extra_pnginfo=None):
+        """
+        Preview video from URL
+        
+        Args:
+            video_url: URL of the video (http/https)
+            prompt: Hidden parameter for ComfyUI workflow
+            extra_pnginfo: Hidden parameter for ComfyUI metadata
+            
+        Returns:
+            dict: UI output with video file for preview
+        """
+        if not video_url or video_url.strip() == "":
+            raise ValueError("video_url cannot be empty")
+        
+        if not video_url.startswith("http://") and not video_url.startswith("https://"):
+            raise ValueError("video_url must be a valid HTTP or HTTPS URL")
+        
+        print(f"Downloading video from URL: {video_url}")
+        
+        # Download video from URL
+        try:
+            response = requests.get(video_url, stream=True, timeout=60)
+            response.raise_for_status()
+            
+            # Determine file extension from URL or Content-Type
+            content_type = response.headers.get('Content-Type', '')
+            extension = self._get_extension_from_content_type(content_type, video_url)
+            
+            filename_prefix = str(uuid.uuid4()) + "_video_url_preview"
+            
+            # Get save path
+            full_output_folder = self.output_dir            
+            filename = f"{filename_prefix}.{extension}"
+            filepath = os.path.join(full_output_folder, filename)
+  
+            
+            # Save video to temp directory
+            print(f"Saving video to: {filepath}")
+            with open(filepath, 'wb') as f:
+                for chunk in response.iter_content(chunk_size=8192):
+                    if chunk:
+                        f.write(chunk)
+            
+            file_size = os.path.getsize(filepath)
+            print(f"Video downloaded successfully: {filename} ({file_size / (1024*1024):.2f} MB)")
+            
+            return {
+                "ui": {
+                    "images": [{
+                        "filename": filename,
+                        "subfolder": "",
+                        "type": self.type,
+                        "format": extension
+                    }],
+                    "animated": (True,),
+                    "has_audio": (True,)
+                }
+            }
+            
+        except requests.exceptions.RequestException as e:
+            raise Exception(f"Failed to download video from URL: {str(e)}")
+        except Exception as e:
+            raise Exception(f"Error previewing video: {str(e)}")
+    
+    def _get_extension_from_content_type(self, content_type, url):
+        """
+        Determine file extension from Content-Type header or URL
+        """
+        # Map common video MIME types to extensions
+        content_type_map = {
+            'video/mp4': 'mp4',
+            'video/webm': 'webm',
+            'video/quicktime': 'mov',
+            'video/x-matroska': 'mkv',
+            'video/x-msvideo': 'avi',
+            'image/gif': 'gif',
+        }
+        
+        # Try to get extension from Content-Type
+        for mime_type, ext in content_type_map.items():
+            if mime_type in content_type.lower():
+                return ext
+        
+        # Try to get extension from URL
+        url_path = url.split('?')[0]  # Remove query parameters
+        if '.' in url_path:
+            url_ext = url_path.rsplit('.', 1)[-1].lower()
+            if url_ext in ['mp4', 'webm', 'mov', 'mkv', 'avi', 'gif', 'webp']:
+                return url_ext
+        
+        # Default to mp4
+        return 'mp4'

nodes/video_nodes/remove_video_background_node.py

@@ -0,0 +1,130 @@
+import os
+import uuid
+import requests
+import folder_paths
+from ..common import bria_json_headers, poll_status_until_completed
+from .video_utils import upload_video_to_s3
+
+class RemoveVideoBackgroundNode():
+    """
+    Removes the background from a video using the Bria API.
+
+    Parameters:
+        api_key (str): Your Bria API key.
+        video_url (str): Local path or URL of the video to process.
+        preserve_audio (bool, optional): Whether to keep the audio track. Default is True.
+        output_container_and_codec (str, optional): Desired output format and codec. Default is "webm_vp9".
+        background_color Predefined string only - one of the predefined enum values
+
+    Returns:
+        result_video_url (STRING): URL of the video with background removed.
+    """
+    @classmethod
+    def INPUT_TYPES(self):
+        return {
+            "required": {
+                "api_key": ("STRING", {"default": "BRIA_API_TOKEN"}),
+                "video_url": ("STRING", {
+                    "default": "",
+                    "tooltip": "URL of video to process (provide either frames or video_url)"
+                }),
+            },
+            "optional": {
+                "preserve_audio": ("BOOLEAN", {"default": True}), 
+                "output_container_and_codec": ([
+                    "mp4_h264",
+                    "mp4_h265",
+                    "webm_vp9",
+                    "mov_h265",
+                    "mov_proresks",
+                    "mkv_h264",
+                    "mkv_h265",
+                    "mkv_vp9",
+                    "gif"
+                ], {"default": "webm_vp9"}),
+                "background_color": ([
+                    "Transparent",
+                    "Black",
+                    "White",
+                    "Gray",
+                    "Red",
+                    "Green",
+                    "Blue",
+                    "Yellow",
+                    "Cyan",
+                    "Magenta",
+                    "Orange"
+                ], {"default": "Black"})
+            }
+        }
+
+    RETURN_TYPES = ("STRING",)
+    RETURN_NAMES = ("result_video_url",)
+    CATEGORY = "API Nodes"
+    FUNCTION = "execute"
+
+    def __init__(self):
+        self.api_url = "https://engine.prod.bria-api.com/v2/video/edit/remove_background"
+
+    def execute(self, api_key, video_url, preserve_audio=True, output_container_and_codec="webm_vp9",background_color="Black"):
+        if api_key.strip() == "" or api_key.strip() == "BRIA_API_TOKEN":
+            raise Exception("Please insert a valid API key.")
+        video_path = None
+
+        input_video_url = ""
+        if video_url and video_url.strip() != "":
+            if os.path.exists(video_url):
+                filename = f"{ str(uuid.uuid4())}_{os.path.basename(video_url)}"
+                input_video_url = upload_video_to_s3(video_url, filename, api_key)
+                if video_url.startswith(folder_paths.get_temp_directory()):
+                    video_path = None 
+            else:
+                input_video_url = video_url
+        
+        try:
+            
+            print("Step 3: Calling Bria API for background removal...")
+            payload = {
+                "video": input_video_url,
+                "preserve_audio": preserve_audio,
+                "output_container_and_codec": output_container_and_codec,
+                "background_color":background_color
+            }
+
+            headers = bria_json_headers(api_key)
+
+            response = requests.post(self.api_url, json=payload, headers=headers)
+            
+            if response.status_code == 200 or response.status_code == 202:
+                print('Initial Video RMBG request successful, polling for completion...')
+                response_dict = response.json()
+
+                status_url = response_dict.get('status_url')
+                request_id = response_dict.get('request_id')
+                
+                if not status_url:
+                    raise Exception("No status_url returned from API")
+                
+                print(f"Request ID: {request_id}, Status URL: {status_url}")
+                
+                final_response = poll_status_until_completed(status_url, api_key, timeout=3600, check_interval=5)
+                
+                result_video_url = final_response['result']['video_url']
+                
+                print(f"Video processing completed. Result URL: {result_video_url}")
+                print(f"Background removal complete! Use Preview Video URL node to view the result.")
+                
+                return (result_video_url,)
+            else:
+                raise Exception(f"Error: API request failed with status code {response.status_code} {response.text}")
+
+        except Exception as e:
+            raise Exception(f"{e}")
+        finally:
+            if video_path:
+                try:
+                    if os.path.exists(video_path):
+                        os.unlink(video_path)
+                except:
+                    pass
+

nodes/video_nodes/replace_video_background_node.py

@@ -0,0 +1,152 @@
+import os
+import uuid
+import requests
+from ..common import (
+    bria_json_headers,
+    normalize_images_input,
+    poll_status_until_completed,
+    upload_pil_image_to_temp
+)
+from .video_utils import upload_video_to_s3
+
+
+class ReplaceVideoBackgroundNode():
+    """
+    Composites a new background (image or video URL, or an IMAGE from another node) behind the
+    foreground video using the Bria API (POST /v2/video/edit/replace_background).
+
+    When ``background_image`` is connected, only the first image is used (no batch); it is uploaded
+    via the platform anonymous image presigned URL (same pattern as video) and the resulting
+    ``https://temp.bria.ai/...`` URL is sent in ``background_url``.
+
+    The background asset must match the foreground aspect ratio; otherwise the API may return
+    BACKGROUND_ASPECT_RATIO_MISMATCH (surfaced with foreground and background aspect ratio values).
+    """
+    @classmethod
+    def INPUT_TYPES(self):
+        return {
+            "required": {
+                "api_key": ("STRING", {"default": "BRIA_API_TOKEN"}),
+                "video_url": ("STRING", {
+                    "default": "",
+                    "tooltip": "Local path or publicly accessible URL of the foreground video.",
+                }),
+            },
+            "optional": {
+                "background_url": ("STRING", {
+                    "default": "",
+                    "tooltip": "Public HTTPS image or video URL, if not using background_image.",
+                }),
+                "background_image": ("IMAGE",),
+                "output_container_and_codec": ([
+                    "mp4_h264",
+                    "mp4_h265",
+                    "webm_vp9",
+                    "mov_h265",
+                    "mov_proresks",
+                    "mkv_h264",
+                    "mkv_h265",
+                    "mkv_vp9",
+                    "gif"
+                ], {"default": "mp4_h264"}),
+                "preserve_audio": ("BOOLEAN", {"default": True}),
+            }
+        }
+
+    RETURN_TYPES = ("STRING",)
+    RETURN_NAMES = ("result_video_url",)
+    CATEGORY = "API Nodes"
+    FUNCTION = "execute"
+
+    def __init__(self):
+        self.api_url = "https://engine.prod.bria-api.com/v2/video/edit/replace_background"
+
+    @staticmethod
+    def _background_image_to_temp_url(background_image, api_key):
+        """First image only; upload to temp bucket (format from file_name extension; .png if none)."""
+        if background_image is None:
+            return None
+        try:
+            pil_images = normalize_images_input(background_image)
+        except (ValueError, TypeError) as e:
+            raise Exception(f"Invalid background_image: {e}") from e
+        if not pil_images:
+            raise Exception("background_image produced no images.")
+        file_name = f"{uuid.uuid4()}_background"
+        return upload_pil_image_to_temp(pil_images[0], api_key, file_name=file_name)
+
+    def execute(
+        self,
+        api_key,
+        video_url,
+        background_url="",
+        background_image=None,
+        output_container_and_codec="mp4_h264",
+        preserve_audio=True,
+    ):
+        if api_key.strip() == "" or api_key.strip() == "BRIA_API_TOKEN":
+            raise Exception("Please insert a valid API key.")
+
+        if not video_url or not str(video_url).strip():
+            raise Exception("video_url is required: provide a local path or a publicly accessible video URL.")
+
+        bg_from_image = self._background_image_to_temp_url(background_image, api_key)
+        bg_from_url = str(background_url).strip() if background_url else ""
+
+        if bg_from_image:
+            bg = bg_from_image
+        elif bg_from_url:
+            bg = bg_from_url
+        else:
+            raise Exception(
+                "Provide either background_image (IMAGE from Load Image, Generate Image, etc.) "
+                "or a non-empty background_url (HTTPS image or video URL)."
+            )
+
+        if os.path.exists(video_url):
+            filename = f"{str(uuid.uuid4())}_{os.path.basename(video_url)}"
+            input_video_url = upload_video_to_s3(video_url, filename, api_key)
+            if not input_video_url or not (
+                input_video_url.startswith("http://") or input_video_url.startswith("https://")
+            ):
+                raise Exception(f"Failed to upload video to S3. Got: {input_video_url}")
+        else:
+            input_video_url = video_url.strip()
+
+        try:
+            print("Calling Bria API for video replace background...")
+            payload = {
+                "video": input_video_url,
+                "background_url": bg,
+                "output_container_and_codec": output_container_and_codec,
+                "preserve_audio": preserve_audio,
+            }
+
+            headers = bria_json_headers(api_key)
+
+            response = requests.post(self.api_url, json=payload, headers=headers)
+
+            if response.status_code == 200 or response.status_code == 202:
+                print("Initial video replace-background request accepted, polling for completion...")
+                response_dict = response.json()
+
+                status_url = response_dict.get("status_url")
+                request_id = response_dict.get("request_id")
+
+                if not status_url:
+                    raise Exception("No status_url returned from API")
+
+                print(f"Request ID: {request_id}, Status URL: {status_url}")
+
+                final_response = poll_status_until_completed(
+                    status_url, api_key, timeout=3600, check_interval=5
+                )
+
+                result_video_url = final_response["result"]["video_url"]
+
+                print(f"Video processing completed. Result URL: {result_video_url}")
+                return (result_video_url,)
+
+            raise Exception(f"Error: API request failed with status code {response.status_code} {response.text}")
+        except Exception as e:
+            raise Exception(f"{e}")

nodes/video_nodes/video_erase_elements_node.py

@@ -0,0 +1,123 @@
+import os
+import uuid
+import requests
+import folder_paths
+from ..common import bria_json_headers, poll_status_until_completed
+from .video_utils import upload_video_to_s3
+
+class VideoEraseElementsNode():
+    """
+    Erase elements from a video using the Bria API.
+
+    Parameters:
+        api_key (str): Your Bria API key.
+        video_url (str): Local path or URL of the video to process.
+        mask_url (str, optional): URL of a mask video for selective erasing.
+        output_container_and_codec (str, optional): Desired output format and codec. Default is "mp4_h264".
+        preserve_audio (bool, optional): Whether to keep the audio track. Default is True.
+
+    Returns:
+        result_video_url (STRING): URL of the processed video with elements erased.
+    """
+    @classmethod
+    def INPUT_TYPES(self):
+        return {
+            "required": {
+                "api_key": ("STRING", {"default": "BRIA_API_TOKEN"}),
+                "video_url": ("STRING", {
+                    "default": "",
+                    "tooltip": "URL of video to process (provide either frames or video_url)"
+                }),
+            },
+            "optional": { 
+                "mask_url": ("STRING", {
+                    "default": "",
+                    "tooltip": "URL of mask video (optional)"
+                }),
+                "output_container_and_codec": ([
+                    "mp4_h264",
+                    "mp4_h265",
+                    "webm_vp9",
+                    "mov_h265",
+                    "mov_proresks",
+                    "mkv_h264",
+                    "mkv_h265",
+                    "mkv_vp9",
+                    "gif"
+                ], {"default": "mp4_h264"}),
+                "preserve_audio": ("BOOLEAN", {"default": True}),
+            }
+        }
+
+    RETURN_TYPES = ("STRING",)
+    RETURN_NAMES = ("result_video_url",)
+    CATEGORY = "API Nodes"
+    FUNCTION = "execute"
+
+    def __init__(self):
+        self.api_url = "https://engine.prod.bria-api.com/v2/video/edit/erase"
+
+    def execute(self, api_key, video_url, mask_url="", output_container_and_codec="mp4_h264", preserve_audio=True):
+        if api_key.strip() == "" or api_key.strip() == "BRIA_API_TOKEN":
+            raise Exception("Please insert a valid API key.")
+        video_path = None
+        
+        if video_url and video_url.strip() != "":
+            # Check if video_url is a local file path or a URL
+            if os.path.exists(video_url):
+                filename = f"{ str(uuid.uuid4())}_{os.path.basename(video_url)}"
+                input_video_url = upload_video_to_s3(video_url, filename, api_key)
+                
+                if not input_video_url or not (input_video_url.startswith('http://') or input_video_url.startswith('https://')):
+                    raise Exception(f"Failed to upload video to S3. Got: {input_video_url}")                
+                if video_url.startswith(folder_paths.get_temp_directory()):
+                    video_path = None
+            else:
+                input_video_url = video_url
+        
+        try:
+            
+            print("Step 3: Calling Bria API for element erasure...")
+            payload = {
+                "video": input_video_url,
+                "mask": mask_url,
+                "output_container_and_codec": output_container_and_codec,
+                "preserve_audio": preserve_audio
+            }
+
+            headers = bria_json_headers(api_key)
+
+            response = requests.post(self.api_url, json=payload, headers=headers)
+            
+            if response.status_code == 200 or response.status_code == 202:
+                print('Initial Video Erase Elements request successful, polling for completion...')
+                response_dict = response.json()
+
+                status_url = response_dict.get('status_url')
+                request_id = response_dict.get('request_id')
+                
+                if not status_url:
+                    raise Exception("No status_url returned from API")
+                
+                print(f"Request ID: {request_id}, Status URL: {status_url}")
+                
+                final_response = poll_status_until_completed(status_url, api_key, timeout=3600, check_interval=5)
+                
+                result_video_url = final_response['result']['video_url']
+                
+                print(f"Video processing completed. Result URL: {result_video_url}")
+                print(f"Element erasure complete! Use Preview Video URL node to view the result.")
+                
+                return (result_video_url,)
+            else:
+                raise Exception(f"Error: API request failed with status code {response.status_code} {response.text}")
+
+        except Exception as e:
+            raise Exception(f"{e}")
+        finally:
+            if video_path:
+                try:
+                    if os.path.exists(video_path):
+                        os.unlink(video_path)
+                except:
+                    pass

nodes/video_nodes/video_increase_resolution_node.py

@@ -0,0 +1,120 @@
+import os
+import uuid
+import requests
+import folder_paths
+from ..common import bria_json_headers, poll_status_until_completed
+from .video_utils import  upload_video_to_s3
+
+class VideoIncreaseResolutionNode():
+    """
+    Increase the resolution of a video using the Bria API.
+
+    Parameters:
+        api_key (str): Your Bria API key.
+        video_url (str): Local path or URL of the video to process.
+        desired_increase (str, optional): Resolution increase factor, '2' or '4'. Default is '2'.
+        output_container_and_codec (str, optional): Desired output format and codec. Default is "mp4_h264".
+        preserve_audio (bool, optional): Whether to keep the audio track. Default is True.
+
+    Returns:
+        result_video_url (STRING): URL of the processed video with increased resolution.
+    """
+    @classmethod
+    def INPUT_TYPES(self):
+        return {
+            "required": {
+                "api_key": ("STRING", {"default": "BRIA_API_TOKEN"}),
+                "video_url": ("STRING", {
+                    "default": "",
+                    "tooltip": "URL of video to process (provide either frames or video_url)"
+                }),
+            },
+            "optional": {
+                "desired_increase": (['2', '4'], {"default": '2'}),
+                "output_container_and_codec": ([
+                    "mp4_h264",
+                    "mp4_h265",
+                    "webm_vp9",
+                    "mov_h265",
+                    "mov_proresks",
+                    "mkv_h264",
+                    "mkv_h265",
+                    "mkv_vp9",
+                    "gif"
+                ], {"default": "mp4_h264"}),
+                "preserve_audio": ("BOOLEAN", {"default": True}),
+            }
+        }
+
+    RETURN_TYPES = ("STRING",)
+    RETURN_NAMES = ("result_video_url",)
+    CATEGORY = "API Nodes"
+    FUNCTION = "execute"
+
+    def __init__(self):
+        self.api_url = "https://engine.prod.bria-api.com/v2/video/edit/increase_resolution"
+
+    def execute(self, api_key, video_url, desired_increase='2', output_container_and_codec="mp4_h264", preserve_audio=True):
+        if api_key.strip() == "" or api_key.strip() == "BRIA_API_TOKEN":
+            raise Exception("Please insert a valid API key.")
+        video_path = None
+        
+        if video_url and video_url.strip() != "":
+            if os.path.exists(video_url):
+                filename = f"{ str(uuid.uuid4())}_{os.path.basename(video_url)}"
+                input_video_url = upload_video_to_s3(video_url, filename, api_key)
+                
+                if not input_video_url or not (input_video_url.startswith('http://') or input_video_url.startswith('https://')):
+                    raise Exception(f"Failed to upload video to S3. Got: {input_video_url}")
+                
+                if video_url.startswith(folder_paths.get_temp_directory()):
+                    video_path = None
+            else:
+                input_video_url = video_url
+        
+        try:
+            
+            print("Step 3: Calling Bria API for resolution increase...")
+            payload = {
+                "video": input_video_url,
+                "desired_increase": desired_increase,
+                "output_container_and_codec": output_container_and_codec,
+                "preserve_audio": preserve_audio
+            }
+
+            headers = bria_json_headers(api_key)
+
+            response = requests.post(self.api_url, json=payload, headers=headers)
+            
+            if response.status_code == 200 or response.status_code == 202:
+                print('Initial Video Increase Resolution request successful, polling for completion...')
+                response_dict = response.json()
+
+                status_url = response_dict.get('status_url')
+                request_id = response_dict.get('request_id')
+                
+                if not status_url:
+                    raise Exception("No status_url returned from API")
+                
+                print(f"Request ID: {request_id}, Status URL: {status_url}")
+                
+                final_response = poll_status_until_completed(status_url, api_key, timeout=3600, check_interval=5)
+                
+                result_video_url = final_response['result']['video_url']
+                
+                print(f"Video processing completed. Result URL: {result_video_url}")
+                print(f"Resolution increase complete! Use Preview Video URL node to view the result.")
+                
+                return (result_video_url,)
+            else:
+                raise Exception(f"Error: API request failed with status code {response.status_code} {response.text}")
+
+        except Exception as e:
+            raise Exception(f"{e}")
+        finally:
+            if video_path:
+                try:
+                    if os.path.exists(video_path):
+                        os.unlink(video_path)
+                except:
+                    pass

nodes/video_nodes/video_mask_by_key_points_node.py

@@ -0,0 +1,127 @@
+import os
+import uuid
+import requests
+import folder_paths
+from ..common import bria_json_headers, poll_status_until_completed
+from .video_utils import upload_video_to_s3
+import json
+
+class VideoMaskByKeyPointsNode():
+    """
+    Generate a video mask using key points with the Bria API.
+
+    Parameters:
+        key_points (str): JSON string of key points for masking.
+        api_key (str): Your Bria API key.
+        video_url (str): Local path or URL of the video to process.
+        output_container_and_codec (str, optional): Desired output format and codec. Default is "mp4_h264".
+        preserve_audio (bool, optional): Whether to keep the audio track. Default is True.
+
+    Returns:
+        mask_url (STRING): URL of the generated video mask.
+    """
+    @classmethod
+    def INPUT_TYPES(self):
+        return {
+            "required": {
+                "key_points": ("STRING", {"default": "[]", "multiline": True}),
+                "api_key": ("STRING", {"default": "BRIA_API_TOKEN"}),
+                "video_url": ("STRING", {
+                    "default": "",
+                    "tooltip": "URL of video to process (provide either frames or video_url)"
+                }),
+            },
+            "optional": {
+                "output_container_and_codec": ([
+                    "mp4_h264",
+                    "mp4_h265",
+                    "webm_vp9",
+                    "mov_h265",
+                    "mov_proresks",
+                    "mkv_h264",
+                    "mkv_h265",
+                    "mkv_vp9",
+                    "gif"
+                ], {"default": "mp4_h264"}),
+                "preserve_audio": ("BOOLEAN", {"default": True}),
+            }
+        }
+
+    RETURN_TYPES = ("STRING",)
+    RETURN_NAMES = ("mask_url",)
+    CATEGORY = "API Nodes"
+    FUNCTION = "execute"
+
+    def __init__(self):
+        self.api_url = "https://engine.prod.bria-api.com/v2/video/segment/mask_by_key_points"
+
+    def execute(self, key_points, api_key, video_url, output_container_and_codec="mp4_h264", preserve_audio=True):
+        if api_key.strip() == "" or api_key.strip() == "BRIA_API_TOKEN":
+            raise Exception("Please insert a valid API key.")
+        try:
+            key_points_array = json.loads(key_points)
+        except json.JSONDecodeError as e:
+            raise Exception(f"Invalid JSON format for key_points: {e}")
+
+        video_path = None
+        
+        if video_url and video_url.strip() != "":
+            if os.path.exists(video_url):
+                filename = f"{ str(uuid.uuid4())}_{os.path.basename(video_url)}"
+                input_video_url = upload_video_to_s3(video_url, filename, api_key)
+                
+                if not input_video_url or not (input_video_url.startswith('http://') or input_video_url.startswith('https://')):
+                    raise Exception(f"Failed to upload video to S3. Got: {input_video_url}")
+                
+                
+                if video_url.startswith(folder_paths.get_temp_directory()):
+                    video_path = None
+            else:
+                input_video_url = video_url
+        
+        try:
+            
+            print("Step 3: Calling Bria API for video mask generation by key points...")
+            payload = {
+                "video": input_video_url,
+                "key_points": key_points_array,
+                "output_container_and_codec": output_container_and_codec,
+                "preserve_audio": preserve_audio
+            }
+
+            headers = bria_json_headers(api_key)
+
+            response = requests.post(self.api_url, json=payload, headers=headers)
+            
+            if response.status_code == 200 or response.status_code == 202:
+                print('Initial Video Mask by Key Points request successful, polling for completion...')
+                response_dict = response.json()
+
+                status_url = response_dict.get('status_url')
+                request_id = response_dict.get('request_id')
+                
+                if not status_url:
+                    raise Exception("No status_url returned from API")
+                
+                print(f"Request ID: {request_id}, Status URL: {status_url}")
+                
+                final_response = poll_status_until_completed(status_url, api_key, timeout=3600, check_interval=5)
+                
+                result_mask_url = final_response['result']['mask_url']
+                
+                print(f"Video mask processing completed. Result URL: {result_mask_url}")
+                print(f"Video mask generation complete! Use Preview Video URL node to view the result.")
+                
+                return (result_mask_url,)
+            else:
+                raise Exception(f"Error: API request failed with status code {response.status_code} {response.text}")
+
+        except Exception as e:
+            raise Exception(f"{e}")
+        finally:
+            if video_path:
+                try:
+                    if os.path.exists(video_path):
+                        os.unlink(video_path)
+                except:
+                    pass

nodes/video_nodes/video_mask_by_prompt_node.py

@@ -0,0 +1,121 @@
+import os
+import uuid
+import requests
+import folder_paths
+from ..common import bria_json_headers, poll_status_until_completed
+from .video_utils import upload_video_to_s3
+
+class VideoMaskByPromptNode():
+    """
+    Generate a video mask using a text prompt with the Bria API.
+
+    Parameters:
+        prompt (str): Text prompt describing what to mask in the video.
+        api_key (str): Your Bria API key.
+        video_url (str): Local path or URL of the video to process.
+        output_container_and_codec (str, optional): Desired output format and codec. Default is "mp4_h264".
+        preserve_audio (bool, optional): Whether to keep the audio track. Default is True.
+
+    Returns:
+        mask_url (STRING): URL of the generated video mask.
+    """
+    @classmethod
+    def INPUT_TYPES(self):
+        return {
+            "required": {
+                "prompt": ("STRING", {"default": ""}),
+                "api_key": ("STRING", {"default": "BRIA_API_TOKEN"}),
+                "video_url": ("STRING", {
+                    "default": "",
+                    "tooltip": "URL of video to process (provide either frames or video_url)"
+                }),
+            },
+            "optional": {
+                "output_container_and_codec": ([
+                    "mp4_h264",
+                    "mp4_h265",
+                    "webm_vp9",
+                    "mov_h265",
+                    "mov_proresks",
+                    "mkv_h264",
+                    "mkv_h265",
+                    "mkv_vp9",
+                    "gif"
+                ], {"default": "mp4_h264"}),
+                "preserve_audio": ("BOOLEAN", {"default": True}),
+            }
+        }
+
+    RETURN_TYPES = ("STRING",)
+    RETURN_NAMES = ("mask_url",)
+    CATEGORY = "API Nodes"
+    FUNCTION = "execute"
+
+    def __init__(self):
+        self.api_url = "https://engine.prod.bria-api.com/v2/video/segment/mask_by_prompt"
+
+    def execute(self, prompt, api_key, video_url, output_container_and_codec="mp4_h264", preserve_audio=True):
+        if api_key.strip() == "" or api_key.strip() == "BRIA_API_TOKEN":
+            raise Exception("Please insert a valid API key.")
+        video_path = None
+        
+        if video_url and video_url.strip() != "":
+            if os.path.exists(video_url):
+                filename = f"{ str(uuid.uuid4())}_{os.path.basename(video_url)}"
+                input_video_url = upload_video_to_s3(video_url, filename, api_key)
+                
+                if not input_video_url or not (input_video_url.startswith('http://') or input_video_url.startswith('https://')):
+                    raise Exception(f"Failed to upload video to S3. Got: {input_video_url}")
+                
+                
+                if video_url.startswith(folder_paths.get_temp_directory()):
+                    video_path = None
+            else:
+                input_video_url = video_url
+        
+        try:
+            
+            print("Step 3: Calling Bria API for video mask generation...")
+            payload = {
+                "video": input_video_url,
+                "prompt": prompt,
+                "output_container_and_codec": output_container_and_codec,
+                "preserve_audio": preserve_audio
+            }
+
+            headers = bria_json_headers(api_key)
+
+            response = requests.post(self.api_url, json=payload, headers=headers)
+            
+            if response.status_code == 200 or response.status_code == 202:
+                print('Initial Video Mask by Prompt request successful, polling for completion...')
+                response_dict = response.json()
+
+                status_url = response_dict.get('status_url')
+                request_id = response_dict.get('request_id')
+                
+                if not status_url:
+                    raise Exception("No status_url returned from API")
+                
+                print(f"Request ID: {request_id}, Status URL: {status_url}")
+                
+                final_response = poll_status_until_completed(status_url, api_key, timeout=3600, check_interval=5)
+                
+                result_mask_url = final_response['result']['mask_url']
+                
+                print(f"Video mask processing completed. Result URL: {result_mask_url}")
+                print(f"Video mask generation complete! Use Preview Video URL node to view the result.")
+                
+                return (result_mask_url,)
+            else:
+                raise Exception(f"Error: API request failed with status code {response.status_code} {response.text}")
+
+        except Exception as e:
+            raise Exception(f"{e}")
+        finally:
+            if video_path:
+                try:
+                    if os.path.exists(video_path):
+                        os.unlink(video_path)
+                except:
+                    pass

nodes/video_nodes/video_solid_color_background_node.py

@@ -0,0 +1,133 @@
+import os
+import uuid
+import requests
+import folder_paths
+from ..common import bria_json_headers, poll_status_until_completed
+from .video_utils import upload_video_to_s3
+
+class VideoSolidColorBackgroundNode():
+    """
+    Apply a solid color background to a video using the Bria API.
+
+    Parameters:
+        api_key (str): Your Bria API key.
+        video_url (str): Local path or URL of the video to process.
+        background_color (str, optional): Color to apply as background. Default is "Transparent".
+        output_container_and_codec (str, optional): Desired output format and codec. Default is "mp4_h264".
+        preserve_audio (bool, optional): Whether to keep the audio track. Default is True.
+
+    Returns:
+        result_video_url (STRING): URL of the video with the solid color background applied.
+    """
+    @classmethod
+    def INPUT_TYPES(self):
+        return {
+            "required": {
+                "api_key": ("STRING", {"default": "BRIA_API_TOKEN"}),
+                "video_url": ("STRING", {
+                    "default": "",
+                    "tooltip": "URL of video to process (provide either frames or video_url)"
+                }),
+            },
+            "optional": {
+                "background_color": ([
+                    "Transparent",
+                    "Black",
+                    "White",
+                    "Gray",
+                    "Red",
+                    "Green",
+                    "Blue",
+                    "Yellow",
+                    "Cyan",
+                    "Magenta",
+                    "Orange"
+                ], {"default": "Transparent"}),
+                "output_container_and_codec": ([
+                    "mp4_h264",
+                    "mp4_h265",
+                    "webm_vp9",
+                    "mov_h265",
+                    "mov_proresks",
+                    "mkv_h264",
+                    "mkv_h265",
+                    "mkv_vp9",
+                    "gif"
+                ], {"default": "webm_vp9"}),
+                "preserve_audio": ("BOOLEAN", {"default": True}),   
+            }
+        }
+
+    RETURN_TYPES = ("STRING",)
+    RETURN_NAMES = ("result_video_url",)
+    CATEGORY = "API Nodes"
+    FUNCTION = "execute"
+
+    def __init__(self):
+        self.api_url = "https://engine.prod.bria-api.com/v2/video/edit/remove_background"
+
+    def execute(self, api_key, video_url, background_color="Transparent", output_container_and_codec="webm_vp9", preserve_audio=True):
+        if api_key.strip() == "" or api_key.strip() == "BRIA_API_TOKEN":
+            raise Exception("Please insert a valid API key.")
+        video_path = None
+        
+        if video_url and video_url.strip() != "":
+            if os.path.exists(video_url):
+                filename = f"{ str(uuid.uuid4())}_{os.path.basename(video_url)}"
+                input_video_url = upload_video_to_s3(video_url, filename, api_key)
+                
+                if not input_video_url or not (input_video_url.startswith('http://') or input_video_url.startswith('https://')):
+                    raise Exception(f"Failed to upload video to S3. Got: {input_video_url}")
+                
+                
+                if video_url.startswith(folder_paths.get_temp_directory()):
+                    video_path = None
+            else:
+                input_video_url = video_url
+        
+        try:
+            
+            print("Step 3: Calling Bria API for solid color background...")
+            payload = {
+                "video": input_video_url,
+                "background_color": background_color,
+                "output_container_and_codec": output_container_and_codec,
+                "preserve_audio": preserve_audio
+            }
+
+            headers = bria_json_headers(api_key)
+
+            response = requests.post(self.api_url, json=payload, headers=headers)
+            
+            if response.status_code == 200 or response.status_code == 202:
+                print('Initial Video Solid Color Background request successful, polling for completion...')
+                response_dict = response.json()
+
+                status_url = response_dict.get('status_url')
+                request_id = response_dict.get('request_id')
+                
+                if not status_url:
+                    raise Exception("No status_url returned from API")
+                
+                print(f"Request ID: {request_id}, Status URL: {status_url}")
+                
+                final_response = poll_status_until_completed(status_url, api_key, timeout=3600, check_interval=5)
+                
+                result_video_url = final_response['result']['video_url']
+                
+                print(f"Video processing completed. Result URL: {result_video_url}")
+                print(f"Solid color background processing complete! Use Preview Video URL node to view the result.")
+                
+                return (result_video_url,)
+            else:
+                raise Exception(f"Error: API request failed with status code {response.status_code} {response.text}")
+
+        except Exception as e:
+            raise Exception(f"{e}")
+        finally:
+            if video_path:
+                try:
+                    if os.path.exists(video_path):
+                        os.unlink(video_path)
+                except:
+                    pass

nodes/video_nodes/video_utils.py

@@ -0,0 +1,72 @@
+import os
+import requests
+
+from ..common import BRIA_COMFYUI_USER_AGENT
+
+
+def upload_video_to_s3(video_path, filename, api_token):
+    api_url = "https://platform.prod.bria-api.com/upload-video/anonymous/presigned-url"
+    headers = {
+        "Content-Type": "application/json",
+        "User-Agent": BRIA_COMFYUI_USER_AGENT,
+    }
+    extension = os.path.splitext(filename)[1].lower()
+    content_type_map = {
+            '.mp4': 'video/mp4',
+            '.webm': 'video/webm',
+            '.mov': 'video/quicktime',
+            '.mkv': 'video/x-matroska',
+            '.avi': 'video/x-msvideo',
+            '.gif': 'image/gif',
+            '.webp': 'image/webp'
+    }
+    content_type = content_type_map.get(extension, 'video/mp4')
+    if api_token:
+        headers["api_token"] = api_token
+    
+    payload = {
+        "file_name": filename,
+        "content_type":content_type
+    }
+    
+    print(f"Requesting presigned URL for: {filename}")
+    
+    try:
+        response = requests.post(api_url, json=payload, headers=headers)
+        
+        if response.status_code != 200:
+            raise Exception(f"Failed to get presigned URL: {response.status_code} {response.text}")
+        
+        response_data = response.json()
+        video_url = response_data.get("video_url")
+        upload_url = response_data.get("upload_url")
+        
+        if not video_url or not upload_url:
+            raise Exception(f"Invalid response from presigned URL API: {response_data}")
+        
+        print(f"Received presigned URL")
+        print(f"Video URL: {video_url}")
+        
+        # Step 2: Upload video to presigned URL
+        print(f"Uploading video to S3...")
+        
+        with open(video_path, 'rb') as f:
+            video_data = f.read()
+        
+        # Determine content type based on file extension
+        upload_headers = {
+            "Content-Type": content_type
+        }
+
+        upload_response = requests.put(upload_url, data=video_data, headers=upload_headers)
+        
+        if upload_response.status_code not in [200, 204]:
+            raise Exception(f"Failed to upload video to S3: {upload_response.status_code}")
+        
+        print(f"Video uploaded successfully to S3")
+        
+        return video_url
+        
+    except Exception as e:
+        raise Exception(f"Error uploading video to S3: {str(e)}")
+

pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "comfyui-bria-api"
 description = "Custom nodes for ComfyUI using BRIA's API."
-version = "2.1.8"
+version = "2.1.19"
 license = {file = "LICENSE"}
 
 [project.urls]

web/select_images_from_folder.js

@@ -0,0 +1,145 @@
+import { app } from "/scripts/app.js";
+import { api } from "/scripts/api.js";
+
+app.registerExtension({
+  name: "BriaMultiImageSelect",
+
+  async nodeCreated(node) {
+    if (node.comfyClass !== "BriaMultiImageSelect") return;
+
+    const getWidget = (name) =>
+      node.widgets?.find(w => w.name === name);
+
+    const pathsWidget = getWidget("selected_paths");
+    if (!pathsWidget) return;
+
+    pathsWidget.hidden = true;
+    pathsWidget.draw = () => {};
+    pathsWidget.computeSize = () => [0, 0];
+
+    const viewURLFromRel = (rel) => {
+      const parts = (rel || "").split("/");
+      const filename = parts.pop();
+      const subfolder = parts.join("/");
+      const params = new URLSearchParams({ filename, type: "input", subfolder });
+      return api.apiURL(`/view?${params.toString()}`);
+    };
+
+    const loadImg = (url) =>
+      new Promise((resolve, reject) => {
+        const img = new Image();
+        img.crossOrigin = "anonymous";
+        img.onload = () => resolve(img);
+        img.onerror = () => reject();
+        img.src = url;
+      });
+
+    const refreshPreview = async () => {
+      let raw = node.properties.selected_paths;
+      if (!raw) raw = pathsWidget.value;
+      
+      let rels = [];
+      try {
+        rels = raw ? JSON.parse(raw) : [];
+      } catch (e) {
+        console.error("Failed to parse selected_paths:", e);
+        node.imgs = null;
+        node.imgError = "Invalid image data";
+        node.setDirtyCanvas(true, true);
+        return;
+      }
+
+      if (!rels.length) {
+        node.imgs = null;
+        node.imgError = "No images selected";
+        node.setDirtyCanvas(true, true);
+        return;
+      }
+
+      const imgs = [];
+      await Promise.allSettled(
+        rels.map(async (rel) => {
+          try {
+            const img = await loadImg(viewURLFromRel(rel));
+            imgs.push(img);
+          } catch (e) {
+            console.warn(`Failed to load image: ${rel}`, e);
+          }
+        })
+      );
+
+      if (imgs.length) {
+        node.imgs = imgs;
+        node.imgError = null;
+      } else {
+        node.imgs = null;
+        node.imgError = "Failed to load images";
+      }
+      node.setDirtyCanvas(true, true);
+    };
+    // Update node size to accommodate preview
+    const originalComputeSize = node.computeSize;
+    node.computeSize = function () {
+      const size = originalComputeSize ? originalComputeSize.apply(this, arguments) : [200, 100];
+      size[1] = Math.max(size[1], 200); // Ensure minimum height for preview
+      return size;
+    };
+
+    const btn = node.addWidget("button", "Select Images", null, async () => {
+      const input = document.createElement("input");
+      input.type = "file";
+      input.multiple = true;
+      input.accept = "image/*";
+      input.style.display = "none";
+      document.body.appendChild(input);
+
+      input.onchange = async () => {
+        const files = Array.from(input.files || []);
+        document.body.removeChild(input);
+        if (!files.length) return;
+
+        const rels = [];
+
+        for (const f of files) {
+          const form = new FormData();
+          form.append("image", f, f.name);
+
+          const resp = await api.fetchApi("/upload/image", {
+            method: "POST",
+            body: form,
+          });
+          if (!resp.ok) continue;
+
+          const data = await resp.json();
+          const rel = data.subfolder
+            ? `${data.subfolder}/${data.name}`
+            : data.name;
+          rels.push(rel);
+        }
+
+        const json = JSON.stringify(rels);
+        pathsWidget.value = json;
+        node.properties.selected_paths = json;
+
+        await refreshPreview();
+      };
+
+      input.click();
+    });
+
+    node.widgets.unshift(
+      node.widgets.splice(node.widgets.indexOf(btn), 1)[0]
+    );
+
+    // Initialize properties if not present
+    if (!node.properties) {
+      node.properties = {};
+    }
+
+    await refreshPreview();
+
+    setTimeout(async () => {
+      await refreshPreview();
+    }, 100);
+  },
+});

workflows/Bria Video RMBG workflow.json

@@ -0,0 +1,150 @@
+{
+  "id": "faacd69e-30b7-4ae8-a9e0-11e3523139f9",
+  "revision": 0,
+  "last_node_id": 14,
+  "last_link_id": 9,
+  "nodes": [
+    {
+      "id": 5,
+      "type": "LoadVideoFramesNode",
+      "pos": [
+        -377.4424627503264,
+        265.04224992480954
+      ],
+      "size": [
+        270,
+        276.890625
+      ],
+      "flags": {},
+      "order": 0,
+      "mode": 0,
+      "inputs": [],
+      "outputs": [
+        {
+          "name": "video_path",
+          "type": "STRING",
+          "links": [
+            3
+          ]
+        }
+      ],
+      "properties": {
+        "Node name for S&R": "LoadVideoFramesNode"
+      },
+      "widgets_values": [
+        "6952253-uhd_3840_2160_25fps.mp4",
+        "image"
+      ]
+    },
+    {
+      "id": 7,
+      "type": "RemoveVideoBackgroundNode",
+      "pos": [
+        176.90761152601942,
+        244.1153564319173
+      ],
+      "size": [
+        550.953125,
+        348.203125
+      ],
+      "flags": {},
+      "order": 1,
+      "mode": 0,
+      "inputs": [
+        {
+          "name": "video_url",
+          "type": "STRING",
+          "widget": {
+            "name": "video_url"
+          },
+          "link": 3
+        }
+      ],
+      "outputs": [
+        {
+          "name": "result_video_url",
+          "type": "STRING",
+          "links": [
+            4
+          ]
+        }
+      ],
+      "properties": {
+        "Node name for S&R": "RemoveVideoBackgroundNode"
+      },
+      "widgets_values": [
+        "",
+        "",
+        true,
+        "webm_vp9"
+      ]
+    },
+    {
+      "id": 8,
+      "type": "PreviewVideoURLNode",
+      "pos": [
+        1020.6454228509147,
+        264.9389174606007
+      ],
+      "size": [
+        270,
+        252.890625
+      ],
+      "flags": {},
+      "order": 2,
+      "mode": 0,
+      "inputs": [
+        {
+          "name": "video_url",
+          "type": "STRING",
+          "widget": {
+            "name": "video_url"
+          },
+          "link": 4
+        }
+      ],
+      "outputs": [],
+      "properties": {
+        "Node name for S&R": "PreviewVideoURLNode"
+      },
+      "widgets_values": [
+        ""
+      ]
+    }
+  ],
+  "links": [
+    [
+      3,
+      5,
+      0,
+      7,
+      0,
+      "STRING"
+    ],
+    [
+      4,
+      7,
+      0,
+      8,
+      0,
+      "STRING"
+    ]
+  ],
+  "groups": [],
+  "config": {},
+  "extra": {
+    "ds": {
+      "scale": 0.43094644375454877,
+      "offset": [
+        685.6754838043107,
+        494.0393339664482
+      ]
+    },
+    "frontendVersion": "1.43.18",
+    "VHS_latentpreview": false,
+    "VHS_latentpreviewrate": 0,
+    "VHS_MetadataImage": true,
+    "VHS_KeepIntermediate": true
+  },
+  "version": 0.4
+}

workflows/Fibo_edit_workflow.json

@@ -0,0 +1,331 @@
+{
+  "id": "3eb93704-25f0-4511-b147-cf0403c5d060",
+  "revision": 0,
+  "last_node_id": 10,
+  "last_link_id": 11,
+  "nodes": [
+    {
+      "id": 7,
+      "type": "PreviewImage",
+      "pos": [
+        756.945556640625,
+        406.1631774902344
+      ],
+      "size": [
+        140,
+        246
+      ],
+      "flags": {},
+      "order": 5,
+      "mode": 0,
+      "inputs": [
+        {
+          "name": "images",
+          "type": "IMAGE",
+          "link": 9
+        }
+      ],
+      "outputs": [],
+      "properties": {
+        "Node name for S&R": "PreviewImage"
+      },
+      "widgets_values": []
+    },
+    {
+      "id": 6,
+      "type": "PreviewImage",
+      "pos": [
+        702.7532348632812,
+        830.3753662109375
+      ],
+      "size": [
+        140,
+        246
+      ],
+      "flags": {},
+      "order": 3,
+      "mode": 0,
+      "inputs": [
+        {
+          "name": "images",
+          "type": "IMAGE",
+          "link": 5
+        }
+      ],
+      "outputs": [],
+      "properties": {
+        "Node name for S&R": "PreviewImage"
+      },
+      "widgets_values": []
+    },
+    {
+      "id": 8,
+      "type": "FIBOEditNode",
+      "pos": [
+        390.8927307128906,
+        339.14031982421875
+      ],
+      "size": [
+        278.720703125,
+        266
+      ],
+      "flags": {},
+      "order": 4,
+      "mode": 0,
+      "inputs": [
+        {
+          "name": "images",
+          "type": "IMAGE",
+          "link": 8
+        },
+        {
+          "name": "mask",
+          "shape": 7,
+          "type": "MASK",
+          "link": null
+        },
+        {
+          "name": "structured_instruction",
+          "shape": 7,
+          "type": "STRING",
+          "widget": {
+            "name": "structured_instruction"
+          },
+          "link": 11
+        }
+      ],
+      "outputs": [
+        {
+          "name": "IMAGE",
+          "type": "IMAGE",
+          "links": [
+            9
+          ]
+        },
+        {
+          "name": "structured_instruction",
+          "type": "STRING",
+          "links": null
+        },
+        {
+          "name": "seed",
+          "type": "INT",
+          "links": null
+        }
+      ],
+      "properties": {
+        "Node name for S&R": "FIBOEditNode"
+      },
+      "widgets_values": [
+        "",
+        "",
+        "",
+        "",
+        50,
+        5,
+        1199,
+        "randomize"
+      ]
+    },
+    {
+      "id": 4,
+      "type": "FIBOEditNode",
+      "pos": [
+        301.3495788574219,
+        884.5100708007812
+      ],
+      "size": [
+        278.720703125,
+        266
+      ],
+      "flags": {},
+      "order": 1,
+      "mode": 0,
+      "inputs": [
+        {
+          "name": "images",
+          "type": "IMAGE",
+          "link": 4
+        },
+        {
+          "name": "mask",
+          "shape": 7,
+          "type": "MASK",
+          "link": null
+        }
+      ],
+      "outputs": [
+        {
+          "name": "IMAGE",
+          "type": "IMAGE",
+          "links": [
+            5
+          ]
+        },
+        {
+          "name": "structured_instruction",
+          "type": "STRING",
+          "links": null
+        },
+        {
+          "name": "seed",
+          "type": "INT",
+          "links": null
+        }
+      ],
+      "properties": {
+        "Node name for S&R": "FIBOEditNode"
+      },
+      "widgets_values": [
+        "",
+        "change the lamp to a radio",
+        "",
+        "",
+        50,
+        5,
+        55,
+        "randomize"
+      ]
+    },
+    {
+      "id": 2,
+      "type": "LoadImage",
+      "pos": [
+        -245.09060668945312,
+        882.640869140625
+      ],
+      "size": [
+        274.080078125,
+        314
+      ],
+      "flags": {},
+      "order": 0,
+      "mode": 0,
+      "inputs": [],
+      "outputs": [
+        {
+          "name": "IMAGE",
+          "type": "IMAGE",
+          "links": [
+            4,
+            8,
+            10
+          ]
+        },
+        {
+          "name": "MASK",
+          "type": "MASK",
+          "links": null
+        }
+      ],
+      "properties": {
+        "Node name for S&R": "LoadImage"
+      },
+      "widgets_values": [
+        "0a9d91e579872d653daf3243df4598f0 (2).png",
+        "image"
+      ]
+    },
+    {
+      "id": 10,
+      "type": "FIBOEditStructuredInstructionNode",
+      "pos": [
+        -139.52833557128906,
+        410.5189208984375
+      ],
+      "size": [
+        314.6372985839844,
+        82
+      ],
+      "flags": {},
+      "order": 2,
+      "mode": 0,
+      "inputs": [
+        {
+          "name": "images",
+          "type": "IMAGE",
+          "link": 10
+        }
+      ],
+      "outputs": [
+        {
+          "name": "structured_instruction",
+          "type": "STRING",
+          "links": [
+            11
+          ]
+        }
+      ],
+      "properties": {
+        "Node name for S&R": "FIBOEditStructuredInstructionNode"
+      },
+      "widgets_values": [
+        "BRIA_API_TOKEN",
+        ""
+      ]
+    }
+  ],
+  "links": [
+    [
+      4,
+      2,
+      0,
+      4,
+      0,
+      "IMAGE"
+    ],
+    [
+      5,
+      4,
+      0,
+      6,
+      0,
+      "IMAGE"
+    ],
+    [
+      8,
+      2,
+      0,
+      8,
+      0,
+      "IMAGE"
+    ],
+    [
+      9,
+      8,
+      0,
+      7,
+      0,
+      "IMAGE"
+    ],
+    [
+      10,
+      2,
+      0,
+      10,
+      0,
+      "IMAGE"
+    ],
+    [
+      11,
+      10,
+      0,
+      8,
+      2,
+      "STRING"
+    ]
+  ],
+  "groups": [],
+  "config": {},
+  "extra": {
+    "ds": {
+      "scale": 0.7522123482651067,
+      "offset": [
+        684.0553164416729,
+        -258.9658284524182
+      ]
+    },
+    "frontendVersion": "1.25.11"
+  },
+  "version": 0.4
+}

workflows/Video_Editig_Workflow.json

@@ -0,0 +1,495 @@
+{
+  "id": "17df2a89-3a7b-4b17-9ed1-fe236151bd3c",
+  "revision": 0,
+  "last_node_id": 30,
+  "last_link_id": 34,
+  "nodes": [
+    {
+      "id": 23,
+      "type": "PreviewVideoURLNode",
+      "pos": [
+        1146.3548583984375,
+        426.4760437011719
+      ],
+      "size": [
+        270,
+        177.875
+      ],
+      "flags": {},
+      "order": 9,
+      "mode": 0,
+      "inputs": [
+        {
+          "name": "video_url",
+          "type": "STRING",
+          "widget": {
+            "name": "video_url"
+          },
+          "link": 28
+        }
+      ],
+      "outputs": [],
+      "properties": {
+        "Node name for S&R": "PreviewVideoURLNode"
+      },
+      "widgets_values": [
+        ""
+      ]
+    },
+    {
+      "id": 24,
+      "type": "PreviewVideoURLNode",
+      "pos": [
+        285.0629577636719,
+        889.9742431640625
+      ],
+      "size": [
+        270,
+        177.875
+      ],
+      "flags": {},
+      "order": 5,
+      "mode": 0,
+      "inputs": [
+        {
+          "name": "video_url",
+          "type": "STRING",
+          "widget": {
+            "name": "video_url"
+          },
+          "link": 29
+        }
+      ],
+      "outputs": [],
+      "properties": {
+        "Node name for S&R": "PreviewVideoURLNode"
+      },
+      "widgets_values": [
+        ""
+      ]
+    },
+    {
+      "id": 28,
+      "type": "PreviewVideoURLNode",
+      "pos": [
+        560.3937377929688,
+        -124.61785125732422
+      ],
+      "size": [
+        270,
+        177.875
+      ],
+      "flags": {},
+      "order": 7,
+      "mode": 0,
+      "inputs": [
+        {
+          "name": "video_url",
+          "type": "STRING",
+          "widget": {
+            "name": "video_url"
+          },
+          "link": 33
+        }
+      ],
+      "outputs": [],
+      "properties": {
+        "Node name for S&R": "PreviewVideoURLNode"
+      },
+      "widgets_values": [
+        ""
+      ]
+    },
+    {
+      "id": 29,
+      "type": "PreviewVideoURLNode",
+      "pos": [
+        484.5365295410156,
+        -391.69635009765625
+      ],
+      "size": [
+        270,
+        177.875
+      ],
+      "flags": {},
+      "order": 8,
+      "mode": 0,
+      "inputs": [
+        {
+          "name": "video_url",
+          "type": "STRING",
+          "widget": {
+            "name": "video_url"
+          },
+          "link": 34
+        }
+      ],
+      "outputs": [],
+      "properties": {
+        "Node name for S&R": "PreviewVideoURLNode"
+      },
+      "widgets_values": [
+        ""
+      ]
+    },
+    {
+      "id": 19,
+      "type": "LoadVideoFramesNode",
+      "pos": [
+        -898.2652587890625,
+        20.335519790649414
+      ],
+      "size": [
+        270,
+        554
+      ],
+      "flags": {},
+      "order": 0,
+      "mode": 0,
+      "inputs": [],
+      "outputs": [
+        {
+          "name": "video_path",
+          "type": "STRING",
+          "links": [
+            24,
+            25,
+            26,
+            31,
+            32
+          ]
+        }
+      ],
+      "properties": {
+        "Node name for S&R": "LoadVideoFramesNode"
+      },
+      "widgets_values": [
+        "plane.mp4",
+        "image"
+      ]
+    },
+    {
+      "id": 27,
+      "type": "VideoIncreaseResolutionNode",
+      "pos": [
+        -83.17711639404297,
+        -394.345458984375
+      ],
+      "size": [
+        318.79296875,
+        154
+      ],
+      "flags": {},
+      "order": 4,
+      "mode": 0,
+      "inputs": [
+        {
+          "name": "video_url",
+          "type": "STRING",
+          "widget": {
+            "name": "video_url"
+          },
+          "link": 32
+        }
+      ],
+      "outputs": [
+        {
+          "name": "result_video_url",
+          "type": "STRING",
+          "links": [
+            34
+          ]
+        }
+      ],
+      "properties": {
+        "Node name for S&R": "VideoIncreaseResolutionNode"
+      },
+      "widgets_values": [
+        "BRIA_API_TOKEN",
+        "",
+        "2",
+        "mp4_h264",
+        true
+      ]
+    },
+    {
+      "id": 26,
+      "type": "RemoveVideoBackgroundNode",
+      "pos": [
+        22.839176177978516,
+        -98.48242950439453
+      ],
+      "size": [
+        318.79296875,
+        130
+      ],
+      "flags": {},
+      "order": 3,
+      "mode": 0,
+      "inputs": [
+        {
+          "name": "video_url",
+          "type": "STRING",
+          "widget": {
+            "name": "video_url"
+          },
+          "link": 31
+        }
+      ],
+      "outputs": [
+        {
+          "name": "result_video_url",
+          "type": "STRING",
+          "links": [
+            33
+          ]
+        }
+      ],
+      "properties": {
+        "Node name for S&R": "RemoveVideoBackgroundNode"
+      },
+      "widgets_values": [
+        "BRIA_API_TOKEN",
+        "",
+        true,
+        "webm_vp9"
+      ]
+    },
+    {
+      "id": 21,
+      "type": "VideoMaskByPromptNode",
+      "pos": [
+        53.91518783569336,
+        377.6769714355469
+      ],
+      "size": [
+        318.79296875,
+        154
+      ],
+      "flags": {},
+      "order": 2,
+      "mode": 0,
+      "inputs": [
+        {
+          "name": "video_url",
+          "type": "STRING",
+          "widget": {
+            "name": "video_url"
+          },
+          "link": 25
+        }
+      ],
+      "outputs": [
+        {
+          "name": "mask_url",
+          "type": "STRING",
+          "links": [
+            27
+          ]
+        }
+      ],
+      "properties": {
+        "Node name for S&R": "VideoMaskByPromptNode"
+      },
+      "widgets_values": [
+        "airplane",
+        "BRIA_API_TOKEN",
+        "",
+        "mp4_h264",
+        true
+      ]
+    },
+    {
+      "id": 22,
+      "type": "VideoEraseElementsNode",
+      "pos": [
+        658.7935180664062,
+        377.9380187988281
+      ],
+      "size": [
+        318.79296875,
+        154
+      ],
+      "flags": {},
+      "order": 6,
+      "mode": 0,
+      "inputs": [
+        {
+          "name": "video_url",
+          "type": "STRING",
+          "widget": {
+            "name": "video_url"
+          },
+          "link": 26
+        },
+        {
+          "name": "mask_url",
+          "shape": 7,
+          "type": "STRING",
+          "widget": {
+            "name": "mask_url"
+          },
+          "link": 27
+        }
+      ],
+      "outputs": [
+        {
+          "name": "result_video_url",
+          "type": "STRING",
+          "links": [
+            28
+          ]
+        }
+      ],
+      "properties": {
+        "Node name for S&R": "VideoEraseElementsNode"
+      },
+      "widgets_values": [
+        "BRIA_API_TOKEN",
+        "",
+        "",
+        "mp4_h264",
+        true
+      ]
+    },
+    {
+      "id": 20,
+      "type": "VideoSolidColorBackgroundNode",
+      "pos": [
+        -133.50149536132812,
+        856.9811401367188
+      ],
+      "size": [
+        318.79296875,
+        154
+      ],
+      "flags": {},
+      "order": 1,
+      "mode": 0,
+      "inputs": [
+        {
+          "name": "video_url",
+          "type": "STRING",
+          "widget": {
+            "name": "video_url"
+          },
+          "link": 24
+        }
+      ],
+      "outputs": [
+        {
+          "name": "result_video_url",
+          "type": "STRING",
+          "links": [
+            29
+          ]
+        }
+      ],
+      "properties": {
+        "Node name for S&R": "VideoSolidColorBackgroundNode"
+      },
+      "widgets_values": [
+        "BRIA_API_TOKEN",
+        "",
+        "Transparent",
+        "webm_vp9",
+        true
+      ]
+    }
+  ],
+  "links": [
+    [
+      24,
+      19,
+      0,
+      20,
+      0,
+      "STRING"
+    ],
+    [
+      25,
+      19,
+      0,
+      21,
+      0,
+      "STRING"
+    ],
+    [
+      26,
+      19,
+      0,
+      22,
+      0,
+      "STRING"
+    ],
+    [
+      27,
+      21,
+      0,
+      22,
+      1,
+      "STRING"
+    ],
+    [
+      28,
+      22,
+      0,
+      23,
+      0,
+      "STRING"
+    ],
+    [
+      29,
+      20,
+      0,
+      24,
+      0,
+      "STRING"
+    ],
+    [
+      31,
+      19,
+      0,
+      26,
+      0,
+      "STRING"
+    ],
+    [
+      32,
+      19,
+      0,
+      27,
+      0,
+      "STRING"
+    ],
+    [
+      33,
+      26,
+      0,
+      28,
+      0,
+      "STRING"
+    ],
+    [
+      34,
+      27,
+      0,
+      29,
+      0,
+      "STRING"
+    ]
+  ],
+  "groups": [],
+  "config": {},
+  "extra": {
+    "ds": {
+      "scale": 0.5644739300537782,
+      "offset": [
+        1311.5448975965508,
+        244.532748842859
+      ]
+    },
+    "frontendVersion": "1.25.11"
+  },
+  "version": 0.4
+}

workflows/background_generation_workflow.json

@@ -1 +1,491 @@
-{"last_node_id":39,"last_link_id":65,"nodes":[{"id":34,"type":"LoadImage","pos":[19.94045066833496,1075.806640625],"size":[315,314],"flags":{},"order":0,"mode":0,"inputs":[],"outputs":[{"name":"IMAGE","type":"IMAGE","links":[61],"slot_index":0,"localized_name":"IMAGE"},{"name":"MASK","type":"MASK","links":null,"localized_name":"MASK"}],"properties":{"Node name for S&R":"LoadImage"},"widgets_values":["quirky-red-brick-brick-wallpaper.jpg","image"]},{"id":36,"type":"PreviewImage","pos":[468.7108154296875,1210.288818359375],"size":[210,246],"flags":{},"order":5,"mode":0,"inputs":[{"name":"images","type":"IMAGE","link":63,"localized_name":"images"}],"outputs":[],"properties":{"Node name for S&R":"PreviewImage"},"widgets_values":[]},{"id":33,"type":"PreviewImage","pos":[1317.4083251953125,1118.4864501953125],"size":[210,246],"flags":{},"order":8,"mode":0,"inputs":[{"name":"images","type":"IMAGE","link":59,"localized_name":"images"}],"outputs":[],"properties":{"Node name for S&R":"PreviewImage"},"widgets_values":[]},{"id":31,"type":"PreviewImage","pos":[1668.734619140625,796.7755737304688],"size":[210,246],"flags":{},"order":10,"mode":0,"inputs":[{"name":"images","type":"IMAGE","link":57,"localized_name":"images"}],"outputs":[],"properties":{"Node name for S&R":"PreviewImage"},"widgets_values":[]},{"id":29,"type":"PreviewImage","pos":[872.8858032226562,679.9429321289062],"size":[210,246],"flags":{},"order":6,"mode":0,"inputs":[{"name":"images","type":"IMAGE","link":55,"localized_name":"images"}],"outputs":[],"properties":{"Node name for S&R":"PreviewImage"},"widgets_values":[]},{"id":28,"type":"LoadImage","pos":[29.629886627197266,676.8370361328125],"size":[315,314],"flags":{},"order":1,"mode":0,"inputs":[],"outputs":[{"name":"IMAGE","type":"IMAGE","links":[54],"slot_index":0,"localized_name":"IMAGE"},{"name":"MASK","type":"MASK","links":null,"localized_name":"MASK"}],"properties":{"Node name for S&R":"LoadImage"},"widgets_values":["pexels-photo-1808399.jpeg","image"]},{"id":35,"type":"RemoveForegroundNode","pos":[414.2638854980469,1079.1705322265625],"size":[315,58],"flags":{},"order":3,"mode":0,"inputs":[{"name":"image","type":"IMAGE","link":61,"localized_name":"image"}],"outputs":[{"name":"output_image","type":"IMAGE","links":[62,63],"slot_index":0,"localized_name":"output_image"}],"properties":{"Node name for S&R":"RemoveForegroundNode"},"widgets_values":["BRIA_API_TOKEN"]},{"id":32,"type":"ReplaceBgNode","pos":[834.6115112304688,1063.0406494140625],"size":[315,294],"flags":{},"order":7,"mode":0,"inputs":[{"name":"image","type":"IMAGE","link":65,"localized_name":"image"},{"name":"ref_image","type":"IMAGE","link":62,"shape":7,"localized_name":"ref_image"}],"outputs":[{"name":"output_image","type":"IMAGE","links":[59,64],"slot_index":0,"localized_name":"output_image"}],"properties":{"Node name for S&R":"ReplaceBgNode"},"widgets_values":["BRIA_API_TOKEN",false,"",true,true,true,false,"",1978,"randomize"]},{"id":30,"type":"ImageExpansionNode","pos":[1265.001220703125,798.8323974609375],"size":[315,226],"flags":{},"order":9,"mode":0,"inputs":[{"name":"image","type":"IMAGE","link":64,"localized_name":"image"}],"outputs":[{"name":"output_image","type":"IMAGE","links":[57],"slot_index":0,"localized_name":"output_image"}],"properties":{"Node name for S&R":"ImageExpansionNode"},"widgets_values":["600,760","200, 0","BRIA_API_TOKEN","1200, 800","",1729,"randomize","Ugly, mutated"]},{"id":38,"type":"Note","pos":[436.7110900878906,566.2047119140625],"size":[306.28387451171875,58],"flags":{},"order":2,"mode":0,"inputs":[],"outputs":[],"properties":{},"widgets_values":["You can get your BRIA API token at:\nhttps://bria.ai/api/"],"color":"#432","bgcolor":"#653"},{"id":27,"type":"RmbgNode","pos":[430.8815002441406,678.7727661132812],"size":[315,58],"flags":{},"order":4,"mode":0,"inputs":[{"name":"image","type":"IMAGE","link":54,"localized_name":"image"}],"outputs":[{"name":"output_image","type":"IMAGE","links":[55,65],"slot_index":0,"localized_name":"output_image"}],"properties":{"Node name for S&R":"RmbgNode"},"widgets_values":["BRIA_API_TOKEN"]}],"links":[[51,5,0,15,2,"STRING"],[54,28,0,27,0,"IMAGE"],[55,27,0,29,0,"IMAGE"],[57,30,0,31,0,"IMAGE"],[59,32,0,33,0,"IMAGE"],[61,34,0,35,0,"IMAGE"],[62,35,0,32,1,"IMAGE"],[63,35,0,36,0,"IMAGE"],[64,32,0,30,0,"IMAGE"],[65,27,0,32,0,"IMAGE"]],"groups":[],"config":{},"extra":{"ds":{"scale":0.8140274938684037,"offset":[101.53311990208498,-477.0342684311694]},"node_versions":{"comfyui-bria-api":"c72754d15b53a13ee0c0419d70401232c56b7fdb","comfy-core":"v0.3.8-1-gc441048","ComfyUI-Jjk-Nodes":"b3c99bb78a99551776b5eab1a820e1cd58f84f31"}},"version":0.4}
+{
+  "id": "f47c8ffe-20e8-473c-a56b-c6241da2c469",
+  "revision": 0,
+  "last_node_id": 43,
+  "last_link_id": 78,
+  "nodes": [
+    {
+      "id": 31,
+      "type": "PreviewImage",
+      "pos": [
+        1668.734619140625,
+        796.7755737304688
+      ],
+      "size": [
+        210,
+        246
+      ],
+      "flags": {},
+      "order": 10,
+      "mode": 0,
+      "inputs": [
+        {
+          "name": "images",
+          "type": "IMAGE",
+          "link": 78
+        }
+      ],
+      "outputs": [],
+      "properties": {
+        "Node name for S&R": "PreviewImage"
+      },
+      "widgets_values": []
+    },
+    {
+      "id": 29,
+      "type": "PreviewImage",
+      "pos": [
+        872.8858032226562,
+        679.9429321289062
+      ],
+      "size": [
+        210,
+        246
+      ],
+      "flags": {},
+      "order": 5,
+      "mode": 0,
+      "inputs": [
+        {
+          "name": "images",
+          "type": "IMAGE",
+          "link": 66
+        }
+      ],
+      "outputs": [],
+      "properties": {
+        "Node name for S&R": "PreviewImage"
+      },
+      "widgets_values": []
+    },
+    {
+      "id": 38,
+      "type": "Note",
+      "pos": [
+        436.7110900878906,
+        566.2047119140625
+      ],
+      "size": [
+        306.28387451171875,
+        88
+      ],
+      "flags": {},
+      "order": 0,
+      "mode": 0,
+      "inputs": [],
+      "outputs": [],
+      "properties": {},
+      "widgets_values": [
+        "You can get your BRIA API token at:\nhttps://bria.ai/api/"
+      ],
+      "color": "#432",
+      "bgcolor": "#653"
+    },
+    {
+      "id": 28,
+      "type": "LoadImage",
+      "pos": [
+        29.629886627197266,
+        676.8370361328125
+      ],
+      "size": [
+        315,
+        314
+      ],
+      "flags": {},
+      "order": 1,
+      "mode": 0,
+      "inputs": [],
+      "outputs": [
+        {
+          "name": "IMAGE",
+          "type": "IMAGE",
+          "slot_index": 0,
+          "links": [
+            68
+          ]
+        },
+        {
+          "name": "MASK",
+          "type": "MASK",
+          "links": null
+        }
+      ],
+      "properties": {
+        "Node name for S&R": "LoadImage"
+      },
+      "widgets_values": [
+        "pexels-photo-1808399.jpeg",
+        "image"
+      ]
+    },
+    {
+      "id": 34,
+      "type": "LoadImage",
+      "pos": [
+        19.94045066833496,
+        1075.806640625
+      ],
+      "size": [
+        315,
+        314
+      ],
+      "flags": {},
+      "order": 2,
+      "mode": 0,
+      "inputs": [],
+      "outputs": [
+        {
+          "name": "IMAGE",
+          "type": "IMAGE",
+          "slot_index": 0,
+          "links": [
+            69
+          ]
+        },
+        {
+          "name": "MASK",
+          "type": "MASK",
+          "links": null
+        }
+      ],
+      "properties": {
+        "Node name for S&R": "LoadImage"
+      },
+      "widgets_values": [
+        "quirky-red-brick-brick-wallpaper.jpg",
+        "image"
+      ]
+    },
+    {
+      "id": 36,
+      "type": "PreviewImage",
+      "pos": [
+        455.6301574707031,
+        1240.017578125
+      ],
+      "size": [
+        210,
+        246
+      ],
+      "flags": {},
+      "order": 6,
+      "mode": 0,
+      "inputs": [
+        {
+          "name": "images",
+          "type": "IMAGE",
+          "link": 71
+        }
+      ],
+      "outputs": [],
+      "properties": {
+        "Node name for S&R": "PreviewImage"
+      },
+      "widgets_values": []
+    },
+    {
+      "id": 40,
+      "type": "RmbgNode",
+      "pos": [
+        414.095703125,
+        749.6687622070312
+      ],
+      "size": [
+        347.462890625,
+        130
+      ],
+      "flags": {},
+      "order": 3,
+      "mode": 0,
+      "inputs": [
+        {
+          "name": "images",
+          "type": "IMAGE",
+          "link": 68
+        }
+      ],
+      "outputs": [
+        {
+          "name": "output_images",
+          "type": "IMAGE",
+          "links": [
+            66,
+            74
+          ]
+        }
+      ],
+      "properties": {
+        "Node name for S&R": "RmbgNode"
+      },
+      "widgets_values": [
+        "BRIA_API_TOKEN",
+        false,
+        false,
+        true
+      ]
+    },
+    {
+      "id": 41,
+      "type": "RemoveForegroundNode",
+      "pos": [
+        397.44732666015625,
+        1127.818359375
+      ],
+      "size": [
+        347.462890625,
+        130
+      ],
+      "flags": {},
+      "order": 4,
+      "mode": 0,
+      "inputs": [
+        {
+          "name": "images",
+          "type": "IMAGE",
+          "link": 69
+        }
+      ],
+      "outputs": [
+        {
+          "name": "output_images",
+          "type": "IMAGE",
+          "links": [
+            71,
+            75
+          ]
+        }
+      ],
+      "properties": {
+        "Node name for S&R": "RemoveForegroundNode"
+      },
+      "widgets_values": [
+        "BRIA_API_TOKEN",
+        false,
+        false,
+        true
+      ]
+    },
+    {
+      "id": 42,
+      "type": "ReplaceBgNode",
+      "pos": [
+        800.56884765625,
+        1050.5228271484375
+      ],
+      "size": [
+        347.462890625,
+        318
+      ],
+      "flags": {},
+      "order": 7,
+      "mode": 0,
+      "inputs": [
+        {
+          "name": "images",
+          "type": "IMAGE",
+          "link": 74
+        },
+        {
+          "name": "ref_images",
+          "shape": 7,
+          "type": "IMAGE",
+          "link": 75
+        }
+      ],
+      "outputs": [
+        {
+          "name": "output_images",
+          "type": "IMAGE",
+          "links": [
+            72,
+            77
+          ]
+        }
+      ],
+      "properties": {
+        "Node name for S&R": "ReplaceBgNode"
+      },
+      "widgets_values": [
+        "BRIA_API_TOKEN",
+        "base",
+        "",
+        true,
+        true,
+        false,
+        "",
+        "681794",
+        false,
+        false,
+        false
+      ]
+    },
+    {
+      "id": 43,
+      "type": "ImageExpansionNode",
+      "pos": [
+        1241.743896484375,
+        731.8318481445312
+      ],
+      "size": [
+        347.462890625,
+        322
+      ],
+      "flags": {},
+      "order": 9,
+      "mode": 0,
+      "inputs": [
+        {
+          "name": "images",
+          "type": "IMAGE",
+          "link": 77
+        }
+      ],
+      "outputs": [
+        {
+          "name": "output_images",
+          "type": "IMAGE",
+          "links": [
+            78
+          ]
+        }
+      ],
+      "properties": {
+        "Node name for S&R": "ImageExpansionNode"
+      },
+      "widgets_values": [
+        "BRIA_API_TOKEN",
+        "",
+        "",
+        "1000, 1000",
+        "None",
+        "",
+        "681794",
+        "Ugly, mutated",
+        false,
+        true,
+        false,
+        false
+      ]
+    },
+    {
+      "id": 33,
+      "type": "PreviewImage",
+      "pos": [
+        1316.2191162109375,
+        1169.61962890625
+      ],
+      "size": [
+        210,
+        246
+      ],
+      "flags": {},
+      "order": 8,
+      "mode": 0,
+      "inputs": [
+        {
+          "name": "images",
+          "type": "IMAGE",
+          "link": 72
+        }
+      ],
+      "outputs": [],
+      "properties": {
+        "Node name for S&R": "PreviewImage"
+      },
+      "widgets_values": []
+    }
+  ],
+  "links": [
+    [
+      66,
+      40,
+      0,
+      29,
+      0,
+      "IMAGE"
+    ],
+    [
+      68,
+      28,
+      0,
+      40,
+      0,
+      "IMAGE"
+    ],
+    [
+      69,
+      34,
+      0,
+      41,
+      0,
+      "IMAGE"
+    ],
+    [
+      71,
+      41,
+      0,
+      36,
+      0,
+      "IMAGE"
+    ],
+    [
+      72,
+      42,
+      0,
+      33,
+      0,
+      "IMAGE"
+    ],
+    [
+      74,
+      40,
+      0,
+      42,
+      0,
+      "IMAGE"
+    ],
+    [
+      75,
+      41,
+      0,
+      42,
+      1,
+      "IMAGE"
+    ],
+    [
+      77,
+      42,
+      0,
+      43,
+      0,
+      "IMAGE"
+    ],
+    [
+      78,
+      43,
+      0,
+      31,
+      0,
+      "IMAGE"
+    ]
+  ],
+  "groups": [],
+  "config": {},
+  "extra": {
+    "ds": {
+      "scale": 0.61159090448415,
+      "offset": [
+        138.07082928418134,
+        -488.9478800208311
+      ]
+    },
+    "node_versions": {
+      "comfyui-bria-api": "c72754d15b53a13ee0c0419d70401232c56b7fdb",
+      "comfy-core": "v0.3.8-1-gc441048",
+      "ComfyUI-Jjk-Nodes": "b3c99bb78a99551776b5eab1a820e1cd58f84f31"
+    },
+    "frontendVersion": "1.25.11"
+  },
+  "version": 0.4
+}

workflows/generate_image_fibo-lite_workflow.json

@@ -0,0 +1,661 @@
+{
+  "id": "1c31a92d-1e48-46b0-b4ad-60d537260922",
+  "revision": 0,
+  "last_node_id": 27,
+  "last_link_id": 46,
+  "nodes": [
+    {
+      "id": 5,
+      "type": "PreviewImage",
+      "pos": [
+        1235.1895751953125,
+        10.958272933959961
+      ],
+      "size": [
+        140,
+        26
+      ],
+      "flags": {},
+      "order": 10,
+      "mode": 0,
+      "inputs": [
+        {
+          "name": "images",
+          "type": "IMAGE",
+          "link": 43
+        }
+      ],
+      "outputs": [],
+      "properties": {
+        "Node name for S&R": "PreviewImage"
+      },
+      "widgets_values": []
+    },
+    {
+      "id": 6,
+      "type": "PreviewImage",
+      "pos": [
+        774.4347534179688,
+        -58.6872444152832
+      ],
+      "size": [
+        140,
+        26
+      ],
+      "flags": {},
+      "order": 6,
+      "mode": 0,
+      "inputs": [
+        {
+          "name": "images",
+          "type": "IMAGE",
+          "link": 29
+        }
+      ],
+      "outputs": [],
+      "properties": {
+        "Node name for S&R": "PreviewImage"
+      },
+      "widgets_values": []
+    },
+    {
+      "id": 1,
+      "type": "Note",
+      "pos": [
+        -175.84976196289062,
+        102.76270294189453
+      ],
+      "size": [
+        210,
+        88
+      ],
+      "flags": {},
+      "order": 0,
+      "mode": 0,
+      "inputs": [],
+      "outputs": [],
+      "properties": {},
+      "widgets_values": [
+        "If you would like to start with prompt"
+      ],
+      "color": "#432",
+      "bgcolor": "#653"
+    },
+    {
+      "id": 2,
+      "type": "Note",
+      "pos": [
+        -225.1866455078125,
+        624.7235107421875
+      ],
+      "size": [
+        210,
+        88
+      ],
+      "flags": {},
+      "order": 1,
+      "mode": 0,
+      "inputs": [],
+      "outputs": [],
+      "properties": {},
+      "widgets_values": [
+        "If you would like to start with reference image + prompt"
+      ],
+      "color": "#432",
+      "bgcolor": "#653"
+    },
+    {
+      "id": 15,
+      "type": "PreviewImage",
+      "pos": [
+        719.5359497070312,
+        443.11285400390625
+      ],
+      "size": [
+        140,
+        26
+      ],
+      "flags": {},
+      "order": 8,
+      "mode": 0,
+      "inputs": [
+        {
+          "name": "images",
+          "type": "IMAGE",
+          "link": 36
+        }
+      ],
+      "outputs": [],
+      "properties": {
+        "Node name for S&R": "PreviewImage"
+      },
+      "widgets_values": []
+    },
+    {
+      "id": 17,
+      "type": "PreviewImage",
+      "pos": [
+        1132.783935546875,
+        517.3931884765625
+      ],
+      "size": [
+        140,
+        26
+      ],
+      "flags": {},
+      "order": 11,
+      "mode": 0,
+      "inputs": [
+        {
+          "name": "images",
+          "type": "IMAGE",
+          "link": 44
+        }
+      ],
+      "outputs": [],
+      "properties": {
+        "Node name for S&R": "PreviewImage"
+      },
+      "widgets_values": []
+    },
+    {
+      "id": 20,
+      "type": "GenerateStructuredPromptLiteNodeV2",
+      "pos": [
+        68.9760971069336,
+        82.11421966552734
+      ],
+      "size": [
+        310.7749938964844,
+        174
+      ],
+      "flags": {},
+      "order": 2,
+      "mode": 0,
+      "inputs": [
+        {
+          "name": "images",
+          "shape": 7,
+          "type": "IMAGE",
+          "link": null
+        }
+      ],
+      "outputs": [
+        {
+          "name": "structured_prompt",
+          "type": "STRING",
+          "links": [
+            27
+          ]
+        },
+        {
+          "name": "seed",
+          "type": "INT",
+          "links": [
+            28
+          ]
+        }
+      ],
+      "properties": {
+        "Node name for S&R": "GenerateStructuredPromptLiteNodeV2"
+      },
+      "widgets_values": [
+        "BRIA_API_TOKEN",
+        "",
+        "",
+        123456,
+        "randomize"
+      ]
+    },
+    {
+      "id": 24,
+      "type": "GenerateImageLiteNodeV2",
+      "pos": [
+        460.6510314941406,
+        44.938873291015625
+      ],
+      "size": [
+        270,
+        290
+      ],
+      "flags": {},
+      "order": 4,
+      "mode": 0,
+      "inputs": [
+        {
+          "name": "images",
+          "shape": 7,
+          "type": "IMAGE",
+          "link": null
+        },
+        {
+          "name": "structured_prompt",
+          "shape": 7,
+          "type": "STRING",
+          "widget": {
+            "name": "structured_prompt"
+          },
+          "link": 27
+        },
+        {
+          "name": "seed",
+          "shape": 7,
+          "type": "INT",
+          "widget": {
+            "name": "seed"
+          },
+          "link": 28
+        }
+      ],
+      "outputs": [
+        {
+          "name": "image",
+          "type": "IMAGE",
+          "links": [
+            29
+          ]
+        },
+        {
+          "name": "structured_prompt",
+          "type": "STRING",
+          "links": [
+            41
+          ]
+        },
+        {
+          "name": "seed",
+          "type": "INT",
+          "links": [
+            42
+          ]
+        }
+      ],
+      "properties": {
+        "Node name for S&R": "GenerateImageLiteNodeV2"
+      },
+      "widgets_values": [
+        "BRIA_API_TOKEN",
+        "",
+        "FIBO",
+        "",
+        "1:1",
+        8,
+        5,
+        123456,
+        "randomize"
+      ]
+    },
+    {
+      "id": 26,
+      "type": "RefineImageLiteNodeV2",
+      "pos": [
+        834.9641723632812,
+        46.47432327270508
+      ],
+      "size": [
+        270,
+        290
+      ],
+      "flags": {},
+      "order": 7,
+      "mode": 0,
+      "inputs": [
+        {
+          "name": "structured_prompt",
+          "type": "STRING",
+          "widget": {
+            "name": "structured_prompt"
+          },
+          "link": 41
+        },
+        {
+          "name": "seed",
+          "shape": 7,
+          "type": "INT",
+          "widget": {
+            "name": "seed"
+          },
+          "link": 42
+        }
+      ],
+      "outputs": [
+        {
+          "name": "image",
+          "type": "IMAGE",
+          "links": [
+            43
+          ]
+        },
+        {
+          "name": "structured_prompt",
+          "type": "STRING",
+          "links": null
+        },
+        {
+          "name": "seed",
+          "type": "INT",
+          "links": null
+        }
+      ],
+      "properties": {
+        "Node name for S&R": "RefineImageLiteNodeV2"
+      },
+      "widgets_values": [
+        "BRIA_API_TOKEN",
+        "",
+        "",
+        "FIBO",
+        "1:1",
+        8,
+        5,
+        123456,
+        "randomize"
+      ]
+    },
+    {
+      "id": 25,
+      "type": "GenerateImageLiteNodeV2",
+      "pos": [
+        404.58843994140625,
+        532.5309448242188
+      ],
+      "size": [
+        270,
+        290
+      ],
+      "flags": {},
+      "order": 5,
+      "mode": 0,
+      "inputs": [
+        {
+          "name": "images",
+          "shape": 7,
+          "type": "IMAGE",
+          "link": null
+        },
+        {
+          "name": "structured_prompt",
+          "shape": 7,
+          "type": "STRING",
+          "widget": {
+            "name": "structured_prompt"
+          },
+          "link": 34
+        },
+        {
+          "name": "seed",
+          "shape": 7,
+          "type": "INT",
+          "widget": {
+            "name": "seed"
+          },
+          "link": 35
+        }
+      ],
+      "outputs": [
+        {
+          "name": "image",
+          "type": "IMAGE",
+          "links": [
+            36
+          ]
+        },
+        {
+          "name": "structured_prompt",
+          "type": "STRING",
+          "links": [
+            45
+          ]
+        },
+        {
+          "name": "seed",
+          "type": "INT",
+          "links": [
+            46
+          ]
+        }
+      ],
+      "properties": {
+        "Node name for S&R": "GenerateImageLiteNodeV2"
+      },
+      "widgets_values": [
+        "BRIA_API_TOKEN",
+        "",
+        "FIBO",
+        "",
+        "1:1",
+        8,
+        5,
+        123456,
+        "randomize"
+      ]
+    },
+    {
+      "id": 27,
+      "type": "RefineImageLiteNodeV2",
+      "pos": [
+        820.26611328125,
+        558.9785766601562
+      ],
+      "size": [
+        270,
+        290
+      ],
+      "flags": {},
+      "order": 9,
+      "mode": 0,
+      "inputs": [
+        {
+          "name": "structured_prompt",
+          "type": "STRING",
+          "widget": {
+            "name": "structured_prompt"
+          },
+          "link": 45
+        },
+        {
+          "name": "seed",
+          "shape": 7,
+          "type": "INT",
+          "widget": {
+            "name": "seed"
+          },
+          "link": 46
+        }
+      ],
+      "outputs": [
+        {
+          "name": "image",
+          "type": "IMAGE",
+          "links": [
+            44
+          ]
+        },
+        {
+          "name": "structured_prompt",
+          "type": "STRING",
+          "links": null
+        },
+        {
+          "name": "seed",
+          "type": "INT",
+          "links": null
+        }
+      ],
+      "properties": {
+        "Node name for S&R": "RefineImageLiteNodeV2"
+      },
+      "widgets_values": [
+        "BRIA_API_TOKEN",
+        "",
+        "",
+        "FIBO",
+        "1:1",
+        8,
+        5,
+        123456,
+        "randomize"
+      ]
+    },
+    {
+      "id": 21,
+      "type": "GenerateStructuredPromptLiteNodeV2",
+      "pos": [
+        3.162916421890259,
+        587.301025390625
+      ],
+      "size": [
+        310.7749938964844,
+        174
+      ],
+      "flags": {},
+      "order": 3,
+      "mode": 0,
+      "inputs": [
+        {
+          "name": "images",
+          "shape": 7,
+          "type": "IMAGE",
+          "link": null
+        }
+      ],
+      "outputs": [
+        {
+          "name": "structured_prompt",
+          "type": "STRING",
+          "links": [
+            34
+          ]
+        },
+        {
+          "name": "seed",
+          "type": "INT",
+          "links": [
+            35
+          ]
+        }
+      ],
+      "properties": {
+        "Node name for S&R": "GenerateStructuredPromptLiteNodeV2"
+      },
+      "widgets_values": [
+        "BRIA_API_TOKEN",
+        "",
+        "",
+        123456,
+        "randomize"
+      ]
+    }
+  ],
+  "links": [
+    [
+      27,
+      20,
+      0,
+      24,
+      1,
+      "STRING"
+    ],
+    [
+      28,
+      20,
+      1,
+      24,
+      2,
+      "INT"
+    ],
+    [
+      29,
+      24,
+      0,
+      6,
+      0,
+      "IMAGE"
+    ],
+    [
+      34,
+      21,
+      0,
+      25,
+      1,
+      "STRING"
+    ],
+    [
+      35,
+      21,
+      1,
+      25,
+      2,
+      "INT"
+    ],
+    [
+      36,
+      25,
+      0,
+      15,
+      0,
+      "IMAGE"
+    ],
+    [
+      41,
+      24,
+      1,
+      26,
+      0,
+      "STRING"
+    ],
+    [
+      42,
+      24,
+      2,
+      26,
+      1,
+      "INT"
+    ],
+    [
+      43,
+      26,
+      0,
+      5,
+      0,
+      "IMAGE"
+    ],
+    [
+      44,
+      27,
+      0,
+      17,
+      0,
+      "IMAGE"
+    ],
+    [
+      45,
+      25,
+      1,
+      27,
+      0,
+      "STRING"
+    ],
+    [
+      46,
+      25,
+      2,
+      27,
+      1,
+      "INT"
+    ]
+  ],
+  "groups": [],
+  "config": {},
+  "extra": {
+    "ds": {
+      "scale": 0.6276708501927047,
+      "offset": [
+        650.5506889681789,
+        132.55696596915172
+      ]
+    },
+    "frontendVersion": "1.25.11"
+  },
+  "version": 0.4
+}