GPT Image 2

import json
import os
from urllib.request import Request, HTTPRedirectHandler, build_opener

class NoRedirect(HTTPRedirectHandler):
    def redirect_request(self, req, fp, code, msg, headers, newurl):
        return None

opener = build_opener(NoRedirect)

def post(url, payload):
    request = Request(url, data=json.dumps(payload).encode("utf-8"), headers={
        "Authorization": "Bearer " + os.environ["KVANTORA_API_KEY"],
        "Content-Type": "application/json",
        "Idempotency-Key": os.environ["REQUEST_ID"],
    }, method="POST")
    with opener.open(request, timeout=120) as response:
        return json.load(response)

def get(url):
    request = Request(url, headers={
        "Authorization": "Bearer " + os.environ["KVANTORA_API_KEY"],
    }, method="GET")
    with opener.open(request, timeout=120) as response:
        return json.load(response)

def format_rubles(value):
    amount = int(value)
    sign, absolute = ("-" if amount < 0 else ""), abs(amount)
    if 0 < absolute < 10_000:
        return sign + "<0,01 ₽"
    cents = (absolute + 5_000) // 10_000
    return f"{sign}{cents // 100},{cents % 100:02d} ₽"

quote = post("https://api.kvantora.ai/v1/quotes", {
  "model": "openai/gpt-image-2",
  "prompt": "Иллюстрация для статьи об API, без надписей",
  "n": 1,
  "settings": {
    "n": 1,
    "quality": "medium",
    "resolution": "1024x1024"
  },
  "operation": "image_generation"
})
print("Стоимость:", format_rubles(quote["estimated_cost_microrub"]))

# Укажите допустимый бюджет вместо max_cost_microrub: 0.
payload = {
  "model": "openai/gpt-image-2",
  "prompt": "Иллюстрация для статьи об API, без надписей",
  "n": 1,
  "settings": {
    "n": 1,
    "quality": "medium",
    "resolution": "1024x1024"
  },
  "max_cost_microrub": 0
}
job = post("https://api.kvantora.ai/v1/images/generations", payload)
print(get("https://api.kvantora.ai/v1/jobs" + "/" + job["id"]))