4cfad5ae0f
- 全新ui - 全面优化websocket逻辑，提高数字人和ui连接的稳定性及资源开销 - 全面优化唤醒逻辑，提供稳定的普通唤醒模式和前置词唤醒模式 - 优化拾音质量，支持多声道麦克风拾音 - 优化自动播放服务器的对接机制，提供稳定和兼容旧版ue工程的对接模式 - 数字人接口输出机器人表情，以适应新fay ui及单片机的数字人表情输出 - 使用更高级的音频时长计算方式，可以更精准控制音频播放完成后的逻辑 - 修复点击关闭按钮会导致程序退出的bug - 修复没有麦克风的设备开启麦克风会出错的问题 - 为服务器主机地址提供配置项，以方便服务器部署
100 lines
3.9 KiB
Python
100 lines
3.9 KiB
Python
import json
|
||
import requests
|
||
import time
|
||
from core.authorize_tb import Authorize_Tb
|
||
from utils import config_util as cfg
|
||
from utils import util
|
||
|
||
def get_sentiment(cont):
|
||
emotion = Emotion()
|
||
answer = emotion.get_sentiment(cont)
|
||
return answer
|
||
|
||
class Emotion:
|
||
|
||
def __init__(self):
|
||
self.app_id = cfg.baidu_emotion_app_id
|
||
self.authorize_tb = Authorize_Tb()
|
||
|
||
def get_sentiment(self, cont):
|
||
token = self.__check_token()
|
||
if token is None or token == 'expired':
|
||
token_info = self.__get_token()
|
||
if token_info is not None and token_info['access_token'] is not None:
|
||
#转换过期时间
|
||
updated_in_seconds = time.time()
|
||
expires_timedelta = token_info['expires_in']
|
||
expiry_timestamp_in_seconds = updated_in_seconds + expires_timedelta
|
||
expiry_timestamp_in_milliseconds = expiry_timestamp_in_seconds * 1000
|
||
if token == 'expired':
|
||
self.authorize_tb.update_by_userid(self.app_id, token_info['access_token'], expiry_timestamp_in_milliseconds)
|
||
else:
|
||
self.authorize_tb.add(self.app_id, token_info['access_token'], expiry_timestamp_in_milliseconds)
|
||
token = token_info['access_token']
|
||
else:
|
||
token = None
|
||
|
||
if token is not None:
|
||
try:
|
||
url=f"https://aip.baidubce.com/rpc/2.0/nlp/v1/sentiment_classify?access_token={token}"
|
||
req = json.dumps({"text": cont})
|
||
headers = {
|
||
'Content-Type': 'application/json',
|
||
'Accept': 'application/json'
|
||
}
|
||
r = requests.post(url, headers=headers, data=req)
|
||
if r.status_code != 200:
|
||
util.log(1, f"百度情感分析对接有误: {r.text}")
|
||
return 0
|
||
info = json.loads(r.text)
|
||
if not self.has_field(info,'error_code'):
|
||
return info['items'][0]['sentiment']
|
||
else:
|
||
util.log(1, f"百度情感分析对接有误： {info['error_msg']}")
|
||
return 0
|
||
except Exception as e:
|
||
util.log(1, f"百度情感分析对接有误： {str(e)}")
|
||
return 0
|
||
else:
|
||
return 0
|
||
|
||
def __check_token(self):
|
||
self.authorize_tb.init_tb()
|
||
info = self.authorize_tb.find_by_userid(self.app_id)
|
||
if info is not None:
|
||
if info[1] >= int(time.time())*1000:
|
||
return info[0]
|
||
else:
|
||
return 'expired'
|
||
else:
|
||
return None
|
||
|
||
def __get_token(self):
|
||
try:
|
||
url=f"https://aip.baidubce.com/oauth/2.0/token?client_id={cfg.baidu_emotion_api_key}&client_secret={cfg.baidu_emotion_secret_key}&grant_type=client_credentials"
|
||
headers = {'Content-Type':'application/json;charset=UTF-8'}
|
||
r = requests.post(url, headers=headers)
|
||
if r.status_code != 200:
|
||
info = json.loads(r.text)
|
||
if info["error"] == "invalid_client":
|
||
util.log(1, f"请检查baidu_emotion_api_key")
|
||
else:
|
||
util.log(1, f"请检查baidu_emotion_secret_key")
|
||
return None
|
||
info = json.loads(r.text)
|
||
if not self.has_field(info,'error_code'):
|
||
return info
|
||
else:
|
||
util.log(1, f"百度情感分析对接有误： {info['error_msg']}")
|
||
util.log(1, f"请检查baidu_emotion_api_key和baidu_emotion_secret_key")
|
||
return None
|
||
except Exception as e:
|
||
util.log(1, f"百度情感分析有1误： {str(e)}")
|
||
return None
|
||
|
||
|
||
def has_field(self, array, field):
|
||
return any(field in item for item in array)
|
||
|
||
|