bizyengine 1.2.68__py3-none-any.whl → 1.2.69__py3-none-any.whl

bizyengine/bizyair_extras/third_party_api/nodes_gemini.py

@@ -0,0 +1,403 @@
+from bizyairsdk import tensor_to_bytesio
+
+from .trd_nodes_base import BizyAirTrdApiBaseNode
+
+
+# FROM: https://github.com/ShmuelRonen/ComfyUI-NanoBanano/blob/9eeb8f2411fd0ff08791bdf5e24eec347456c8b8/nano_banano.py#L191
+def build_prompt_for_operation(
+    prompt,
+    operation,
+    has_references=False,
+    aspect_ratio="auto",
+    character_consistency=True,
+):
+    """Build optimized prompt based on operation type"""
+
+    auto_aspect = (
+        "keep the original image aspect ratio"
+        if has_references
+        else "use an appropriate aspect ratio"
+    )
+    aspect_instructions = {
+        "1:1": "square format",
+        "16:9": "widescreen landscape format",
+        "9:16": "portrait format",
+        "4:3": "standard landscape format",
+        "3:4": "standard portrait format",
+        "auto": auto_aspect,
+    }
+
+    base_quality = "Generate a high-quality, photorealistic image"
+    format_instruction = f"in {aspect_instructions.get(aspect_ratio, auto_aspect)}"
+
+    if operation == "generate":
+        if has_references:
+            final_prompt = f"{base_quality} inspired by the style and elements of the reference images. {prompt}. {format_instruction}."
+        else:
+            final_prompt = f"{base_quality} of: {prompt}. {format_instruction}."
+
+    elif operation == "edit":
+        if not has_references:
+            return "Error: Edit operation requires reference images"
+        # No aspect ratio for edit - preserve original image dimensions
+        final_prompt = f"Edit the provided reference image(s). {prompt}. Maintain the original composition and quality while making the requested changes."
+
+    elif operation == "style_transfer":
+        if not has_references:
+            return "Error: Style transfer requires reference images"
+        final_prompt = f"Apply the style from the reference images to create: {prompt}. Blend the stylistic elements naturally. {format_instruction}."
+
+    elif operation == "object_insertion":
+        if not has_references:
+            return "Error: Object insertion requires reference images"
+        final_prompt = f"Insert or blend the following into the reference image(s): {prompt}. Ensure natural lighting, shadows, and perspective. {format_instruction}."
+
+    if character_consistency and has_references:
+        final_prompt += " Maintain character consistency and visual identity from the reference images."
+
+    return final_prompt
+
+
+class NanoBanana(BizyAirTrdApiBaseNode):
+    NODE_DISPLAY_NAME = "NanoBanana"
+    RETURN_TYPES = ("IMAGE", "STRING")
+    CATEGORY = "☁️BizyAir/External APIs/Gemini"
+
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+            "required": {
+                "prompt": (
+                    "STRING",
+                    {
+                        "multiline": True,
+                        "default": "",
+                    },
+                ),
+                "operation": (
+                    ["generate", "edit", "style_transfer", "object_insertion"],
+                    {
+                        "default": "generate",
+                        "tooltip": "Choose the type of image operation",
+                    },
+                ),
+                "temperature": (
+                    "FLOAT",
+                    {"default": 1.0, "min": 0.0, "max": 1.0, "step": 0.05},
+                ),
+                "top_p": (
+                    "FLOAT",
+                    {"default": 1.0, "min": 0.0, "max": 1.0, "step": 0.05},
+                ),
+                "seed": ("INT", {"default": 0, "min": 0, "max": 2147483647}),
+                "max_tokens": ("INT", {"default": 8192, "min": 1, "max": 8192}),
+            },
+            "optional": {
+                "image": ("IMAGE",),
+                "image2": ("IMAGE",),
+                "image3": ("IMAGE",),
+                "image4": ("IMAGE",),
+                "image5": ("IMAGE",),
+                "quality": (
+                    ["standard", "high"],
+                    {"default": "high", "tooltip": "Image generation quality"},
+                ),
+                "aspect_ratio": (
+                    ["1:1", "16:9", "9:16", "4:3", "3:4", "auto"],
+                    {"default": "auto", "tooltip": "Output image aspect ratio"},
+                ),
+                "character_consistency": (
+                    "BOOLEAN",
+                    {
+                        "default": True,
+                        "tooltip": "Maintain character consistency across edits",
+                    },
+                ),
+            },
+        }
+
+    def handle_inputs(self, headers, prompt_id, **kwargs):
+        operation = kwargs.get("operation", "generate")
+        temperature = kwargs.get("temperature", 1.0)
+        top_p = kwargs.get("top_p", 1.0)
+        seed = kwargs.get("seed", 0)
+        max_tokens = kwargs.get("max_tokens", 8192)
+        quality = kwargs.get("quality", "high")
+        aspect_ratio = kwargs.get("aspect_ratio", "auto")
+        character_consistency = kwargs.get("character_consistency", True)
+        prompt = kwargs.get("prompt", "")
+
+        parts = []
+        for i, img in enumerate(
+            [
+                kwargs.get("image", None),
+                kwargs.get("image2", None),
+                kwargs.get("image3", None),
+                kwargs.get("image4", None),
+                kwargs.get("image5", None),
+            ],
+            1,
+        ):
+            if img is not None:
+                url = self.upload_file(
+                    tensor_to_bytesio(image=img, total_pixels=4096 * 4096),
+                    f"{prompt_id}_{i}.png",
+                    headers,
+                )
+                parts.append(url)
+
+        prompt = build_prompt_for_operation(
+            prompt,
+            operation,
+            has_references=len(parts) > 0,
+            aspect_ratio=aspect_ratio,
+            character_consistency=character_consistency,
+        )
+        if quality == "high":
+            prompt += " Use the highest quality settings available."
+
+        data = {
+            "urls": parts,
+            "prompt": prompt,
+            "temperature": temperature,
+            "top_p": top_p,
+            "seed": seed,
+            "max_tokens": max_tokens,
+        }
+        if aspect_ratio != "auto":
+            data["aspect_ratio"] = aspect_ratio
+        return data, "gemini-2.5-flash-image"
+
+    def handle_outputs(self, outputs):
+        text = None
+        if len(outputs[1]) > 0:
+            image = outputs[1][0]
+        else:
+            raise ValueError("No image found in response")
+        if len(outputs[2]) > 0:
+            text = outputs[2][0]
+        return (image, text)
+
+
+class NanoBananaPro(BizyAirTrdApiBaseNode):
+    NODE_DISPLAY_NAME = "NanoBananaPro"
+    RETURN_TYPES = ("IMAGE", "STRING")
+    RETURN_NAMES = ("image", "string")
+    CATEGORY = "☁️BizyAir/External APIs/Gemini"
+
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+            "required": {
+                "prompt": (
+                    "STRING",
+                    {
+                        "multiline": True,
+                        "default": "",
+                    },
+                ),
+                "operation": (
+                    ["generate", "edit", "style_transfer", "object_insertion"],
+                    {
+                        "default": "generate",
+                        "tooltip": "Choose the type of image operation",
+                    },
+                ),
+                "temperature": (
+                    "FLOAT",
+                    {"default": 1.0, "min": 0.0, "max": 1.0, "step": 0.05},
+                ),
+                "top_p": (
+                    "FLOAT",
+                    {"default": 1.0, "min": 0.0, "max": 1.0, "step": 0.05},
+                ),
+                "seed": ("INT", {"default": 0, "min": 0, "max": 2147483647}),
+                "max_tokens": ("INT", {"default": 32768, "min": 1, "max": 32768}),
+                "aspect_ratio": (
+                    [
+                        "1:1",
+                        "2:3",
+                        "3:2",
+                        "3:4",
+                        "4:3",
+                        "4:5",
+                        "5:4",
+                        "9:16",
+                        "16:9",
+                        "21:9",
+                        "auto",
+                    ],
+                    {"default": "auto", "tooltip": "Output image aspect ratio"},
+                ),
+                "resolution": (["1K", "2K", "4K", "auto"], {"default": "auto"}),
+            },
+            "optional": {
+                "images": ("IMAGE",),
+                "quality": (
+                    ["standard", "high"],
+                    {"default": "high", "tooltip": "Image generation quality"},
+                ),
+                "character_consistency": (
+                    "BOOLEAN",
+                    {
+                        "default": True,
+                        "tooltip": "Maintain character consistency across edits",
+                    },
+                ),
+            },
+        }
+
+    def handle_inputs(self, headers, prompt_id, **kwargs):
+        operation = kwargs.get("operation", "generate")
+        temperature = kwargs.get("temperature", 1.0)
+        top_p = kwargs.get("top_p", 1.0)
+        seed = kwargs.get("seed", 0)
+        max_tokens = kwargs.get("max_tokens", 32768)
+        quality = kwargs.get("quality", "high")
+        aspect_ratio = kwargs.get("aspect_ratio", "auto")
+        resolution = kwargs.get("resolution", "auto")
+        images = kwargs.get("images", None)
+        character_consistency = kwargs.get("character_consistency", True)
+        prompt = kwargs.get("prompt", "")
+
+        if images is not None and len(images) > 14:
+            raise ValueError("Maximum number of images is 14")
+        parts = []
+        for batch_number, img in enumerate(images if images is not None else []):
+            if img is not None:
+                url = self.upload_file(
+                    tensor_to_bytesio(image=img, total_pixels=4096 * 4096),
+                    f"{prompt_id}_{batch_number}.png",
+                    headers,
+                )
+                parts.append(url)
+
+        prompt = build_prompt_for_operation(
+            prompt,
+            operation,
+            has_references=len(parts) > 0,
+            aspect_ratio=aspect_ratio,
+            character_consistency=character_consistency,
+        )
+        if quality == "high":
+            prompt += " Use the highest quality settings available."
+
+        data = {
+            "urls": parts,
+            "prompt": prompt,
+            "temperature": temperature,
+            "top_p": top_p,
+            "seed": seed,
+            "max_tokens": max_tokens,
+        }
+        if aspect_ratio != "auto":
+            data["aspect_ratio"] = aspect_ratio
+        if resolution != "auto":
+            data["resolution"] = resolution
+        return data, "gemini-3-pro-image-preview"
+
+    def handle_outputs(self, outputs):
+        text = None
+        if len(outputs[1]) > 0:
+            image = outputs[1][0]
+        else:
+            raise ValueError("No image found in response")
+        if len(outputs[2]) > 0:
+            text = outputs[2][0]
+        return (image, text)
+
+
+class NanoBananaProOfficial(BizyAirTrdApiBaseNode):
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+            "required": {
+                "prompt": (
+                    "STRING",
+                    {
+                        "multiline": True,
+                        "default": "",
+                    },
+                ),
+                "temperature": (
+                    "FLOAT",
+                    {"default": 1.0, "min": 0.0, "max": 1.0, "step": 0.05},
+                ),
+                "top_p": (
+                    "FLOAT",
+                    {"default": 1.0, "min": 0.0, "max": 1.0, "step": 0.05},
+                ),
+                "seed": ("INT", {"default": 0, "min": 0, "max": 2147483647}),
+                "max_tokens": ("INT", {"default": 32768, "min": 1, "max": 32768}),
+                "aspect_ratio": (
+                    [
+                        "1:1",
+                        "2:3",
+                        "3:2",
+                        "3:4",
+                        "4:3",
+                        "4:5",
+                        "5:4",
+                        "9:16",
+                        "16:9",
+                        "21:9",
+                        "auto",
+                    ],
+                    {"default": "auto", "tooltip": "Output image aspect ratio"},
+                ),
+                "resolution": (["1K", "2K", "4K", "auto"], {"default": "auto"}),
+            },
+            "optional": {
+                "images": ("IMAGE",),
+            },
+        }
+
+    RETURN_TYPES = ("IMAGE", "STRING")
+    RETURN_NAMES = ("image", "string")
+    CATEGORY = "☁️BizyAir/External APIs/Gemini"
+    NODE_DISPLAY_NAME = "NanoBananaPro (Official Parameters)"
+
+    def handle_inputs(self, headers, prompt_id, **kwargs):
+        prompt = kwargs.get("prompt", "")
+        temperature = kwargs.get("temperature", 1.0)
+        top_p = kwargs.get("top_p", 1.0)
+        seed = kwargs.get("seed", 0)
+        max_tokens = kwargs.get("max_tokens", 32768)
+        aspect_ratio = kwargs.get("aspect_ratio", "auto")
+        resolution = kwargs.get("resolution", "auto")
+        images = kwargs.get("images", None)
+        if images is not None and len(images) > 14:
+            raise ValueError("Maximum number of images is 14")
+        parts = []
+        for batch_number, img in enumerate(images if images is not None else []):
+            if img is not None:
+                url = self.upload_file(
+                    tensor_to_bytesio(image=img, total_pixels=4096 * 4096),
+                    f"{prompt_id}_{batch_number}.png",
+                    headers,
+                )
+                parts.append(url)
+
+        data = {
+            "urls": parts,
+            "prompt": prompt,
+            "temperature": temperature,
+            "top_p": top_p,
+            "seed": seed,
+            "max_tokens": max_tokens,
+        }
+        if aspect_ratio != "auto":
+            data["aspect_ratio"] = aspect_ratio
+        if resolution != "auto":
+            data["resolution"] = resolution
+        return data, "gemini-3-pro-image-preview"
+
+    def handle_outputs(self, outputs):
+        text = None
+        if len(outputs[1]) > 0:
+            image = outputs[1][0]
+        else:
+            raise ValueError("No image found in response")
+        if len(outputs[2]) > 0:
+            text = outputs[2][0]
+        return (image, text)

bizyengine/bizyair_extras/third_party_api/nodes_gpt.py

@@ -0,0 +1,101 @@
+from bizyairsdk import tensor_to_bytesio
+
+from .trd_nodes_base import BizyAirTrdApiBaseNode
+
+
+class GPT_IMAGE_1_T2I_API(BizyAirTrdApiBaseNode):
+    NODE_DISPLAY_NAME = "GPT Image 1 Text To Image"
+    RETURN_TYPES = ("IMAGE",)
+    RETURN_NAMES = ("images",)
+    CATEGORY = "☁️BizyAir/External APIs/OpenAI"
+
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "prompt": (
+                    "STRING",
+                    {
+                        "multiline": True,
+                        "default": "",
+                    },
+                ),
+                "size": (["1:1", "2:3", "3:2"], {"default": "1:1"}),
+                "variants": ([1, 2, 4], {"default": 1}),
+            },
+        }
+
+    def handle_inputs(self, headers, prompt_id, **kwargs):
+        prompt = kwargs.get("prompt", "")
+        size = kwargs.get("size", "1:1")
+        variants = kwargs.get("variants", 1)
+        data = {
+            "prompt": prompt,
+            "size": size,
+            "variants": variants,
+        }
+        if variants == 4:
+            data["provider"] = "KieAI"
+        return data, "gpt-image-1"
+
+    def handle_outputs(self, outputs):
+        images = self.combine_images(outputs[1])
+        return (images,)
+
+
+class GPT_IMAGE_EDIT_API(BizyAirTrdApiBaseNode):
+    NODE_DISPLAY_NAME = "GPT Image Edit"
+    RETURN_TYPES = ("IMAGE",)
+    RETURN_NAMES = ("images",)
+    CATEGORY = "☁️BizyAir/External APIs/OpenAI"
+
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "images": ("IMAGE",),
+                "prompt": (
+                    "STRING",
+                    {
+                        "multiline": True,
+                        "default": "",
+                    },
+                ),
+                "size": (["1:1", "2:3", "3:2"], {"default": "1:1"}),
+                "variants": ([1, 2, 4], {"default": 1}),
+            },
+        }
+
+    def handle_inputs(self, headers, prompt_id, **kwargs):
+        images = kwargs.get("images", None)
+        if images is None:
+            raise ValueError("Images are required")
+        prompt = kwargs.get("prompt", "")
+        size = kwargs.get("size", "1:1")
+        variants = kwargs.get("variants", 1)
+
+        urls = []
+        for batch_number, img in enumerate(images if images is not None else []):
+            if img is not None:
+                url = self.upload_file(
+                    tensor_to_bytesio(image=img, total_pixels=4096 * 4096),
+                    f"{prompt_id}_{batch_number}.png",
+                    headers,
+                )
+                urls.append(url)
+        if len(urls) == 0:
+            raise ValueError("At least one image is required")
+
+        data = {
+            "urls": urls,
+            "prompt": prompt,
+            "size": size,
+            "variants": variants,
+        }
+        if variants == 4:
+            data["provider"] = "KieAI"
+        return data, "gpt-image-1"
+
+    def handle_outputs(self, outputs):
+        images = self.combine_images(outputs[1])
+        return (images,)

bizyengine/bizyair_extras/third_party_api/nodes_hailuo.py

@@ -0,0 +1,115 @@
+from bizyairsdk import tensor_to_bytesio
+
+from .trd_nodes_base import BizyAirTrdApiBaseNode
+
+
+class Hailuo2_3_T2V(BizyAirTrdApiBaseNode):
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "prompt": (
+                    "STRING",
+                    {
+                        "multiline": True,
+                        "default": "",
+                    },
+                ),
+                "model": (
+                    ["MiniMax-Hailuo-2.3"],
+                    {"default": "MiniMax-Hailuo-2.3"},
+                ),
+                "duration": ([6, 10], {"default": 6}),
+                "resolution": (
+                    ["768P", "1080P"],
+                    {"default": "1080P"},
+                ),
+            },
+        }
+
+    NODE_DISPLAY_NAME = "Hailuo2.3 Text To Video"
+    RETURN_TYPES = ("VIDEO",)
+    RETURN_NAMES = ("video",)
+    CATEGORY = "☁️BizyAir/External APIs/Hailuo"
+
+    def handle_inputs(self, headers, prompt_id, **kwargs):
+        model = kwargs.get("model", "MiniMax-Hailuo-2.3")
+        duration = kwargs.get("duration", 6)
+        prompt = kwargs.get("prompt", "")
+        resolution = kwargs.get("resolution", "1080P")
+        if resolution == "1080P" and duration == 10:
+            raise ValueError(
+                "Hailuo2.3 1080P resolution + 10s duration is not supported"
+            )
+        data = {
+            "model": model,
+            "duration": duration,
+            "prompt": prompt,
+            "resolution": resolution,
+        }
+        return data, "MiniMax-Hailuo-2.3-t2v"
+
+    def handle_outputs(self, outputs):
+        return (outputs[0][0],)
+
+
+class Hailuo2_3_I2V(BizyAirTrdApiBaseNode):
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "first_frame_image": ("IMAGE",),
+                "prompt": (
+                    "STRING",
+                    {
+                        "multiline": True,
+                        "default": "",
+                    },
+                ),
+                "model": (
+                    ["MiniMax-Hailuo-2.3", "MiniMax-Hailuo-2.3-Fast"],
+                    {"default": "MiniMax-Hailuo-2.3"},
+                ),
+                "duration": ([6, 10], {"default": 6}),
+                "resolution": (
+                    ["768P", "1080P"],
+                    {"default": "1080P"},
+                ),
+            },
+        }
+
+    NODE_DISPLAY_NAME = "Hailuo2.3 Image To Video"
+    RETURN_TYPES = ("VIDEO",)
+    RETURN_NAMES = ("video",)
+    CATEGORY = "☁️BizyAir/External APIs/Hailuo"
+
+    def handle_inputs(self, headers, prompt_id, **kwargs):
+        model = kwargs.get("model", "MiniMax-Hailuo-2.3")
+        duration = kwargs.get("duration", 6)
+        prompt = kwargs.get("prompt", "")
+        resolution = kwargs.get("resolution", "1080P")
+        first_frame_image = kwargs.get("first_frame_image", None)
+        if first_frame_image is None:
+            raise ValueError("First frame image is required")
+        if resolution == "1080P" and duration == 10:
+            raise ValueError(
+                "Hailuo2.3 1080P resolution + 10s duration is not supported"
+            )
+        # 上传首帧图片
+        first_frame_image_url = self.upload_file(
+            tensor_to_bytesio(image=first_frame_image, total_pixels=4096 * 4096),
+            f"{prompt_id}.png",
+            headers,
+        )
+
+        data = {
+            "model": model,
+            "duration": duration,
+            "prompt": prompt,
+            "resolution": resolution,
+            "first_frame_image": first_frame_image_url,
+        }
+        return data, "MiniMax-Hailuo-2.3-i2v"
+
+    def handle_outputs(self, outputs):
+        return (outputs[0][0],)