from Util.CommonUtil import *

# 请求URL
url = 'https://goapi.gptnb.me/v1/chat/completions'
# 令牌
token = 'sk-amQHwiEzPIZIB2KuF5A10dC23a0e4b02B48a7a2b6aFa0662'

# POST请求的数据
modelData = {
    "model": "pika-text-to-video",
    "messages": [
        {
            "role": "user",
            "content": "1girl walk"
        }
    ],
    "temperature": 0.7
}

# 发送POST请求
# JWT授权为啥要在 Authorization标头里加个Bearer 呢
# https://blog.csdn.net/liuzaixi/article/details/133686087
headers = {
    'Accept': 'application/json',
    'Authorization': 'Bearer ' + token,
    'User-Agent': 'Apifox/1.0.0 (https://apifox.com)',
    'Content-Type': 'application/json'
}

# response = requests.post(url=url, headers=headers, data=json.dumps(data))

# 打印返回的响应内容
# print(response.text)

# 下面是返回的示例
res = {"id": "chatcmpl-89D7nWn2yyvlNGn5O6JzCJlO6KJy2", "object": "chat.completion", "created": 1717131288,
       "model": "pika-text-to-video", "choices": [{"index": 0, "message": {"role": "assistant",
                                                                           "content": "✅成功创建视频任务：331fa3f9-799f-4fb7-a765-bb5518857212|5297d397-fca2-4d43-98ca-214ffda59937\n\n⌛️视频生成中.................✅视频生成成功\n\n[1girl walk](https://cdn.pika.art/v1/5297d397-fca2-4d43-98ca-214ffda59937/1girl_walk_seed8131565178219680.mp4)\n[⏬下载](https://cdn.pika.art/v1/5297d397-fca2-4d43-98ca-214ffda59937/1girl_walk_seed8131565178219680.mp4)\n\n"},
                                                   "finish_reason": "stop"}],
       "usage": {"prompt_tokens": 10, "completion_tokens": 161, "total_tokens": 171}}
content = res["choices"][0]['message']['content']
flagDown = '[⏬下载]'
pos = content.find(flagDown) + len(flagDown)
url = content[pos:].replace('(', '').replace(')', '')
print(url)
'''
{"id":"chatcmpl-89D7nWn2yyvlNGn5O6JzCJlO6KJy2","object":"chat.completion","created":1717131288,"model":"pika-text-to-video",
"choices":[{"index":0,"message":{"role":"assistant","content":"✅成功创建视频任务：331fa3f9-799f-4fb7-a765-bb5518857212|5297d397-fca2-4d43-98ca-214ffda59937\n\n⌛️
视频生成中.................✅视频生成成功\n\n[1girl walk]
(https://cdn.pika.art/v1/5297d397-fca2-4d43-98ca-214ffda59937/1girl_walk_seed8131565178219680.mp4)\n[⏬下载]
(https://cdn.pika.art/v1/5297d397-fca2-4d43-98ca-214ffda59937/1girl_walk_seed8131565178219680.mp4)\n\n"},
"finish_reason":"stop"}],"usage":{"prompt_tokens":10,"completion_tokens":161,"total_tokens":171}}
'''
# Image/pika_sample.mp4 是生成的样例

''' 费用情况
2024-05-31 12:55:08	消费	littlehb的初始令牌	pika-text-to-video	<1秒	非流	10	161	$0.01026	模型调用
2024-05-31 12:53:32	消费	littlehb的初始令牌	pika-text-to-video	<1秒	非流	10	181	$0.5	模型调用
'''