108 lines
3.1 KiB
Python
108 lines
3.1 KiB
Python
|
import json
|
|||
|
|
import time
|
|||
|
|
|
|||
|
|
from ws4py.client.threadedclient import WebSocketClient
|
|||
|
|
import base64
|
|||
|
|
import hashlib
|
|||
|
|
import uuid
|
|||
|
|
from utils import config_util as cfg
|
|||
|
|
|
|||
|
|
base_url = "ws://wsapi.xfyun.cn/v1/aiui"
|
|||
|
|
|
|||
|
|
end_tag = "--end--"
|
|||
|
|
|
|||
|
|
|
|||
|
|
# qa 通讯类
|
|||
|
|
class __WSClient(WebSocketClient):
|
|||
|
|
q_msg = ''
|
|||
|
|
a_msg = ''
|
|||
|
|
|
|||
|
|
def opened(self):
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
def closed(self, code, reason=None):
|
|||
|
|
# if code == 1000:
|
|||
|
|
# print("qa close")
|
|||
|
|
# else:
|
|||
|
|
# print("连接异常关闭，code：" + str(code) + " ，reason：" + str(reason))
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
def received_message(self, m):
|
|||
|
|
|
|||
|
|
s = json.loads(str(m))
|
|||
|
|
|
|||
|
|
if s['action'] == "started":
|
|||
|
|
|
|||
|
|
# 输入内容并发送
|
|||
|
|
str_content = self.q_msg
|
|||
|
|
self.send(bytes(str_content.encode('utf-8')))
|
|||
|
|
time.sleep(0.04)
|
|||
|
|
|
|||
|
|
# 数据发送结束之后发送结束标识
|
|||
|
|
self.send(bytes(end_tag.encode("utf-8")))
|
|||
|
|
|
|||
|
|
elif s['action'] == "result":
|
|||
|
|
data = s['data']
|
|||
|
|
# with open('qa/out.txt', 'w') as file:
|
|||
|
|
# file.write(str(data))
|
|||
|
|
if data['sub'] == "iat":
|
|||
|
|
print("user: ", data["text"])
|
|||
|
|
elif data['sub'] == "nlp":
|
|||
|
|
intent = data['intent']
|
|||
|
|
if intent['rc'] == 0:
|
|||
|
|
self.a_msg = intent['answer']['text']
|
|||
|
|
else:
|
|||
|
|
self.a_msg = "我没有理解你说的话啊"
|
|||
|
|
elif data['sub'] == "tts":
|
|||
|
|
# TODO 播报pcm音频
|
|||
|
|
print('tts')
|
|||
|
|
pass
|
|||
|
|
elif s['action'] == "error":
|
|||
|
|
print('[NLP错误] ' + s['desc'])
|
|||
|
|
else:
|
|||
|
|
print(s)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def __get_auth_id():
|
|||
|
|
mac = uuid.UUID(int=uuid.getnode()).hex[-12:]
|
|||
|
|
return hashlib.md5(":".join([mac[e:e + 2] for e in range(0, 11, 2)]).encode("utf-8")).hexdigest()
|
|||
|
|
|
|||
|
|
|
|||
|
|
def question(text):
|
|||
|
|
ws = None
|
|||
|
|
try:
|
|||
|
|
# 构造握手参数
|
|||
|
|
curTime = int(time.time())
|
|||
|
|
|
|||
|
|
auth_id = __get_auth_id()
|
|||
|
|
|
|||
|
|
param = """{{
|
|||
|
|
"auth_id": "{0}",
|
|||
|
|
"data_type": "text",
|
|||
|
|
"scene": "main_box",
|
|||
|
|
"ver_type": "monitor",
|
|||
|
|
"close_delay": "200",
|
|||
|
|
"ent":"xtts",
|
|||
|
|
"vcn":"x_xiaoyan",
|
|||
|
|
"speed":"50",
|
|||
|
|
"interact_mode":"continuous",
|
|||
|
|
"context": "{{\\\"sdk_support\\\":[\\\"iat\\\",\\\"nlp\\\",\\\"tts\\\"]}}"
|
|||
|
|
}}"""
|
|||
|
|
|
|||
|
|
param = param.format(auth_id).encode(encoding="utf-8")
|
|||
|
|
paramBase64 = base64.b64encode(param).decode()
|
|||
|
|
checkSumPre = cfg.key_xf_aiui_api_key + str(curTime) + paramBase64
|
|||
|
|
checksum = hashlib.md5(checkSumPre.encode("utf-8")).hexdigest()
|
|||
|
|
connParam = "?appid=" + cfg.key_xf_aiui_app_id + "&checksum=" + checksum + "¶m=" + paramBase64 + "&curtime=" + str(curTime) + "&signtype=md5"
|
|||
|
|
|
|||
|
|
ws = __WSClient(base_url + connParam, protocols=['chat'], headers=[("Origin", "https://wsapi.xfyun.cn")])
|
|||
|
|
ws.q_msg = text
|
|||
|
|
ws.connect()
|
|||
|
|
ws.run_forever()
|
|||
|
|
|
|||
|
|
except KeyboardInterrupt:
|
|||
|
|
if ws is not None:
|
|||
|
|
ws.close()
|
|||
|
|
|
|||
|
|
return ws.a_msg
|