# Copyright (c) 2024 Julian Müller (ChaoticByte)
from os import getpid as _getpid
from sanic import Sanic as _Sanic
from sanic import empty as _empty
from sanic import Request as _Request
from . import env as _env
from .msg import ComponentLogger as _ComponentLogger
from .stt import STT as _STT
def get_app() -> _Sanic:
app = _Sanic("TranscriptAPI")
@app.get("/ping")
async def ping(_):
return _empty(status=200)
@app.post('/')
async def transcribe(request: _Request):
audio = request.files.get("audio").body
if len(audio) < 1:
return _empty(400)
resp = await request.respond(content_type="text/plain")
for s in app.ctx.stt.transcribe(audio):
await resp.send(s)
await resp.eof()
@app.before_server_start
async def setup_stt(app):
app.ctx.stt = _STT(_env.API_STT_MODEL, n_threads=_env.API_STT_THREADS, logger=_ComponentLogger(f"{_getpid()}/STT"))
@app.after_server_start
async def init_stt(app):
app.ctx.stt.init()
@app.on_response
async def middleware(_, response):
response.headers["Access-Control-Allow-Origin"] = _env.ACCESS_CONTROL_ALLOW_ORIGIN
return app
