import os
import json
import requests

from JiMeng.Kit.VolcanoConst import VOLCANO_API_KEY

# API请求URL
url = 'https://ark.cn-beijing.volces.com/api/v3/chat/completions'

# 请求头
headers = {
    'Content-Type': 'application/json',
    'Authorization': f'Bearer {VOLCANO_API_KEY}'
}

# 请求体数据
payload = {
    'model': 'doubao-1-5-vision-pro-32k-250115',
    'messages': [
        {
            'content': [
                {
                    'image_url': {
                        'url': 'https://dsideal.obs.cn-north-1.myhuaweicloud.com/HuangHai/Backup/Text2Img.jpg'
                    },
                    'type': 'image_url'
                },
                {
                    'text': '以这张图片为首帧，帮我生成一个5秒的视频提示词。提示词包括：镜号、运镜、画面内容',
                    'type': 'text'
                }
            ],
            'role': 'user'
        }
    ]
}

# 发送POST请求
try:
    response = requests.post(url, headers=headers, data=json.dumps(payload), timeout=30)

    # 检查响应状态码
    response.raise_for_status()

    # 解析响应数据
    result = response.json()
    print('请求成功:')
    
    # 提取choices[0].message.content的内容
    if 'choices' in result and len(result['choices']) > 0:
        choice = result['choices'][0]
        if 'message' in choice and 'content' in choice['message']:
            content = choice['message']['content']
            print('提取的内容:')
            print(content)
            
            # 这里可以添加对content的进一步处理逻辑
            # 例如保存到文件、解析表格内容等
        else:
            print('响应中未找到message或content字段')
    else:
        print('响应中未找到choices字段或choices为空')

except requests.exceptions.RequestException as e:
    print(f'请求发生错误: {e}')
    if hasattr(e, 'response') and e.response is not None:
        print(f'错误响应状态码: {e.response.status_code}')
        print(f'错误响应内容: {e.response.text}')

except json.JSONDecodeError:
    print('响应解析失败: 返回的内容不是有效的JSON格式')
    print(f'响应内容: {response.text}')