#!/usr/bin/env python3
import sys
import json
import requests

def analyze_douyin_video(video_url, api_key):
    url = "https://ark.cn-beijing.volces.com/api/v3/chat/completions"
    
    headers = {
        "Content-Type": "application/json",
        "Authorization": f"Bearer {api_key}"
    }
    
    data = {
        "model": "doubao-seed-2-0-mini-260215",
        "messages": [{
            "role": "user",
            "content": [{
                "type": "video_url",
                "video_url": {"url": video_url}
            }, {
                "type": "text",
                "text": "请详细分析这个视频的内容，包括：1. 视频主题 2. 主要内容 3. 关键信息"
            }]
        }]
    }
    
    try:
        response = requests.post(url, headers=headers, json=data, timeout=60)
        response.raise_for_status()
        result = response.json()
        return result['choices'][0]['message']['content']
    except Exception as e:
        return f"错误: {str(e)}"

if __name__ == "__main__":
    if len(sys.argv) < 3:
        print("用法: python3 analyze_douyin_simple.py <API_KEY> <抖音视频链接>")
        sys.exit(1)
    
    result = analyze_douyin_video(sys.argv[2], sys.argv[1])
    print(result)