mirror of
https://github.com/zhayujie/chatgpt-on-wechat.git
synced 2026-03-19 21:38:18 +08:00
74 lines
2.7 KiB
Python
74 lines
2.7 KiB
Python
|
||
"""
|
||
baidu voice service
|
||
"""
|
||
import time
|
||
from aip import AipSpeech
|
||
from bridge.reply import Reply, ReplyType
|
||
from common.log import logger
|
||
from common.tmp_dir import TmpDir
|
||
from voice.voice import Voice
|
||
from voice.audio_convert import get_pcm_from_wav
|
||
from config import conf
|
||
"""
|
||
百度的语音识别API.
|
||
dev_pid:
|
||
- 1936: 普通话远场
|
||
- 1536：普通话(支持简单的英文识别)
|
||
- 1537：普通话(纯中文识别)
|
||
- 1737：英语
|
||
- 1637：粤语
|
||
- 1837：四川话
|
||
要使用本模块, 首先到 yuyin.baidu.com 注册一个开发者账号,
|
||
之后创建一个新应用, 然后在应用管理的"查看key"中获得 API Key 和 Secret Key
|
||
填入 config.json 中.
|
||
baidu_app_id: ''
|
||
baidu_api_key: ''
|
||
baidu_secret_key: ''
|
||
baidu_dev_pid: '1536'
|
||
"""
|
||
|
||
|
||
class BaiduVoice(Voice):
|
||
APP_ID = conf().get('baidu_app_id')
|
||
API_KEY = conf().get('baidu_api_key')
|
||
SECRET_KEY = conf().get('baidu_secret_key')
|
||
DEV_ID = conf().get('baidu_dev_pid')
|
||
client = AipSpeech(APP_ID, API_KEY, SECRET_KEY)
|
||
|
||
def __init__(self):
|
||
pass
|
||
|
||
def voiceToText(self, voice_file):
|
||
# 识别本地文件
|
||
logger.debug('[Baidu] voice file name={}'.format(voice_file))
|
||
pcm = get_pcm_from_wav(voice_file)
|
||
res = self.client.asr(pcm, "pcm", 16000, {"dev_pid": self.DEV_ID})
|
||
if res["err_no"] == 0:
|
||
logger.info("百度语音识别到了：{}".format(res["result"]))
|
||
text = "".join(res["result"])
|
||
reply = Reply(ReplyType.TEXT, text)
|
||
else:
|
||
logger.info("百度语音识别出错了: {}".format(res["err_msg"]))
|
||
if res["err_msg"] == "request pv too much":
|
||
logger.info(" 出现这个原因很可能是你的百度语音服务调用量超出限制，或未开通付费")
|
||
reply = Reply(ReplyType.ERROR,
|
||
"百度语音识别出错了；{0}".format(res["err_msg"]))
|
||
return reply
|
||
|
||
def textToVoice(self, text):
|
||
result = self.client.synthesis(text, 'zh', 1, {
|
||
'spd': 5, 'pit': 5, 'vol': 5, 'per': 111
|
||
})
|
||
if not isinstance(result, dict):
|
||
fileName = TmpDir().path() + '语音回复_' + str(int(time.time())) + '.mp3'
|
||
with open(fileName, 'wb') as f:
|
||
f.write(result)
|
||
logger.info(
|
||
'[Baidu] textToVoice text={} voice file name={}'.format(text, fileName))
|
||
reply = Reply(ReplyType.VOICE, fileName)
|
||
else:
|
||
logger.error('[Baidu] textToVoice error={}'.format(result))
|
||
reply = Reply(ReplyType.ERROR, "抱歉，语音合成失败")
|
||
return reply
|