Images API: генерация и редактирование изображений через LaoZhang API

export LAOZHANG_API_KEY="YOUR_LAOZHANG_API_KEY"

python -m pip install openai pillow httpx

"""Сохранение всех изображений из ответа OpenAI Images API."""
import base64
import io
import json
import sys
from pathlib import Path

import httpx
from PIL import Image

EXTENSIONS = {"PNG": "png", "JPEG": "jpg", "WEBP": "webp"}


def image_bytes(item):
    """Возвращает байты изображения из одного элемента data[]."""
    encoded = item.get("b64_json")
    if encoded:
        encoded = "".join(encoded.split())
        if encoded.startswith("data:"):
            encoded = encoded.split(",", 1)[1]  # убираем префикс data:image/...;base64,
        encoded += "=" * (-len(encoded) % 4)  # восстанавливаем выравнивание
        return base64.b64decode(encoded)
    if item.get("url"):
        # Скачиваем результат, не отправляя свой API-ключ LaoZhang API.
        download = httpx.get(item["url"], timeout=120, follow_redirects=True)
        download.raise_for_status()
        return download.content
    raise ValueError("В элементе data[] нет ни b64_json, ни url")


def save_images(response, stem):
    """Сохраняет каждое изображение как <stem>.<ext>, затем <stem>-2.<ext> и так далее."""
    entries = response.get("data") or []
    if not entries:
        raise RuntimeError(f"В ответе нет изображений: {json.dumps(response)[:500]}")
    for number, item in enumerate(entries, start=1):
        raw = image_bytes(item)
        with Image.open(io.BytesIO(raw)) as image:
            image.load()
            extension = EXTENSIONS.get(image.format, (image.format or "bin").lower())
            size = f"{image.width} × {image.height}"
        suffix = "" if number == 1 else f"-{number}"
        output = Path(f"{stem}{suffix}.{extension}")
        output.write_bytes(raw)
        print(f"Сохранено: {output} ({size})")


if __name__ == "__main__":
    save_images(json.loads(Path(sys.argv[1]).read_text(encoding="utf-8")), sys.argv[2])

POST https://api.laozhang.ai/v1/images/generations

POST https://api.laozhang.ai/v1/images/edits