mirror of
https://github.com/Monadical-SAS/reflector.git
synced 2026-03-26 00:46:46 +00:00
045eae8ff2
* feat: enable daily co in selfhosted + only schedule tasks when necessary * feat: refactor aws storage to be platform agnostic + add local pad tracking with slfhosted support
33 lines
1.0 KiB
Python
33 lines
1.0 KiB
Python
from contextlib import asynccontextmanager
|
|
|
|
from fastapi import FastAPI
|
|
|
|
from .routers.diarization import router as diarization_router
|
|
from .routers.padding import router as padding_router
|
|
from .routers.transcription import router as transcription_router
|
|
from .routers.translation import router as translation_router
|
|
from .services.transcriber import WhisperService
|
|
from .services.diarizer import PyannoteDiarizationService
|
|
from .utils import ensure_dirs
|
|
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI):
|
|
ensure_dirs()
|
|
whisper_service = WhisperService()
|
|
whisper_service.load()
|
|
app.state.whisper = whisper_service
|
|
diarization_service = PyannoteDiarizationService()
|
|
diarization_service.load()
|
|
app.state.diarizer = diarization_service
|
|
yield
|
|
|
|
|
|
def create_app() -> FastAPI:
|
|
app = FastAPI(lifespan=lifespan)
|
|
app.include_router(transcription_router)
|
|
app.include_router(translation_router)
|
|
app.include_router(diarization_router)
|
|
app.include_router(padding_router)
|
|
return app
|