mirrors/whispering
1
0
Fork 0
mirror of https://github.com/shirayu/whispering.git synced 2024-11-25 18:31:00 +00:00
Code Issues Projects Releases Packages Wiki Activity

Changed

Browse source
This commit is contained in:
Yuta Hayashibe 2022-09-24 21:50:04 +09:00
parent f8bda94117
commit 1ebef74d3a
1 changed files with 16 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
whisper_streaming/websocket_client.py
View file

@ -44,22 +44,28 @@ async def transcribe_from_mic_and_send(
async def g():
return await q.get()
logger.debug(f"Segment #: {idx}")
logger.debug(f"Loop #: {idx}")
segment = None
try:
segment = await asyncio.wait_for(g(), timeout=3.0)
segment = await asyncio.wait_for(g(), timeout=0.5)
except asyncio.TimeoutError:
continue
pass
if segment is not None:
logger.debug(f"Segment size: {len(segment)}")
await ws.send(segment)
logger.debug("Sent")
async def recv():
return await ws.recv()
while True:
c = await ws.recv()
if len(c) == 0:
break
try:
c = await asyncio.wait_for(recv(), timeout=0.5)
chunk = ParsedChunk.parse_raw(c)
print(f"{chunk.start:.2f}->{chunk.end:.2f}\t{chunk.text}")
except asyncio.TimeoutError:
break
idx += 1