bizyengine 1.2.58__py3-none-any.whl → 1.2.71__py3-none-any.whl

bizyengine/bizyair_extras/nodes_seedream.py

@@ -1,195 +0,0 @@
-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/bizyair_extras/nodes_wan_api.py

@@ -1,291 +0,0 @@
-import io
-import json
-import logging
-
-import requests
-from comfy_api.latest._input_impl import VideoFromFile
-from comfy_api_nodes.apinode_utils import tensor_to_bytesio
-
-from bizyengine.bizyair_extras.utils.audio import save_audio
-from bizyengine.core import BizyAirBaseNode, pop_api_key_and_prompt_id
-from bizyengine.core.common import client
-from bizyengine.core.common.client import send_request
-from bizyengine.core.common.env_var import BIZYAIR_X_SERVER
-
-from .utils.aliyun_oss import upload_file_without_sdk
-
-
-def parse_upload_token(resp) -> dict:
-    logging.debug(f"parsing token resp: {resp}")
-    if "data" not in resp:
-        logging.error(f"Invalid response, data not found: {resp}")
-        raise ValueError(f"Invalid response: {resp}")
-    data = resp["data"]
-    if "file" not in data:
-        logging.error(f"Invalid response, file not found: {resp}")
-        raise ValueError(f"Invalid response: {resp}")
-    file = data["file"]
-    if "storage" not in data:
-        logging.error(f"Invalid response, storage not found: {resp}")
-        raise ValueError(f"Invalid response: {resp}")
-    storage = data["storage"]
-    return file | storage
-
-
-class Wan_V2_5_I2V_API(BizyAirBaseNode):
-    @classmethod
-    def INPUT_TYPES(cls):
-        return {
-            "required": {
-                "image": ("IMAGE",),
-            },
-            "optional": {
-                "audio": ("AUDIO",),
-                "prompt": (
-                    "STRING",
-                    {
-                        "multiline": True,
-                        "default": "",
-                    },
-                ),
-                "negative_prompt": (
-                    "STRING",
-                    {
-                        "multiline": True,
-                        "default": "",
-                    },
-                ),
-                "resolution": (
-                    ["480P", "720P", "1080P"],
-                    {"default": "1080P"},
-                ),
-                "duration": ([5, 10], {"default": 5}),
-                "prompt_extend": (
-                    "BOOLEAN",
-                    {
-                        "default": True,
-                        "tooltip": "是否开启prompt智能改写。开启后使用大模型对输入prompt进行智能改写。对于较短的prompt生成效果提升明显，但会增加耗时。",
-                    },
-                ),
-                "auto_audio": (
-                    "BOOLEAN",
-                    {
-                        "default": True,
-                        "tooltip": "是否由模型自动生成声音，优先级低于audio参数。",
-                    },
-                ),
-            },
-        }
-
-    NODE_DISPLAY_NAME = "Wan2.5 Image To Video"
-    RETURN_TYPES = ("VIDEO", "STRING")
-    RETURN_NAMES = ("video", "actual_prompt")
-    CATEGORY = "☁️BizyAir/External APIs/WanVideo"
-    FUNCTION = "api_call"
-
-    def api_call(self, image, **kwargs):
-        extra_data = pop_api_key_and_prompt_id(kwargs)
-        headers = client.headers(api_key=extra_data["api_key"])
-        prompt_id = extra_data["prompt_id"]
-        headers["X-BIZYAIR-PROMPT-ID"] = prompt_id
-
-        # 参数
-        prompt = kwargs.get("prompt", "")
-        negative_prompt = kwargs.get("negative_prompt", "")
-        audio = kwargs.get("audio", None)
-        resolution = kwargs.get("resolution", "1080P")
-        duration = kwargs.get("duration", 5)
-        prompt_extend = kwargs.get("prompt_extend", True)
-        auto_audio = kwargs.get("auto_audio", True)
-
-        input = {}
-        if prompt is not None and prompt.strip() != "":
-            input["prompt"] = prompt
-        if negative_prompt is not None and negative_prompt.strip() != "":
-            input["negative_prompt"] = negative_prompt
-
-        # 上传图片&音频
-        if image is not None:
-            oss_token_url = f"{BIZYAIR_X_SERVER}/upload/token?file_name={prompt_id}.png&file_type=inputs"
-            token_resp = send_request("GET", oss_token_url, headers=headers)
-            auth_info = parse_upload_token(token_resp)
-            input["img_url"] = upload_file_without_sdk(
-                tensor_to_bytesio(image), **auth_info
-            )
-        if audio is not None:
-            oss_token_url = f"{BIZYAIR_X_SERVER}/upload/token?file_name={prompt_id}.flac&file_type=inputs"
-            token_resp = send_request("GET", oss_token_url, headers=headers)
-            auth_info = parse_upload_token(token_resp)
-            audio_bytes = save_audio(audio)
-            input["audio_url"] = upload_file_without_sdk(audio_bytes, **auth_info)
-
-        # 调用API
-        model = "wan2.5-i2v-preview"
-        api_url = f"{BIZYAIR_X_SERVER}/proxy_inference/Wan/{model}"
-        data = {
-            "model": model,
-            "input": input,
-            "parameters": {
-                "resolution": resolution,
-                "prompt_extend": prompt_extend,
-                "duration": duration,
-                "audio": auto_audio,
-            },
-        }
-        json_payload = json.dumps(data).encode("utf-8")
-        logging.debug(f"json_payload: {json_payload}")
-        api_resp = send_request(
-            url=api_url,
-            data=json_payload,
-            headers=headers,
-        )
-        logging.debug(f"api resp: {api_resp}")
-        if "output" not in api_resp or "video_url" not in api_resp["output"]:
-            raise ValueError(f"Invalid response: {api_resp}")
-        video_url = api_resp["output"]["video_url"]
-        logging.info(f"video_url: {video_url}")
-        actual_prompt = api_resp["output"].get("actual_prompt", prompt)
-        # 下载视频
-        video_resp = requests.get(video_url, stream=True, timeout=3600)
-        video_resp.raise_for_status()  # 非 2xx 会抛异常
-        return (VideoFromFile(io.BytesIO(video_resp.content)), actual_prompt)
-
-
-class Wan_V2_5_T2V_API(BizyAirBaseNode):
-    @classmethod
-    def INPUT_TYPES(cls):
-        return {
-            "required": {
-                "prompt": (
-                    "STRING",
-                    {
-                        "multiline": True,
-                        "default": "",
-                    },
-                ),
-            },
-            "optional": {
-                "audio": ("AUDIO",),
-                "negative_prompt": (
-                    "STRING",
-                    {
-                        "multiline": True,
-                        "default": "",
-                    },
-                ),
-                "size": (
-                    [
-                        "832*480",
-                        "480*832",
-                        "624*624",
-                        "1280*720",
-                        "720*1280",
-                        "960*960",
-                        "1088*832",
-                        "832*1088",
-                        "1920*1080",
-                        "1080*1920",
-                        "1440*1440",
-                        "1632*1248",
-                        "1248*1632",
-                    ],
-                    {"default": "1920*1080"},
-                ),
-                "duration": ([5, 10], {"default": 5}),
-                "prompt_extend": (
-                    "BOOLEAN",
-                    {
-                        "default": True,
-                        "tooltip": "是否开启prompt智能改写。开启后使用大模型对输入prompt进行智能改写。对于较短的prompt生成效果提升明显，但会增加耗时。",
-                    },
-                ),
-                "auto_audio": (
-                    "BOOLEAN",
-                    {
-                        "default": True,
-                        "tooltip": "是否由模型自动生成声音，优先级低于audio参数。",
-                    },
-                ),
-            },
-        }
-
-    NODE_DISPLAY_NAME = "Wan2.5 Text To Video"
-    RETURN_TYPES = ("VIDEO", "STRING")
-    RETURN_NAMES = ("video", "actual_prompt")
-    CATEGORY = "☁️BizyAir/External APIs/WanVideo"
-    FUNCTION = "api_call"
-
-    def parse_upload_token(self, resp) -> dict:
-        logging.debug(f"parsing token resp: {resp}")
-        if "data" not in resp:
-            logging.error(f"Invalid response, data not found: {resp}")
-            raise ValueError(f"Invalid response: {resp}")
-        data = resp["data"]
-        if "file" not in data:
-            logging.error(f"Invalid response, file not found: {resp}")
-            raise ValueError(f"Invalid response: {resp}")
-        file = data["file"]
-        if "storage" not in data:
-            logging.error(f"Invalid response, storage not found: {resp}")
-            raise ValueError(f"Invalid response: {resp}")
-        storage = data["storage"]
-        return file | storage
-
-    def api_call(self, prompt, **kwargs):
-        extra_data = pop_api_key_and_prompt_id(kwargs)
-        headers = client.headers(api_key=extra_data["api_key"])
-        prompt_id = extra_data["prompt_id"]
-        headers["X-BIZYAIR-PROMPT-ID"] = prompt_id
-
-        # 参数
-        negative_prompt = kwargs.get("negative_prompt", "")
-        audio = kwargs.get("audio", None)
-        size = kwargs.get("size", "1920*1080")
-        duration = kwargs.get("duration", 5)
-        prompt_extend = kwargs.get("prompt_extend", True)
-        auto_audio = kwargs.get("auto_audio", True)
-
-        input = {}
-        if prompt is not None and prompt.strip() != "":
-            input["prompt"] = prompt
-        if negative_prompt is not None and negative_prompt.strip() != "":
-            input["negative_prompt"] = negative_prompt
-
-        # 上传音频
-        if audio is not None:
-            oss_token_url = f"{BIZYAIR_X_SERVER}/upload/token?file_name={prompt_id}.flac&file_type=inputs"
-            token_resp = send_request("GET", oss_token_url, headers=headers)
-            auth_info = parse_upload_token(token_resp)
-            audio_bytes = save_audio(audio)
-            input["audio_url"] = upload_file_without_sdk(audio_bytes, **auth_info)
-
-        # 调用API
-        model = "wan2.5-t2v-preview"
-        api_url = f"{BIZYAIR_X_SERVER}/proxy_inference/Wan/{model}"
-        data = {
-            "model": model,
-            "input": input,
-            "parameters": {
-                "size": size,
-                "prompt_extend": prompt_extend,
-                "duration": duration,
-                "audio": auto_audio,
-            },
-        }
-        json_payload = json.dumps(data).encode("utf-8")
-        logging.debug(f"json_payload: {json_payload}")
-        api_resp = send_request(
-            url=api_url,
-            data=json_payload,
-            headers=headers,
-        )
-        logging.debug(f"api resp: {api_resp}")
-        if "output" not in api_resp or "video_url" not in api_resp["output"]:
-            raise ValueError(f"Invalid response: {api_resp}")
-        video_url = api_resp["output"]["video_url"]
-        logging.info(f"video_url: {video_url}")
-        actual_prompt = api_resp["output"].get("actual_prompt", prompt)
-        # 下载视频
-        video_resp = requests.get(video_url, stream=True, timeout=3600)
-        video_resp.raise_for_status()  # 非 2xx 会抛异常
-        return (VideoFromFile(io.BytesIO(video_resp.content)), actual_prompt)