#!/usr/bin/env python3
import sys
import os
import json
from volcenginesdkarkruntime import Ark

def analyze_douyin_video(video_url):
    api_key = os.getenv("DOUBAO_API_KEY")
    if not api_key:
        return {"error": "请设置 DOUBAO_API_KEY 环境变量"}
    
    client = Ark(api_key=api_key)
    
    try:
        response = client.chat.completions.create(
            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. 关键信息"
                }]
            }]
        )
        return response.choices[0].message.content
    except Exception as e:
        return {"error": str(e)}

if __name__ == "__main__":
    if len(sys.argv) < 2:
        print("用法: python3 analyze_douyin.py <抖音视频链接>")
        sys.exit(1)
    
    result = analyze_douyin_video(sys.argv[1])
    print(result if isinstance(result, str) else json.dumps(result, ensure_ascii=False))