asr/app/asr_engine.py

32 lines
819 B
Python
Raw Normal View History

2024-08-03 18:22:53 +08:00
from appPublic.log import info, debug, warning, error, exception, critical
2024-08-03 19:06:37 +08:00
from appPublic.dictObject import DictObject
2024-08-03 17:27:14 +08:00
from ahserver.serverenv import ServerEnv
from aiohttp.web import StreamResponse
async def generate(request, **kw):
2024-08-03 19:18:31 +08:00
params_kw = kw.get('params_kw', DictObject())
2024-08-04 08:50:43 +08:00
info(f'{params_kw=}')
2024-08-03 19:16:33 +08:00
model = params_kw.model
engine = None
g = ServerEnv()
if model=='whisper':
engine = g.whisper_engine
2024-08-03 17:27:14 +08:00
2024-08-03 19:20:42 +08:00
resp = StreamResponse()
2024-08-03 19:16:33 +08:00
await resp.prepare(request)
line = await request.content.readline()
content = ''
while line:
2024-08-03 18:22:53 +08:00
info(f'read from request stream {line=}')
2024-08-03 19:16:33 +08:00
txt = await engine.stt(line)
content += txt
d = json.dumps({'content':content})
btxt = d.encode('utf8')
await resp.write(btxt)
await resp.drain()
line = await request.content.readline()
2024-08-03 18:22:53 +08:00
info('response finish')
2024-08-03 19:16:33 +08:00
return resp
2024-08-03 17:27:14 +08:00