haujetzhao--capswriter-offline
wehub-resource-sync
73 行
2.4 KiB
Python
73 行
2.4 KiB
Python
import json
|
|
import asyncio
|
|
from multiprocessing import Queue
|
|
|
|
from ..state import console
|
|
from ..schema import Result
|
|
from core.protocol import RecognitionMessage
|
|
from core.tools.asyncio_to_thread import to_thread
|
|
from .. import logger
|
|
|
|
|
|
|
|
async def ws_send(app):
|
|
|
|
state = app.state
|
|
queue_out = state.queue_out
|
|
sockets = state.sockets
|
|
|
|
logger.info("WebSocket 发送任务已启动")
|
|
|
|
while True:
|
|
try:
|
|
# 获取识别结果（从多进程队列）
|
|
result: Result = await to_thread(queue_out.get)
|
|
|
|
# 得到退出的通知
|
|
if result is None:
|
|
logger.info("收到退出通知，停止发送任务")
|
|
return
|
|
|
|
# 1. 将内部 Result 转换为标准的协议消息对象
|
|
msg = RecognitionMessage(
|
|
task_id=result.task_id,
|
|
is_final=result.is_final,
|
|
duration=result.duration,
|
|
time_start=result.time_start,
|
|
time_submit=result.time_submit,
|
|
time_complete=result.time_complete,
|
|
text=result.text,
|
|
text_accu=result.text_accu,
|
|
tokens=result.tokens,
|
|
timestamps=result.timestamps
|
|
)
|
|
|
|
# 获得 socket
|
|
websocket = next(
|
|
(ws for ws in sockets.values() if str(ws.id) == result.socket_id),
|
|
None,
|
|
)
|
|
|
|
if not websocket:
|
|
logger.warning(f"客户端 {result.socket_id} 不存在，跳过发送结果，任务ID: {result.task_id}")
|
|
continue
|
|
|
|
# 发送消息
|
|
await websocket.send(msg.to_json())
|
|
logger.debug(f"发送识别结果，任务ID: {result.task_id}, 文本长度: {len(result.text)}")
|
|
|
|
if result.type == 'mic':
|
|
logger.info(f"麦克风识别结果: {result.text}")
|
|
elif result.type == 'file':
|
|
console.print(f' 转录进度：{result.duration:.2f}s', end='\r')
|
|
logger.debug(f"文件转录进度: {result.duration:.2f}s")
|
|
if result.is_final:
|
|
console.print('\n [green]转录完成')
|
|
logger.info(f"文件转录完成，任务ID: {result.task_id}, 总时长: {result.duration:.2f}s")
|
|
|
|
except Exception as e:
|
|
logger.error(f"发送结果时发生错误: {e}", exc_info=True)
|
|
print(e)
|
|
|
|
|