video.py

"""Video generation via NovAI (Doubao Seedance 2.0, async submit + poll).

    pip install requests
    export NOVAI_API_KEY=sk-...
    python video.py
"""
import os
import time
import requests

BASE = "https://aiapi-pro.com/v1"
MODEL = "doubao-seedance-2.0"          # or doubao-seedance-2.0-fast
HEADERS = {
    "Authorization": f"Bearer {os.environ['NOVAI_API_KEY']}",
    "Content-Type": "application/json",
}

# 1) submit the generation job
job = requests.post(
    f"{BASE}/video/generations",
    headers=HEADERS,
    json={
        "model": MODEL,
        "prompt": "a paper plane flying over a neon city at night, cinematic",
        "resolution": "720p",
        "duration": 5,
    },
).json()
task_id = job["id"]
print("submitted:", task_id)

# 2) poll until the video is ready
for _ in range(90):
    r = requests.get(
        f"{BASE}/video/generations/{task_id}",
        headers=HEADERS,
        params={"model": MODEL},
    ).json()
    status = r.get("status")
    if status == "succeeded":
        content = r.get("content", {})
        print("video:", content.get("video_url") or content.get("url"))
        break
    if status == "failed":
        raise RuntimeError(r)
    print("...", status)
    time.sleep(4)