bizyengine 1.2.51__py3-none-any.whl → 1.2.53__py3-none-any.whl

bizyengine/bizyair_extras/__init__.py

@@ -21,6 +21,7 @@ from .nodes_nunchaku import *
 from .nodes_reactor import *
 from .nodes_sam2 import *
 from .nodes_sd3 import *
+from .nodes_seedream import *
 from .nodes_segment_anything import *
 from .nodes_testing_utils import *
 from .nodes_trellis import *

bizyengine/bizyair_extras/nodes_gemini.py

@@ -289,7 +289,7 @@ class NanoBanana(BizyAirBaseNode):
             if prompt_feedback:
                 logging.error(f"Response: {resp}")
                 raise ValueError(f"Prompt blocked: {prompt_feedback}")
-            if len(resp["candidates"]) == 0:
+            if len(resp.get("candidates", [])) == 0:
                 logging.error(f"Response: {resp}")
                 raise ValueError("No candidates found in response")
             if resp["candidates"][0]["finishReason"] != "STOP":

bizyengine/bizyair_extras/nodes_seedream.py

@@ -0,0 +1,195 @@
+import base64
+import io
+import json
+import logging
+import re
+from pathlib import Path
+
+import requests
+import torch
+from comfy_api_nodes.apinode_utils import (
+    bytesio_to_image_tensor,
+    tensor_to_base64_string,
+)
+
+from bizyengine.core import BizyAirBaseNode, pop_api_key_and_prompt_id
+from bizyengine.core.common import client
+from bizyengine.core.common.env_var import BIZYAIR_SERVER_ADDRESS
+
+
+def download_png(url: str) -> bytes:
+    """下载 PNG 图片"""
+    resp = requests.get(url, stream=True, timeout=30)
+    resp.raise_for_status()  # 非 2xx 会抛异常
+    return resp.content
+
+
+class Seedream4(BizyAirBaseNode):
+    def __init__(self):
+        pass
+
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+            "required": {
+                "prompt": (
+                    "STRING",
+                    {
+                        "multiline": True,
+                        "default": "",
+                    },
+                ),
+                "size": (
+                    [
+                        "1K Square (1024x1024)",
+                        "2K Square (2048x2048)",
+                        "4K Square (4096x4096)",
+                        "HD 16:9 (1920x1080)",
+                        "2K 16:9 (2560x1440)",
+                        "4K 16:9 (3840x2160)",
+                        "Portrait 9:16 (1080x1920)",
+                        "Portrait 3:4 (1536x2048)",
+                        "Landscape 4:3 (2048x1536)",
+                        "Ultra-wide 21:9 (3440x1440)",
+                        "Custom",
+                    ],
+                    {
+                        "default": "HD 16:9 (1920x1080)",
+                    },
+                ),
+                "custom_width": ("INT", {"default": 1920, "min": 1024, "max": 4096}),
+                "custom_height": ("INT", {"default": 1080, "min": 1024, "max": 4096}),
+                "model": (
+                    ["doubao-seedream-4-0-250828"],
+                    {"default": "doubao-seedream-4-0-250828"},
+                ),
+            },
+            "optional": {
+                "image": ("IMAGE",),
+                "image2": ("IMAGE",),
+                "image3": ("IMAGE",),
+                "image4": ("IMAGE",),
+                "image5": ("IMAGE",),
+                "image6": ("IMAGE",),
+                "image7": ("IMAGE",),
+                "image8": ("IMAGE",),
+                "image9": ("IMAGE",),
+                "image10": ("IMAGE",),
+            },
+        }
+
+    RETURN_TYPES = ("IMAGE",)
+    FUNCTION = "execute"
+    OUTPUT_NODE = False
+    CATEGORY = "☁️BizyAir/External APIs/Doubao"
+
+    def execute(self, **kwargs):
+        try:
+            model = kwargs.get("model", "doubao-seedream-4-0-250828")
+            url = f"{BIZYAIR_SERVER_ADDRESS}/proxy_inference/Doubao/{model}"
+            extra_data = pop_api_key_and_prompt_id(kwargs)
+
+            prompt = kwargs.get("prompt", "")
+            size = kwargs.get("size", "1K Square (1024x1024)")
+
+            match size:
+                case "1K Square (1024x1024)":
+                    width = 1024
+                    height = 1024
+                case "2K Square (2048x2048)":
+                    width = 2048
+                    height = 2048
+                case "4K Square (4096x4096)":
+                    width = 4096
+                    height = 4096
+                case "HD 16:9 (1920x1080)":
+                    width = 1920
+                    height = 1080
+                case "2K 16:9 (2560x1440)":
+                    width = 2560
+                    height = 1440
+                case "4K 16:9 (3840x2160)":
+                    width = 3840
+                    height = 2160
+                case "Portrait 9:16 (1080x1920)":
+                    width = 1080
+                    height = 1920
+                case "Portrait 3:4 (1536x2048)":
+                    width = 1536
+                    height = 2048
+                case "Landscape 4:3 (2048x1536)":
+                    width = 2048
+                    height = 1536
+                case "Ultra-wide 21:9 (3440x1440)":
+                    width = 3440
+                    height = 1440
+                case "Custom":
+                    width = kwargs.get("custom_width", 1920)
+                    height = kwargs.get("custom_height", 1080)
+
+                case _:
+                    raise ValueError(f"Invalid size: {size}")
+
+            sizeStr = f"{width}x{height}"
+
+            images = []
+            total_size = 0
+            for _, img in enumerate(
+                [
+                    kwargs.get("image", None),
+                    kwargs.get("image2", None),
+                    kwargs.get("image3", None),
+                    kwargs.get("image4", None),
+                    kwargs.get("image5", None),
+                    kwargs.get("image6", None),
+                    kwargs.get("image7", None),
+                    kwargs.get("image8", None),
+                    kwargs.get("image9", None),
+                    kwargs.get("image10", None),
+                ],
+                1,
+            ):
+                if img is not None:
+                    # 都当作PNG就行
+                    b64_data = tensor_to_base64_string(img)
+                    if len(b64_data) > 10 * 1024 * 1024:
+                        raise ValueError(
+                            "Image size is too large, Seedream 4.0 only supports up to 10MB"
+                        )
+                    images.append(f"data:image/png;base64,{b64_data}")
+                    total_size += len(b64_data)
+                    if total_size > 50 * 1024 * 1024:
+                        raise ValueError(
+                            "Total size of images is too large, BizyAir only supports up to 50MB"
+                        )
+
+            data = {
+                "prompt": prompt,
+                "size": sizeStr,
+                "image": images,
+                "model": model,
+                "watermark": False,
+                "response_format": "url",
+            }
+
+            json_payload = json.dumps(data).encode("utf-8")
+            headers = client.headers(api_key=extra_data["api_key"])
+            headers["X-BIZYAIR-PROMPT-ID"] = extra_data["prompt_id"]
+            resp = client.send_request(
+                url=url,
+                data=json_payload,
+                headers=headers,
+            )
+
+            # 结果会包含图片URL，客户端这里负责下载
+            if not "data" in resp:
+                raise ValueError(f"Invalid response: {resp}")
+            if not "url" in resp["data"][0]:
+                raise ValueError(f"Invalid response: {resp}")
+
+            image_data = download_png(resp["data"][0]["url"])
+            return (bytesio_to_image_tensor(io.BytesIO(image_data)),)
+
+        except Exception as e:
+            logging.error(f"Seedream 4.0 API error: {e}")
+            raise e

bizyengine/version.txt

@@ -1 +1 @@
-1.2.51
+1.2.53

{bizyengine-1.2.51.dist-info → bizyengine-1.2.53.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bizyengine
-Version: 1.2.51
+Version: 1.2.53
 Summary: [a/BizyAir](https://github.com/siliconflow/BizyAir) Comfy Nodes that can run in any environment.
 Author-email: SiliconFlow <yaochi@siliconflow.cn>
 Project-URL: Repository, https://github.com/siliconflow/BizyAir

{bizyengine-1.2.51.dist-info → bizyengine-1.2.53.dist-info}/RECORD

@@ -1,5 +1,5 @@
 bizyengine/__init__.py,sha256=GP9V-JM07fz7uv_qTB43QEA2rKdrVJxi5I7LRnn_3ZQ,914
-bizyengine/version.txt,sha256=AvxtajeVm2SVm3pKmMIouv9qTZ-GSKH6a35TmY6KQVo,7
+bizyengine/version.txt,sha256=waYXUIfzzYTFbZKp7xU3Dgr3yzqZ1XpUChdF3MlPTj0,7
 bizyengine/bizy_server/__init__.py,sha256=SP9oSblnPo4KQyh7yOGD26YCskFAcQHAZy04nQBNRIw,200
 bizyengine/bizy_server/api_client.py,sha256=Z7G5IjaEqSJkF6nLLw2R3bpgBAOi5ClQiUbel6NMXmE,43932
 bizyengine/bizy_server/errno.py,sha256=RIyvegX3lzpx_1L1q2XVvu3on0kvYgKiUQ8U3ZtyF68,16823
@@ -10,7 +10,7 @@ bizyengine/bizy_server/resp.py,sha256=iOFT5Ud7VJBP2uqkojJIgc3y2ifMjjEXoj0ewneL9l
 bizyengine/bizy_server/server.py,sha256=isOzHDk2kD6WjdlemeOA7j_xLnZ2vat_NvE1I0bsOFw,57490
 bizyengine/bizy_server/stream_response.py,sha256=H2XHqlVRtQMhgdztAuG7l8-iV_Pm42u2x6WJ0gNVIW0,9654
 bizyengine/bizy_server/utils.py,sha256=Kkn-AATZcdaDhg8Rg_EJW6aKqkyiSE2EYmuyOhUwXso,3863
-bizyengine/bizyair_extras/__init__.py,sha256=lsGuqMnFX-WSOUszSQ5NB6FnKcHlhhO1WG9iiywWQug,1092
+bizyengine/bizyair_extras/__init__.py,sha256=kZ8LQIcdc5mHqqLCrqJi0flIbDr9PtCX0C6TX8El6QQ,1122
 bizyengine/bizyair_extras/nodes_advanced_refluxcontrol.py,sha256=cecfjrtnjJAty9aNkhz8BlmHUC1NImkFlUDiA0COEa4,2242
 bizyengine/bizyair_extras/nodes_cogview4.py,sha256=Ni0TDOycczyDhYPvSR68TxGV_wE2uhaxd8MIj-J4-3o,2031
 bizyengine/bizyair_extras/nodes_comfyui_detail_daemon.py,sha256=i71it24tiGvZ3h-XFWISr4CpZszUtPuz3UrZARYluLk,6169
@@ -22,7 +22,7 @@ bizyengine/bizyair_extras/nodes_custom_sampler.py,sha256=NK-7sdcp8oxJisjTEFfBskk
 bizyengine/bizyair_extras/nodes_dataset.py,sha256=htF0YZb_FHncLhLDEbJfNCVqJ6rvlo1ZLk7iY42Rylc,3440
 bizyengine/bizyair_extras/nodes_differential_diffusion.py,sha256=nSrbD-w0XtrwktwzME5M0Vmi1sI7Z08AqwgymTdThqo,370
 bizyengine/bizyair_extras/nodes_flux.py,sha256=ls94kGBuBNgW5c6uhG36iZLk1TTM2TIoTTcpERgEE5E,2683
-bizyengine/bizyair_extras/nodes_gemini.py,sha256=AdHMxVCYuv7HTBEnxfqSIGLsmICWM5WhDvsLdJWLIqI,10594
+bizyengine/bizyair_extras/nodes_gemini.py,sha256=g1LBpAs5vbORJJZjxt0iv5uNvs8mOQL_-88JL_wA5MY,10602
 bizyengine/bizyair_extras/nodes_hunyuan3d.py,sha256=dWHLeqX68N7zKnfDMzm9nutmCNtFT6-wwt7P5cPDu7Q,2058
 bizyengine/bizyair_extras/nodes_image_utils.py,sha256=BldF_CKD2M01K8-SnG-QV86u3HZqFz_GP5GrCQ5CFDQ,2875
 bizyengine/bizyair_extras/nodes_ip2p.py,sha256=GSEFJvrs4f2tv0xwYkWqc8uhsXrzAJVPvvwcw0gTjR0,619
@@ -32,6 +32,7 @@ bizyengine/bizyair_extras/nodes_nunchaku.py,sha256=iI1Qe_5fQ_43QdUS_reCIOyRFv9-B
 bizyengine/bizyair_extras/nodes_reactor.py,sha256=hbm0HGQWAr_qMcWjp0nMZMtF4mUjTfPw8yK1rq1miAI,8157
 bizyengine/bizyair_extras/nodes_sam2.py,sha256=JTrB7ELwhw6_uOjykRWK9KyOqpYoOtJiGKMxFUbHnNQ,10554
 bizyengine/bizyair_extras/nodes_sd3.py,sha256=lZCxj0IFmuxk1fZTDcRKgVV5QWHjkUdpR4w9-DZbMf4,1727
+bizyengine/bizyair_extras/nodes_seedream.py,sha256=-I22lrdR0S6oRCgN7e5HD8pOJhuGBxDMlhX2S-CyOhY,6855
 bizyengine/bizyair_extras/nodes_segment_anything.py,sha256=x1ei2UggHnB8T6aUtK_ZcUehMALEyLUnDoD5SNJCbFU,7249
 bizyengine/bizyair_extras/nodes_segment_anything_utils.py,sha256=ZefAqrFrevDH3XY_wipr_VwKfeXrgpZEUFaqg_JGOdU,4714
 bizyengine/bizyair_extras/nodes_testing_utils.py,sha256=lYmcyCIkTkQ7WOZfpEPU9wUbEvC_mL6_A46ks68WzZA,3988
@@ -91,7 +92,7 @@ bizyengine/misc/route_sam.py,sha256=-bMIR2QalfnszipGxSxvDAHGJa5gPSrjkYPb5baaRg4,
 bizyengine/misc/segment_anything.py,sha256=wNKYwlYPMszfwj23524geFZJjZaG4eye65SGaUnh77I,8941
 bizyengine/misc/supernode.py,sha256=STN9gaxfTSErH8OiHeZa47d8z-G9S0I7fXuJvHQOBFM,4532
 bizyengine/misc/utils.py,sha256=CKduySGSMNGlJMImHyZmN-giABY5VUaB88f6Kq-HAV0,6831
-bizyengine-1.2.51.dist-info/METADATA,sha256=ubIks74xsnM0PG1697BjTcGvw-LxY5gaG-btFI8-cUk,734
-bizyengine-1.2.51.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-bizyengine-1.2.51.dist-info/top_level.txt,sha256=2zapzqxX-we5cRyJkGf9bd5JinRtXp3-_uDI-xCAnc0,11
-bizyengine-1.2.51.dist-info/RECORD,,
+bizyengine-1.2.53.dist-info/METADATA,sha256=Ps912cD2D6qm1ao5BT8jpr8QkG6RWE8e14qhe-oYfiI,734
+bizyengine-1.2.53.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+bizyengine-1.2.53.dist-info/top_level.txt,sha256=2zapzqxX-we5cRyJkGf9bd5JinRtXp3-_uDI-xCAnc0,11
+bizyengine-1.2.53.dist-info/RECORD,,