Video generation

import os
import time

import requests

BASE = "https://api.naga.ac/v1"
HEADERS = {"Authorization": f"Bearer {os.environ['NAGA_API_KEY']}"}

response = requests.post(
    f"{BASE}/videos",
    headers=HEADERS,
    json={
        "model": "kling-v3.0-pro",
        "prompt": "A red paper boat drifting on a calm lake at dawn",
        "duration": 8,
        "resolution": "720p",
        "aspect_ratio": "16:9",
    },
)
response.raise_for_status()
job = response.json()

while job["status"] in ("pending", "in_progress"):
    time.sleep(10)
    job = requests.get(job["polling_url"], headers=HEADERS).json()

if job["status"] == "completed":
    video = requests.get(f"{BASE}/videos/{job['id']}/content", headers=HEADERS)
    with open("boat.mp4", "wb") as file:
        file.write(video.content)
else:
    print(job["error"])

import { writeFile } from "node:fs/promises";

const BASE = "https://api.naga.ac/v1";
const headers = { Authorization: `Bearer ${process.env.NAGA_API_KEY}` };

const response = await fetch(`${BASE}/videos`, {
  method: "POST",
  headers: { ...headers, "Content-Type": "application/json" },
  body: JSON.stringify({
    model: "kling-v3.0-pro",
    prompt: "A red paper boat drifting on a calm lake at dawn",
    duration: 8,
    resolution: "720p",
    aspect_ratio: "16:9",
  }),
});
if (!response.ok) throw new Error(await response.text());
let job = await response.json();

while (job.status === "pending" || job.status === "in_progress") {
  await new Promise((resolve) => setTimeout(resolve, 10_000));
  job = await (await fetch(job.polling_url, { headers })).json();
}

if (job.status === "completed") {
  const video = await fetch(`${BASE}/videos/${job.id}/content`, { headers });
  await writeFile("boat.mp4", Buffer.from(await video.arrayBuffer()));
} else {
  console.log(job.error);
}

JOB_ID=$(curl https://api.naga.ac/v1/videos \
  -H "Authorization: Bearer $NAGA_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{"model": "kling-v3.0-pro", "prompt": "A red paper boat drifting on a calm lake at dawn", "duration": 8, "resolution": "720p", "aspect_ratio": "16:9"}' \
  | jq -r .id)

# Poll until "status" is "completed" or "failed".
while true; do
  STATUS=$(curl https://api.naga.ac/v1/videos/$JOB_ID \
    -H "Authorization: Bearer $NAGA_API_KEY" | jq -r .status)
  if [ "$STATUS" = "completed" ] || [ "$STATUS" = "failed" ]; then break; fi
  sleep 10
done

if [ "$STATUS" = "completed" ]; then
  curl -L -o boat.mp4 https://api.naga.ac/v1/videos/$JOB_ID/content \
    -H "Authorization: Bearer $NAGA_API_KEY"
fi