mirror of
https://github.com/remsky/Kokoro-FastAPI.git
synced 2025-08-05 16:48:53 +00:00
111 lines
2.8 KiB
Python
111 lines
2.8 KiB
Python
"""
|
|
FastAPI OpenAI Compatible API
|
|
"""
|
|
|
|
import sys
|
|
from contextlib import asynccontextmanager
|
|
|
|
import uvicorn
|
|
from loguru import logger
|
|
from fastapi import FastAPI
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
|
|
from .core.config import settings
|
|
from .services.tts_model import TTSModel
|
|
from .routers.development import router as dev_router
|
|
from .services.tts_service import TTSService
|
|
from .routers.openai_compatible import router as openai_router
|
|
|
|
|
|
def setup_logger():
|
|
"""Configure loguru logger with custom formatting"""
|
|
config = {
|
|
"handlers": [
|
|
{
|
|
"sink": sys.stdout,
|
|
"format": "<fg #2E8B57>{time:hh:mm:ss A}</fg #2E8B57> | "
|
|
"{level: <8} | "
|
|
"{message}",
|
|
"colorize": True,
|
|
"level": "INFO",
|
|
},
|
|
],
|
|
}
|
|
logger.remove()
|
|
logger.configure(**config)
|
|
logger.level("ERROR", color="<red>")
|
|
|
|
|
|
# Configure logger
|
|
setup_logger()
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI):
|
|
"""Lifespan context manager for model initialization"""
|
|
logger.info("Loading TTS model and voice packs...")
|
|
|
|
# Initialize the main model with warm-up
|
|
voicepack_count = await TTSModel.setup()
|
|
# boundary = "█████╗"*9
|
|
boundary = "░" * 24
|
|
startup_msg = f"""
|
|
|
|
{boundary}
|
|
|
|
╔═╗┌─┐┌─┐┌┬┐
|
|
╠╣ ├─┤└─┐ │
|
|
╚ ┴ ┴└─┘ ┴
|
|
╦╔═┌─┐┬┌─┌─┐
|
|
╠╩╗│ │├┴┐│ │
|
|
╩ ╩└─┘┴ ┴└─┘
|
|
|
|
{boundary}
|
|
"""
|
|
# TODO: Improve CPU warmup, threads, memory, etc
|
|
startup_msg += f"\nModel warmed up on {TTSModel.get_device()}"
|
|
startup_msg += f"\n{voicepack_count} voice packs loaded\n"
|
|
startup_msg += f"\n{boundary}\n"
|
|
logger.info(startup_msg)
|
|
|
|
yield
|
|
|
|
|
|
# Initialize FastAPI app
|
|
app = FastAPI(
|
|
title=settings.api_title,
|
|
description=settings.api_description,
|
|
version=settings.api_version,
|
|
lifespan=lifespan,
|
|
openapi_url="/openapi.json", # Explicitly enable OpenAPI schema
|
|
)
|
|
|
|
# Add CORS middleware
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=["*"],
|
|
allow_credentials=True,
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
# Include routers
|
|
app.include_router(openai_router, prefix="/v1")
|
|
app.include_router(dev_router) # New development endpoints
|
|
# app.include_router(text_router) # Deprecated but still live for backwards compatibility
|
|
|
|
|
|
# Health check endpoint
|
|
@app.get("/health")
|
|
async def health_check():
|
|
"""Health check endpoint"""
|
|
return {"status": "healthy"}
|
|
|
|
|
|
@app.get("/v1/test")
|
|
async def test_endpoint():
|
|
"""Test endpoint to verify routing"""
|
|
return {"status": "ok"}
|
|
|
|
|
|
if __name__ == "__main__":
|
|
uvicorn.run("api.src.main:app", host=settings.host, port=settings.port, reload=True)
|