From eda093ea664ea3ef583983fb6d14eb942c8d8c0a Mon Sep 17 00:00:00 2001 From: gabiburtman <102729750+gabiburtman@users.noreply.github.com> Date: Tue, 9 Sep 2025 11:13:41 +0300 Subject: [PATCH] Update shot_by_text_node.py --- nodes/shot_by_text_node.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/nodes/shot_by_text_node.py b/nodes/shot_by_text_node.py index 848c364..fa58eeb 100644 --- a/nodes/shot_by_text_node.py +++ b/nodes/shot_by_text_node.py @@ -11,6 +11,7 @@ class ShotByTextNode(): "image": ("IMAGE",), # Input image from another node "scene_description": ("STRING",), "optimize_description": ("INT", {"default": 1}), + "mode": (["base", "fast", "high_control"], {"default": "high_control"}), "api_key": ("STRING", {"default": "BRIA_API_TOKEN"}) # API Key input with a default value }, "optional": { @@ -27,7 +28,7 @@ class ShotByTextNode(): self.api_url = "https://engine.prod.bria-api.com/v1/product/lifestyle_shot_by_text" # Eraser API URL # Define the execute method as expected by ComfyUI - def execute(self, image, api_key, scene_description, optimize_description, content_moderation): + def execute(self, image, api_key, scene_description, optimize_description, mode, content_moderation): if api_key.strip() == "" or api_key.strip() == "BRIA_API_TOKEN": raise Exception("Please insert a valid API key.") @@ -40,6 +41,7 @@ class ShotByTextNode(): payload = { "file": image_base64, "scene_description": scene_description, + "mode": mode, "optimize_description": optimize_description, "placement_type": "original", "original_quality": True,