Kie.ai 대안: 그대로 쓸 수 있는 모델과 전환 방법

import os
import time
import requests

API = "https://api.omniakey.com"
HEADERS = {
    "Authorization": f"Bearer {os.environ['OMNIAKEY_API_KEY']}",
    "Content-Type": "application/json",
}

# 1. 작업을 만듭니다. Idempotency-Key를 붙이면 다시 요청해도 안전합니다.
task = requests.post(
    f"{API}/v1/media/tasks",
    headers={**HEADERS, "Idempotency-Key": "switch-from-kie-ai-001"},
    json={
        "model": "gemini-nano-banana-2.1",
        "type": "image",
        "operation": "generate",
        "input": {"prompt": "a paper boat on a quiet lake at sunrise"},
        "parameters": {"aspect_ratio": "1:1", "image_size": "1K"},
    },
).json()

# 2. 작업이 끝날 때까지 폴링합니다.
while task["status"] in ("queued", "processing"):
    time.sleep(3)
    task = requests.get(f"{API}/v1/media/tasks/{task['id']}", headers=HEADERS).json()

if task["status"] != "succeeded":
    raise SystemExit(task.get("error"))

# 3. 서명된 링크가 만료되기 전에 이미지를 내려받습니다.
open("out.jpg", "wb").write(requests.get(task["assets"][0]["url"]).content)
print(task["id"], task["assets"][0]["mime_type"], task["assets"][0]["byte_size"])