feat: 项目初始化、完成基本流式传输和语音识别功能

backend/app/api/v1/endpoints/asr.py

@@ -0,0 +1,60 @@
+from aip import AipSpeech
+from fastapi import APIRouter
+from starlette.websockets import WebSocket
+
+from app.constants.asr import APP_ID, API_KEY, SECRET_KEY
+
+asr_client = AipSpeech(APP_ID, API_KEY, SECRET_KEY)
+
+router = APIRouter()
+
+
+@router.websocket("/asr")
+async def chat2(websocket: WebSocket):
+    # 等待websocket接收数据
+    await websocket.accept()
+    temp_buffer = bytes()
+    # 在此处与百度建立websocket连接
+    while True:
+        # 等待websocket接收文本数据
+        receive_data = await websocket.receive()
+        buffer = receive_data.get("bytes")
+        text = receive_data.get("text")
+        if text == "录音完成":
+            asr_text = await asr_buffer(temp_buffer)
+            await websocket.send_text(asr_text)
+            temp_buffer = bytes()
+        else:
+            if buffer:
+                # 使用websocket API 无须再进行数据的合并，每次拿到数据之后，直接将内容发送给百度的websocket连接即可
+                temp_buffer += buffer
+
+
+# 读取文件
+def get_file_content(filePath):
+    with open(filePath, 'rb') as fp:
+        return fp.read()
+
+
+# 识别本地文件
+async def asr_file(filePath):
+    result = await asr_client.asr(get_file_content(filePath), 'pcm', 16000, {
+        'dev_pid': 1537,
+    })
+    if result.get('err_msg') == 'success.':
+        return result.get('result')[0]
+    else:
+        return '语音转换失败'
+
+
+# 识别语音流
+# async的意思是定义异步函数，当使用await修饰异步函数并执行时，如果该异步函数耗时比较长
+# python会自动挂起异步函数，让其他代码运行，等到异步函数完成之后，再回头调用函数
+async def asr_buffer(buffer_data):
+    result = asr_client.asr(buffer_data, 'pcm', 16000, {
+        'dev_pid': 1537,
+    })
+    if result.get('err_msg') == 'success.':
+        return result.get('result')[0]
+    else:
+        return '语音转换失败'