Doubao · 生成

# 安装依赖：python -m pip install requests
import base64
import json
import os
from pathlib import Path
from urllib.parse import quote
import requests

BASE_URL = "https://ai.kedang.net"
MODEL_ID = "MODEL_ID"  # 替换为账户可用、支持当前能力的模型


def request_json(path, body, extra_headers=None):
    headers = {"Authorization": "Bearer " + os.environ["KEDANG_API_KEY"]}
    headers.update(extra_headers or {})
    response = requests.post(BASE_URL + path, headers=headers, json=body, timeout=120)
    response.raise_for_status()
    result = response.json()
    if result.get("error"):
        raise RuntimeError(result["error"])
    return result

import tempfile
from urllib.parse import urlsplit


def image_extension(content):
    if content.startswith(b"\x89PNG\r\n\x1a\n"):
        return ".png"
    if content.startswith(b"\xff\xd8\xff"):
        return ".jpg"
    if content[:4] == b"RIFF" and content[8:12] == b"WEBP":
        return ".webp"
    return ".bin"  # 其他编码先按原始字节保存，再按实际格式命名。


def save_images(result, directory="generated-images"):
    items = result.get("data")
    if not isinstance(items, list) or not items:
        raise ValueError("响应没有 data 图片列表")
    folder = Path(directory)
    folder.mkdir(parents=True, exist_ok=True)
    paths = []
    for index, item in enumerate(items, 1):
        if item.get("b64_json"):
            content = base64.b64decode(item["b64_json"], validate=True)
        elif item.get("url"):
            url = item["url"]
            parsed = urlsplit(url)
            if parsed.scheme != "https" or not parsed.netloc:
                raise ValueError("图片结果需为有效 HTTPS 地址")
            # 媒体地址可能在独立域名；这里不携带平台 API Key。
            response = requests.get(url, timeout=120)
            response.raise_for_status()
            mime = response.headers.get("Content-Type", "").split(";", 1)[0].lower()
            if mime and not (mime.startswith("image/") or mime == "application/octet-stream"):
                raise ValueError("下载响应不是图片或二进制文件")
            content = response.content
        else:
            raise ValueError("图片项缺少 url 或 b64_json")
        if not content:
            raise ValueError("图片内容为空")
        target = folder / ("image-" + str(index) + image_extension(content))
        temporary = None
        try:
            with tempfile.NamedTemporaryFile(dir=folder, delete=False) as file:
                temporary = Path(file.name)
                file.write(content)
            temporary.replace(target)
        finally:
            if temporary is not None:
                temporary.unlink(missing_ok=True)
        paths.append(target)
    return paths

def main():
    result = request_json("/v1/images/generations", {
        "model": MODEL_ID, "prompt": "一只橘猫坐在窗边，柔和自然光。", "n": 1})
    for path in save_images(result):
        print(path)

if __name__ == "__main__":
    main()