mirror of
https://github.com/Monadical-SAS/reflector.git
synced 2026-03-31 11:26:47 +00:00
d164e486cc
* allow memory flags and per service config * feat: mixdown modal services + processor pattern
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
from contextlib import asynccontextmanager
|
|
|
|
from fastapi import FastAPI
|
|
|
|
from .routers.diarization import router as diarization_router
|
|
from .routers.mixdown import router as mixdown_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)
|
|
app.include_router(mixdown_router)
|
|
return app
|