import os
class BaseConfig(object):
# 每日AI标签视频信息表
DAILY_VIDEO = {
'project': 'loghubods',
'table': 'vid_daily_top_not_taged'
}
# ODPS服务配置
ODPS_CONFIG = {
'ENDPOINT': 'http://service.cn.maxcompute.aliyun.com/api',
'ACCESSID': 'LTAIWYUujJAm7CbH',
'ACCESSKEY': 'RfSjdiWwED1sGFlsjXv0DlfTnZTG1P',
}
# OSS配置
OSS_CONFIG = {
'endpoint': 'http://oss-cn-hangzhou-internal.aliyuncs.com',
'accessKeyId': 'LTAIP6x1l3DXfSxm',
'accessKeySecret': 'KbTaM9ars4OX3PMS6Xm7rtxGr1FLon',
'bucket_name': 'art-pubbucket'
}
# 讯飞asr配置
XFASR_HOST = 'https://raasr.xfyun.cn/v2/api'
XF_API = {
'upload': '/upload',
'get_result': '/getResult'
}
XFASR_CONFIG = {
'appid': 'b11df2ff',
'secret_key': '598f38c17bd361f1f473dcfc840d330a'
}
# gpt配置
GPT_HOST = 'https://api.openai.com/v1/chat/completions'
# GPT_OPENAI_API_KEY = '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'
# GPT_OPENAI_API_KEY = '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'
GPT_OPENAI_API_KEY = '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'
GPT_URL = 'http://aigc.piaoquantv.com/aigc-server/aigc/content'
RETRY_MAX_COUNT = 3
# 代理地址
PROXIES = {
'http': 'http://127.0.0.1:4780',
'https': 'http://127.0.0.1:4780'
}
# 飞书应用凭证
FEISHU_TOKEN = {
'app_id': 'cli_a3667697a57b500e',
'app_secret': '5eMszgeNt21U56XnPjCykgmTfZUEEMnp'
}
# stop_words
STOP_WORDS = ['啊', '嗯', '呗', '呢', '呐', '呀', '唉', '哎', '额', '呃', '哦', '呵']
# video tags
TAGS = ['舞蹈', '美食', '时尚', '旅行', '音乐', '运动', '影视', '搞笑', '科技', '综艺',
'游戏', '情感', '健康', '人文', '社会', '热点', '财富', '生活']
TAGS_NEW = ['资讯', '国际', '政治', '科技', '军事', '健康', '社会', '民生政策', '人文', '人物', '旅行', '生活',
'美食', '综艺', '搞笑', '民俗文化', '安全', '休闲娱乐', '艺术', '情感']
# GPT prompt
format_json_array = '[{},{},{}]'
json_format = """[{"category": "<类别1>", "confidence": "<confidence1>"}, {"category": "<类别2>", "confidence": "<confidence2>"}, {"category": "<类别3>", "confidence": "<confidence3>"}]"""
GPT_PROMPT = {
'tags': {
'prompt1': f"""
请对如下文本进行分类。类别为其中的一个：【{' '.join(TAGS)}】。
以json格式返回，key为category与confidence，分别代表类别与分类置信度。给出top 3的分类结果。
-----------------------------
""",
'prompt2': f"""
请对如下文本进行：
1. 分类，类别为其中的一个：【{' '.join(TAGS)}】。如果无法有效分类，请返回“其他”。
2. 用20个字以内对文本内容进行概况。
3. 为文本取一个易于分享，吸引人要求的标题。
4. 列举三个关键词。
以json格式返回，key为category, confidence, summary, title, keywords。分别代表类别，分类置信度，概要，标题，关键词。
-----------------------------
""",
'prompt3': f"""
请对如下文本进行分类，不允许返回"其他"。
以json格式返回，key为categor与confidence, 分别代表类别与分类置信度。给出top 3的分类结果。
-----------------------------
""",
'prompt4': f"""请对如下文本进行分类。类别为其中的一个：【{' '.join(TAGS_NEW)}】。
以json array格式返回，key为category与confidence，分别代表类别与分类置信度。给出top 3的分类结果。
-----------------------------
""",
'prompt5': f"""请对如下文本进行分类。类别为其中的一个：【{' '.join(TAGS_NEW)}】。
以json array格式返回,{json_format},key为category与confidence，分别代表类别与分类置信度。给出top 3的分类结果。
-----------------------------
""",
'prompt6': f"""
请对如下文本进行：
1. 用20个字以内对文本内容进行概况。
2. 列举三个关键词。
仅以json格式返回，key为summary, keywords。分别代表概要，关键词。
-----------------------------
""",
'prompt7': f"""请根据视频的标题、概况和关键词对其进行分类。类别为其中的一个：【{' '.join(TAGS_NEW)}】。
仅以json array格式返回,{json_format}，key为category与confidence，分别代表类别与分类置信度。给出top 3的分类结果。
-----------------------------
""",
'prompt8': """请根据视频的标题、概况和关键词总结下面几个信息。信息我都写到 json 里面了：
```
{
"key_words": [], # 返回关于这个视频的三个关键词
"search_keys": [], # 内容可能的搜索关键词，返回 3 个
"extra_keys": [], # 关心这个视频的用户还会关心哪些关键词， 返回 3 个
}
```
只需要返回一个 json，key 和上面的一样，视频的信息如下：\n-------------------------------\n""",
'prompt9': """请根据视频的标题、概况和关键词总结下面几个信息。信息我都写到 json 里面了：
```
{
"tone": 标题的语气，用一个词概括,
"target_audience": 标题的受众群体，用一个词描述；
"target_age": 标题的受众年龄段，从 老年， 中年，青年，小孩， 不限， 这五个里面选择
}
```
只需要返回一个 json，key 和上面的一样，视频的信息如下：\n-------------------------------\n""",
'prompt10': """请根据视频的标题、概况和关键词总结下面几个信息。信息我都写到 json 里面了：
```
{
"category": "视频的类别, 类别为其中的一个或多个：【'资讯', '国际', '政治', '科技', '军事', '健康', '社会', '民生政策', '人文', '人物', '旅行', '生活',
'美食', '综艺', '搞笑', '民俗文化', '安全', '休闲娱乐', '艺术', '情感'】",
"target_gender": 受众性别：男，女，不限，这三个里面选择一个,
"address": 受众可能属于哪个城市, 用城市名描述
"theme": 内容的主题， 用一个词概括
}
```
只需要返回一个 json，key 和上面的一样，视频的信息如下：\n-------------------------------\n""",
},
'title': {
'prompt1': f"""
请对如下文本进行: 为文本取一个易于分享，吸引人要求的标题。要求在30个字以内。
-----------------------------
""",
'prompt2': f"""
我想让你充当爆款标题生成器，我会给你提供一段视频的讲解文本，你生成一个更吸引眼球的标题。
标题不要超过35个字。
标题要突出惊奇感，让老年人看到就想转发，语义要保持中立，不要向负面倾斜，不要涉及政治敏感话题。
如果原讲解文本中有地名，不要改变讲解文本中的地名。
如果原讲解文本是关于唱歌但是没有说明是哪首歌，你起的标题也不要说歌名。
如果原讲解文本中没有明确指出男女性别，你起的标题也不要说具体的性别。
如果原讲解文本中没有明确指出是谁干了这件事，你起的标题也不要说是谁干的。
如果原讲解文本中没有提及老年人，你起的标题也不要说老年人。
我的原讲解文本是：
"""
}
}
class TestConfig(BaseConfig):
# 后端接口 - 视频添加AI相关标签
ADD_VIDEO_AI_TAGS_URL = 'https://testadmin.piaoquantv.com/manager/video/tag/addVideoAiTags'
# 日志服务配置
ALIYUN_LOG = {
'ENDPOINT': 'cn-hangzhou.log.aliyuncs.com',
'ACCESSID': 'LTAIWYUujJAm7CbH',
'ACCESSKEY': 'RfSjdiWwED1sGFlsjXv0DlfTnZTG1P',
'PROJECT': 'aigc-server-test',
}
class PreConfig(BaseConfig):
# 后端接口 - 视频添加AI相关标签
ADD_VIDEO_AI_TAGS_URL = 'https://preadmin.piaoquantv.com/manager/video/tag/addVideoAiTags'
# 日志服务配置
ALIYUN_LOG = {
'ENDPOINT': 'cn-hangzhou.log.aliyuncs.com',
'ACCESSID': 'LTAIWYUujJAm7CbH',
'ACCESSKEY': 'RfSjdiWwED1sGFlsjXv0DlfTnZTG1P',
'PROJECT': 'aigc-server',
}
class ProConfig(BaseConfig):
# 后端接口 - 视频添加AI相关标签
ADD_VIDEO_AI_TAGS_URL = 'https://admin.piaoquantv.com/manager/video/tag/addVideoAiTags'
# 日志服务配置
ALIYUN_LOG = {
'ENDPOINT': 'cn-hangzhou.log.aliyuncs.com',
'ACCESSID': 'LTAIWYUujJAm7CbH',
'ACCESSKEY': 'RfSjdiWwED1sGFlsjXv0DlfTnZTG1P',
'PROJECT': 'aigc-server',
}
def set_config():
# return TestConfig()
# return PreConfig()
return ProConfig()