# 文生视频
curl --request POST \
  --url https://api.tokenops.ai/v1/videos \
  --header 'Authorization: Bearer <API-KEY>' \
  --header 'Content-Type: application/json' \
  --data '{
    "model": "doubao-seedance-2-0-260128",
    "content": [
        {
            "type": "text",
            "text": "多个镜头。一名侦探进入一间光线昏暗的房间。他检查桌上的线索，手里拿起桌上的某个物品。镜头转向他正在思索。"
        }
    ],
    "ratio": "16:9",
    "seconds": "5",
    "watermark": false,
    "tools": [{"type": "web_search"}]
  }'

# 有声视频（doubao-seedance-1-5-pro-251215）
curl --request POST \
  --url https://api.tokenops.ai/v1/videos \
  --header 'Authorization: Bearer <API-KEY>' \
  --header 'Content-Type: application/json' \
  --data '{
    "model": "doubao-seedance-1-5-pro-251215",
    "content": [
        {
            "type": "text",
            "text": "一位音乐家在舞台上演奏小提琴，观众席上掌声雷动"
        }
    ],
    "ratio": "16:9",
    "seconds": "8",
    "generate_audio": true
        }
    ],
    "generate_audio": true
  }'

# 查询任务状态
curl -X GET "https://api.tokenops.ai/v1/videos/cgt-20251110203910-nmfp7" \
  --header 'Authorization: Bearer <API-KEY>'

# 下载视频文件
curl -X GET "https://api.tokenops.ai/v1/videos/cgt-20251110203910-nmfp7/content" \
  --header 'Authorization: Bearer <API-KEY>' \
  --output doubao_video.mp4

import requests
import time

API_KEY = "<API-KEY>"
BASE_URL = "https://api.tokenops.ai/v1"

def complete_doubao_video_generation(content, **kwargs):
    """
    完整的豆包视频生成流程
    """
    # 步骤1: 创建任务
    print("步骤1: 创建视频生成任务...")
    create_result = create_doubao_video(content, **kwargs)
    if not create_result:
        return False
    
    video_id = create_result['id'] 
    print(f"任务ID: {video_id}")
    
    # 步骤2: 等待任务完成
    print("步骤2: 等待任务完成...")
    max_wait_time = 900  # 15分钟
    start_time = time.time()
    
    while time.time() - start_time < max_wait_time:
        status_info = check_video_status(video_id)
        if not status_info:
            break
            
        status = status_info.get('status')
        print(f"当前状态: {status}")
        
        if status == 'completed':
            print("视频生成完成！")
            break
        elif status == 'failed':
            error_info = status_info.get('error', {})
            print(f"视频生成失败: {error_info.get('message', '未知错误')}")
            return False
        
        time.sleep(20)  # 每20秒检查一次
    else:
        print("等待超时")
        return False
    
    # 步骤3: 下载视频
    print("步骤3: 下载视频...")
    output_path = f"doubao_video_{video_id}.mp4"
    if download_video(video_id, output_path):
        print(f"视频生成完成: {output_path}")
        return True
    
    return False

# 使用示例
if __name__ == "__main__":
    # 示例1: 基础文本到视频生成
    text_content = [
        {
            "type": "text",
            "text": "多个镜头。一名侦探进入一间光线昏暗的房间。他检查桌上的线索，手里拿起桌上的某个物品。镜头转向他正在思索。"
        }
    ]
    
    success = complete_doubao_video_generation(content=text_content)
    
    if success:
        print("✅ 视频生成成功！")
    else:
        print("❌ 视频生成失败！")