记录一次讯飞实时翻译

记录一次讯飞实时翻译

实现方式

1. 与讯飞进行websocket连接
2. 使用PyAudio模块进行实时录音
3. 将录音数据流上传至讯飞
4. 获取讯飞的返回值,并进行拼接

# -*- encoding:utf-8 -*-

import pyaudio
import hashlib
from hashlib import sha1
import hmac
import base64
from socket import *
import json, time, threading
from websocket import create_connection
import websocket
from urllib.parse import quote
import logging

logging.basicConfig()
base_url = "ws://rtasr.xfyun.cn/v1/ws"
app_id = "在讯飞控制台获取"
api_key = "在讯飞控制台获取"
file_path = "./test_1.pcm"

end_tag = "{"end": true}"


class Client():
    def __init__(self):
        # 生成鉴权参数
        ts = str(int(time.time()))
        tmp = app_id + ts
        hl = hashlib.md5()
        hl.update(tmp.encode(encoding='utf-8'))
        b_hl = bytes(hl.hexdigest(), encoding='utf-8')
        b_api_key = bytes(api_key, encoding='utf-8')
        my_sign = hmac.new(b_api_key, b_hl, sha1).digest()
        signa = base64.b64encode(my_sign)

        self.ws = create_connection(base_url + "?appid=" + app_id + "&ts=" + ts + "&signa=" + quote(signa))
        self.trecv = threading.Thread(target=self.recv)
        self.trecv.start()

    def start(self):
        FORMAT = pyaudio.paInt16
        CHANNELS = 1
        RATE = 16000  # 采样率
        CHUNK = 640  # 记录帧数
        RECORD_SECONDS = 30000  # 记录秒数

        # 调用PyAudio模块录音
        p = pyaudio.PyAudio()
        stream = p.open(format=FORMAT,
                        channels=CHANNELS,
                        rate=RATE,
                        input=True,
                        frames_per_buffer=CHUNK)

        print("开始录音...")
        while 1:  # 死循环一直录音
            data = stream.read(CHUNK)
            self.ws.send(data)  # 将数据发送给科大讯飞
            time.sleep(0.04)  # 发送太快可能导致错误,设置延时
        # stream.stop_stream()
        # stream.close()
        # p.terminate()

    def recv(self):
        try:
            while self.ws.connected:
                result = str(self.ws.recv())
                if len(result) == 0:
                    print("receive result end")
                    break
                result_dict = json.loads(result)

                # 解析结果
                if result_dict["action"] == "started":
                    print("handshake success, result: " + result)

                if result_dict["action"] == "result":  # 返回的结果
                    dic = json.loads(result_dict['data'])
                    x = ''
                    for i in dic['cn']['st']['rt']:
                        for g in i['ws']:
                            x += g['cw'][0]['w']
                    print(x)

                if result_dict["action"] == "error":  # 出错
                    print("rtasr error: " + result)
                    self.ws.close()
                    return
        except websocket.WebSocketConnectionClosedException:
            print("receive result end")

    def close(self):
        self.ws.close()
        print("connection closed")


if __name__ == '__main__':
    client = Client()
    client.start()

---