Video generation

import time

from openai import OpenAI

client = OpenAI(
    base_url="https://api.goodputai.cn/v1",
    api_key="YOUR_GOODPUT_API_KEY",
)


def generate(**kwargs) -> str:
    """Submit, poll, download; return the saved path. Shared by all modes."""
    video = client.videos.create(**kwargs)
    print(video.id, video.status)          # video_xxx queued

    deadline = time.time() + 15 * 60       # Overall timeout; do not poll indefinitely
    while video.status in ("queued", "in_progress"):
        if time.time() > deadline:
            raise TimeoutError(f"Timed out before completion: {video.id}")
        time.sleep(5)
        video = client.videos.retrieve(video.id)

    if video.status != "completed":
        raise RuntimeError(f"Generation failed: {video.error}")

    # Completed h3 / h3-fast responses also provide a direct download or playback URL
    video_url = getattr(video, "video_url", None)
    print("Video download URL:", video_url)
    print("Link expiry:", getattr(video, "video_url_expires_at", None))

    path = f"{video.id[:16]}.mp4"
    client.videos.download_content(video.id).write_to_file(path)
    print("usage:", video.usage)           # Billing basis
    return path


# Minimal example: text to video
generate(
    model="wan/wan3.0-video",
    prompt="Sunrise by the sea, waves breaking on rocks, cinematic camera movement",
    size="1280x720",
    seconds="5",
)

# 1) Submit; use the id in the response for subsequent status queries
curl https://api.goodputai.cn/v1/videos \
  -H "Authorization: Bearer $GOODPUT_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{
    "model": "wan/wan3.0-video",
    "prompt": "Sunrise by the sea, waves breaking on rocks",
    "size": "1280x720",
    "seconds": "5"
  }'

# 2) Query status and download URL until status becomes completed or failed
curl https://api.goodputai.cn/v1/videos/{video_id} \
  -H "Authorization: Bearer $GOODPUT_API_KEY"

{
  "id": "video_xxx",
  "status": "completed",
  "video_url": "https://example.com/generated-video.mp4?...",
  "video_url_expires_at": null
}

# 3a) Copy the complete video_url from the status response; no API Key is needed
VIDEO_URL="REPLACE_WITH_COMPLETE_VIDEO_URL_FROM_RESPONSE"
curl -fL "$VIDEO_URL" -o output.mp4

# 3b) If video_url is absent or null, use the download endpoint instead (always supported)
curl https://api.goodputai.cn/v1/videos/{video_id}/content \
  -H "Authorization: Bearer $GOODPUT_API_KEY" \
  -o output.mp4

generate(
    model="wan/wan3.0-video",     # Or wan/wan2.7-t2v, wan/wan2.6-t2v
    prompt="Walk forward along a neon street on a rainy night, first-person view",
    size="720x1280",              # Short side 720 → 720P tier; 9:16 → portrait
    seconds="5",
    prompt_extend=False,          # ⚠️ Defaults to true and rewrites your prompt
)

generate(
    model="minimax/h3",
    prompt="Walk forward along a neon street on a rainy night, first-person view. No subtitles or text.",
    size="768x1344",              # 9:16 portrait; see the six canvases below
    seconds=6,                    # Integer from 4~15
)

generate(
    model="wan/wan2.7-i2v",       # Or wan/wan3.0-video
    prompt="The camera slowly pulls back",
    # Put URLs / base64 in extra_body (see the explanation below)
    extra_body={"input_reference": "https://example.com/first.jpg"},
    size="1280x720",              # Sets the tier and unit price; ⚠️ actual aspect ratio follows the image
    seconds="5",
)

# Option one: public URL or inline data: URL, via extra_body
generate(
    model="minimax/h3",
    prompt="The camera slowly pulls back as the person turns naturally",
    extra_body={"input_reference": "https://example.com/first.jpg"},
    size="768x768",
    seconds=4,
)

# Option two: upload a local file directly (supported by h3, not wan)
with open("first.jpg", "rb") as f:
    generate(
        model="minimax/h3",
        prompt="The camera slowly pulls back as the person turns naturally",
        input_reference=f,
        size="768x768",
        seconds=4,
    )

generate(
    model="wan/wan2.7-r2v",       # Or wan/wan3.0-video
    prompt="The same character walking through snow",
    extra_body={
        "media": [
            {"type": "reference_image", "url": "https://example.com/character.jpg"},
        ]
    },
    size="1280x720",
    seconds="5",
)

# h3 accepts multiple reference images in one request
generate(
    model="minimax/h3",
    prompt="In this room, a woman enters from the left and opens a chest",
    extra_body={
        "media": [
            {"type": "reference_image", "url": "https://example.com/room.jpg"},
            {"type": "reference_image", "url": "https://example.com/person.jpg"},
        ]
    },
    size="1344x768",
    seconds=4,
)

generate(
    model="wan/wan3.0-video",
    prompt="Transition from day to night",
    extra_body={
        "media": [
            {"type": "first_frame", "url": "https://example.com/day.jpg"},
            {"type": "last_frame", "url": "https://example.com/night.jpg"},
        ]
    },
    size="1280x720",
    seconds="5",
)

generate(
    model="minimax/h3",
    prompt="The person moves naturally and continuously from the starting pose to the ending pose",
    extra_body={
        "media": [
            {"type": "first_frame", "url": "https://example.com/start.jpg"},
            {"type": "last_frame", "url": "https://example.com/end.jpg"},
        ]
    },
    size="1344x768",
    seconds=4,
)

{
  "duration_seconds": 5,      // Billed seconds; includes reference video input for Wan3.0
  "SR": 480,                  // Actual output resolution tier; determines the unit price
  "ratio": "16:9",
  "fps": 30,
  "video_count": 1
}

{
  "duration_seconds": 4,      // = requested seconds; billed on this value
  "size": "768x768",
  "inference_time_s": 54.8    // Generation time for reference only; not used for billing
}

{
  "duration_seconds": 5,      // = requested seconds (15 returns 14); billed on this value
  "size": "1344x768",
  "inference_time_s": 17.9    // Generation time for reference only; not used for billing
}