You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

59 lines
2.9 KiB

This file contains ambiguous Unicode characters!

This file contains ambiguous Unicode characters that may be confused with others in your current locale. If your use case is intentional and legitimate, you can safely ignore this warning. Use the Escape button to highlight these characters.

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 模型调用
'''