sora_2_api.py

import time
from openai import OpenAI
import base64

API_KEY="sk-YOUR-KEY"

client = OpenAI(api_key=API_KEY)

response = client.videos.create(
model="sora-2-pro",
prompt="Generate a video of a dog getting pulled over by a police officer",
)

while True:
    response = client.videos.retrieve(response.id)
    print(response)
    if response.status == "completed":
        break
    time.sleep(1)

    response = client.videos.download_content(
        video_id=response.id,
    )
    print(response)
    content = response.read()
    print(f"Downloaded {len(content)} bytes")

# Write the video content to an MP4 file
with open("sora_video.mp4", "wb") as f:
    f.write(content)
    print("Video saved to sora_video.mp4")