mirror of
https://github.com/shirayu/whispering.git
synced 2024-06-13 02:39:23 +00:00
68 lines
1.5 KiB
Python
68 lines
1.5 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import asyncio
|
|
from logging import getLogger
|
|
|
|
import numpy as np
|
|
import websockets
|
|
from websockets.exceptions import ConnectionClosedOK
|
|
|
|
from whispering.transcriber import Context, WhisperStreamingTranscriber
|
|
|
|
logger = getLogger(__name__)
|
|
|
|
|
|
async def serve_with_websocket_main(websocket):
|
|
global g_wsp
|
|
global g_ctx
|
|
idx: int = 0
|
|
ctx: Context = g_ctx.copy(
|
|
deep=True,
|
|
)
|
|
|
|
while True:
|
|
logger.debug(f"Audio #: {idx}")
|
|
try:
|
|
message = await websocket.recv()
|
|
except ConnectionClosedOK:
|
|
break
|
|
|
|
if isinstance(message, str):
|
|
logger.debug(f"Got str: {message}")
|
|
continue
|
|
|
|
logger.debug(f"Message size: {len(message)}")
|
|
audio = np.frombuffer(message, dtype=np.float32)
|
|
for chunk in g_wsp.transcribe(
|
|
audio=audio, # type: ignore
|
|
ctx=ctx,
|
|
):
|
|
await websocket.send(chunk.json())
|
|
idx += 1
|
|
|
|
|
|
async def serve_with_websocket(
|
|
*,
|
|
wsp: WhisperStreamingTranscriber,
|
|
host: str,
|
|
port: int,
|
|
ctx: Context,
|
|
):
|
|
logger.info(f"Serve at {host}:{port}")
|
|
logger.info("Make secure with your responsibility!")
|
|
global g_wsp
|
|
global g_ctx
|
|
g_wsp = wsp
|
|
g_ctx = ctx
|
|
|
|
try:
|
|
async with websockets.serve( # type: ignore
|
|
serve_with_websocket_main,
|
|
host=host,
|
|
port=port,
|
|
max_size=999999999,
|
|
):
|
|
await asyncio.Future()
|
|
except KeyboardInterrupt:
|
|
pass
|