P-Video 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": "pruna-ai/p-video-2/text-to-video",
  "prompt": "Плавная заставка с геометрическими формами, без текста",
  "settings": {
    "seed": -1,
    "draft": false,
    "duration": "5",
    "resolution": "720p",
    "save_audio": true,
    "aspect_ratio": "16:9",
    "prompt_upsampling": true
  },
  "operation": "video_generation"
})
print("Стоимость:", format_rubles(quote["estimated_cost_microrub"]))

# Укажите допустимый бюджет вместо max_cost_microrub: 0.
payload = {
  "model": "pruna-ai/p-video-2/text-to-video",
  "prompt": "Плавная заставка с геометрическими формами, без текста",
  "settings": {
    "seed": -1,
    "draft": false,
    "duration": "5",
    "resolution": "720p",
    "save_audio": true,
    "aspect_ratio": "16:9",
    "prompt_upsampling": true
  },
  "max_cost_microrub": 0
}
job = post("https://api.kvantora.ai/v1/videos", payload)
print(get("https://api.kvantora.ai/v1/jobs" + "/" + job["id"]))