import os
from volcenginesdkarkruntime import Ark
from utils.tools import encode_video
from utils.upload import upload_file_to_tos
from dotenv import load_dotenv

load_dotenv()

video_url = encode_video("./data/test.mp4")

video_url = upload_file_to_tos("./data/raw_test.mp4")
print(video_url)

client = Ark(
    base_url="https://ark.cn-beijing.volces.com/api/v3",
    api_key=os.environ.get("ARK_API_KEY"),
)

# Non-streaming:
print("----- image input request -----")
completion = client.chat.completions.create(
    model="doubao-seed-1-6-250615",
    messages=[
        {
            "role": "user",
            "content": [
                {
                    "type": "video_url",
                    "video_url": {
                        "url": video_url,
                        "fps": 2
                    },
                },
                {"type": "text", "text": "请解析视频内容"},
            ],
        }
    ],
    
)
print(completion.choices[0].message.content)