Sora 2 视频生成

curl --request POST \
  --url https://geekapis.com/v1/video/generations \
  --header "Authorization: Bearer <YOUR_API_KEY>" \
  --header "Content-Type: application/json" \
  --data '{
    "model": "sora2",
    "prompt": "一只金毛犬在草地上奔跑，阳光明媚",
    "duration": 12,
    "size": "1280x720"
  }'

curl --request POST \
  --url https://geekapis.com/v1/video/generations \
  --header "Authorization: Bearer <YOUR_API_KEY>" \
  --header "Content-Type: application/json" \
  --data '{
    "model": "sora2",
    "prompt": "让参考图中的元素轻微运动，保持原有构图和文字不变",
    "duration": 4,
    "size": "1280x720",
    "input_reference": "https://placehold.co/1280x720.png"
  }'

import time
import requests

API_KEY = "<YOUR_API_KEY>"
BASE_URL = "https://geekapis.com/v1"
HEADERS = {
    "Authorization": f"Bearer {API_KEY}",
    "Content-Type": "application/json",
}

def submit_and_poll(payload: dict):
    create_resp = requests.post(
        f"{BASE_URL}/video/generations",
        headers=HEADERS,
        json=payload,
        timeout=60,
    )
    create_resp.raise_for_status()
    task = create_resp.json()
    print("create:")
    print(task)

    task_id = task["task_id"]

    while True:
        status_resp = requests.get(
            f"{BASE_URL}/video/generations/{task_id}",
            headers={"Authorization": f"Bearer {API_KEY}"},
            timeout=60,
        )
        status_resp.raise_for_status()
        result = status_resp.json()
        print("status:")
        print(result)

        status = result["data"]["status"]
        if status == "SUCCESS":
            print("video_url:", result["data"]["result_url"])
            break
        if status == "FAILURE":
            print("failed:", result["data"]["fail_reason"])
            break

        time.sleep(5)

# 文生视频
submit_and_poll({
    "model": "sora2",
    "prompt": "一只金毛犬在草地上奔跑，阳光明媚",
    "duration": 12,
    "size": "1280x720",
})

# 图生视频
# submit_and_poll({
#     "model": "sora2",
#     "prompt": "让参考图中的元素轻微运动，保持原有构图和文字不变",
#     "duration": 4,
#     "size": "1280x720",
#     "input_reference": "https://placehold.co/1280x720.png",
# })

const API_KEY = "<YOUR_API_KEY>";
const BASE_URL = "https://geekapis.com/v1";

function sleep(ms) {
  return new Promise((resolve) => setTimeout(resolve, ms));
}

async function submitAndPoll(payload) {
  const createResp = await fetch(`${BASE_URL}/video/generations`, {
    method: "POST",
    headers: {
      Authorization: `Bearer ${API_KEY}`,
      "Content-Type": "application/json",
    },
    body: JSON.stringify(payload),
  });

  if (!createResp.ok) {
    throw new Error(await createResp.text());
  }

  const task = await createResp.json();
  console.log("create:");
  console.log(task);

  const taskId = task.task_id;

  while (true) {
    const statusResp = await fetch(`${BASE_URL}/video/generations/${taskId}`, {
      headers: {
        Authorization: `Bearer ${API_KEY}`,
      },
    });

    if (!statusResp.ok) {
      throw new Error(await statusResp.text());
    }

    const result = await statusResp.json();
    console.log("status:");
    console.log(result);

    const status = result.data.status;
    if (status === "SUCCESS") {
      console.log("video_url:", result.data.result_url);
      break;
    }
    if (status === "FAILURE") {
      console.log("failed:", result.data.fail_reason);
      break;
    }

    await sleep(5000);
  }
}

// 文生视频
submitAndPoll({
  model: "sora2",
  prompt: "一只金毛犬在草地上奔跑，阳光明媚",
  duration: 12,
  size: "1280x720",
});

// 图生视频
// submitAndPoll({
//   model: "sora2",
//   prompt: "让参考图中的元素轻微运动，保持原有构图和文字不变",
//   duration: 4,
//   size: "1280x720",
//   input_reference: "https://placehold.co/1280x720.png",
// }).catch(console.error);