快速入门：参考素材生成视频

#!/usr/bin/env bash
set -euo pipefail

response=$(curl --fail-with-body -sS "https://apis.viggle.ai/v1/videos" \
  -H "Authorization: Bearer $VIGGLE_API_KEY" \
  -F prompt="keep the same outfit and walk forward" \
  -F quality=high \
  -F reference_video=@./reference.mp4 \
  -F duration_s=5)
video_id=$(echo "$response" | jq -er '.id')

for _ in $(seq 1 60); do
  video=$(curl --fail-with-body -sS "https://apis.viggle.ai/v1/videos/$video_id" \
    -H "Authorization: Bearer $VIGGLE_API_KEY")
  status=$(echo "$video" | jq -r '.status')

  case "$status" in
    ready)
      echo "$video" | jq -r '.video_url'
      exit 0
      ;;
    failed|cancelled)
      echo "Video $status: $(echo "$video" | jq -c '.error')" >&2
      exit 1
      ;;
  esac
  sleep 5
done

echo "Timed out waiting for video $video_id" >&2
exit 1

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

const baseUrl = "https://apis.viggle.ai/v1";
const headers = { Authorization: `Bearer ${process.env.VIGGLE_API_KEY}` };
const TERMINAL = new Set(["ready", "failed", "cancelled"]);

async function createVideo() {
  const form = new FormData();
  form.append("prompt", "keep the same outfit and walk forward");
  form.append("quality", "high");
  form.append("reference_video", new Blob([await readFile("./reference.mp4")], { type: "video/mp4" }), "reference.mp4");
  form.append("duration_s", "5");

  const response = await fetch(`${baseUrl}/videos`, {
    method: "POST",
    headers,
    body: form,
  });
  if (!response.ok) {
    const body = await response.json().catch(() => ({}));
    throw new Error(`HTTP ${response.status}: ${JSON.stringify(body.error ?? body)}`);
  }
  return response.json();
}

async function waitForVideo(id) {
  for (let attempt = 0; attempt < 60; attempt++) {
    const response = await fetch(`${baseUrl}/videos/${id}`, { headers });
    if (!response.ok) {
      const body = await response.json().catch(() => ({}));
      throw new Error(`HTTP ${response.status}: ${JSON.stringify(body.error ?? body)}`);
    }
    const video = await response.json();
    if (TERMINAL.has(video.status)) return video;
    await new Promise((resolve) => setTimeout(resolve, 5000));
  }
  throw new Error(`Timed out waiting for video ${id}`);
}

const created = await createVideo();
const video = await waitForVideo(created.id);

if (video.status !== "ready") {
  throw new Error(`Video ${video.status}: ${JSON.stringify(video.error)}`);
}
console.log(video.video_url);

import os
import time
import requests

base_url = "https://apis.viggle.ai/v1"
headers = {"Authorization": f"Bearer {os.environ['VIGGLE_API_KEY']}"}
TERMINAL_STATES = {"ready", "failed", "cancelled"}

with open("./reference.mp4", "rb") as reference_video:
    response = requests.post(
        f"{base_url}/videos",
        headers=headers,
        data={
            "prompt": "keep the same outfit and walk forward",
            "quality": "high",
            "duration_s": 5,
        },
        files={"reference_video": ("reference.mp4", reference_video, "video/mp4")},
        timeout=120,
    )
response.raise_for_status()
video_id = response.json()["id"]

for _ in range(60):
    response = requests.get(f"{base_url}/videos/{video_id}", headers=headers, timeout=30)
    response.raise_for_status()
    video = response.json()
    if video["status"] in TERMINAL_STATES:
        break
    time.sleep(5)
else:
    raise TimeoutError(f"Timed out waiting for video {video_id}")

if video["status"] != "ready":
    error = video.get("error") or {}
    raise RuntimeError(f"{video['status']}: {error.get('code')} {error.get('message')}")

print(video["video_url"])