bizyengine 1.2.51__py3-none-any.whl → 1.2.52__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_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.52

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bizyengine
-Version: 1.2.51
+Version: 1.2.52
 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.52.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=IjisjQCOb46a60B43ZZzkG9k1UlLoV_j4Wz2PTTdMbM,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
@@ -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.52.dist-info/METADATA,sha256=z4iUCltSLd23LQ2kawCWf1pCDyTqbu81DQNglRCb4AM,734
+bizyengine-1.2.52.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+bizyengine-1.2.52.dist-info/top_level.txt,sha256=2zapzqxX-we5cRyJkGf9bd5JinRtXp3-_uDI-xCAnc0,11
+bizyengine-1.2.52.dist-info/RECORD,,