Initial commit

transcriptapi/server.py

@@ -0,0 +1,44 @@
+# 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, 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