File size: 3,623 Bytes
03977cf 5fb4696 674469e 5fb4696 d43ba60 674469e 2da4544 674469e 5fb4696 03977cf d43ba60 03977cf 2da4544 5fb4696 d43ba60 9db766f 674469e 2da4544 9db766f d43ba60 9db766f 674469e 2da4544 674469e 2da4544 674469e 2da4544 d43ba60 674469e d43ba60 2da4544 674469e 2da4544 d43ba60 674469e d43ba60 2da4544 674469e 2da4544 9db766f 2da4544 d43ba60 9db766f 2da4544 9db766f d43ba60 9db766f d43ba60 56dc677 2da4544 56dc677 2da4544 56dc677 674469e 2da4544 674469e 2da4544 674469e 2da4544 674469e 2da4544 d43ba60 2da4544 674469e 5fb4696 2da4544 674469e 2da4544 5fb4696 9db766f d43ba60 9db766f 2da4544 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 |
import sys
from pathlib import Path
import logging
from contextlib import asynccontextmanager
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
import uvicorn
# --- Logging ---
logging.basicConfig(
level=logging.INFO,
format="%(asctime)s - %(name)s - %(levelname)s - %(message)s"
)
logger = logging.getLogger(__name__)
# --- Ajouter app dir au PATH ---
app_dir = Path(__file__).parent
sys.path.insert(0, str(app_dir))
# --- Config ---
from config import (
API_TITLE, API_DESCRIPTION, API_VERSION,
HUGGINGFACE_API_KEY, HUGGINGFACE_STANCE_MODEL_ID, HUGGINGFACE_LABEL_MODEL_ID,
HOST, PORT, RELOAD,
CORS_ORIGINS, CORS_METHODS, CORS_HEADERS, CORS_CREDENTIALS,
PRELOAD_MODELS_ON_STARTUP, LOAD_STANCE_MODEL, LOAD_KPA_MODEL
)
# --- Import des singletons de services ---
from services.stance_model_manager import stance_model_manager
from services.label_model_manager import kpa_model_manager
from services.stt_service import speech_to_text
from services.tts_service import text_to_speech
# --- Lifespan / startup API ---
@asynccontextmanager
async def lifespan(app: FastAPI):
logger.info("="*60)
logger.info("🚀 API STARTUP - Loading HuggingFace models...")
logger.info("="*60)
if PRELOAD_MODELS_ON_STARTUP:
# Charger stance model
if LOAD_STANCE_MODEL:
try:
stance_model_manager.load_model(HUGGINGFACE_STANCE_MODEL_ID, HUGGINGFACE_API_KEY)
logger.info("✓ Stance model loaded")
except Exception as e:
logger.error(f"✗ Failed loading stance model: {e}")
# Charger KPA model
if LOAD_KPA_MODEL:
try:
kpa_model_manager.load_model(HUGGINGFACE_LABEL_MODEL_ID, HUGGINGFACE_API_KEY)
logger.info("✓ KPA model loaded")
except Exception as e:
logger.error(f"✗ Failed loading KPA model: {e}")
logger.info("✓ Startup complete. API ready.")
yield
logger.info("🛑 Shutting down...")
# --- FastAPI app ---
app = FastAPI(
title=API_TITLE,
description=API_DESCRIPTION,
version=API_VERSION,
lifespan=lifespan
)
# --- CORS ---
app.add_middleware(
CORSMiddleware,
allow_origins=CORS_ORIGINS,
allow_credentials=CORS_CREDENTIALS,
allow_methods=CORS_METHODS,
allow_headers=CORS_HEADERS,
)
# --- Routes ---
try:
from routes.stt_routes import router as stt_router
app.include_router(stt_router, prefix="/api/v1/stt", tags=["Speech To Text"])
logger.info("✓ STT route loaded (Groq Whisper)")
except Exception as e:
logger.warning(f"⚠ Failed loading STT route: {e}")
try:
from routes.tts_routes import router as tts_router
app.include_router(tts_router, prefix="/api/v1/tts", tags=["Text To Speech"])
logger.info("✓ TTS route loaded (Groq PlayAI TTS)")
except Exception as e:
logger.warning(f"⚠ Failed loading TTS route: {e}")
try:
from routes import api_router
app.include_router(api_router)
logger.info("✓ Main API routes loaded")
except Exception as e:
logger.warning(f"⚠ Failed loading main API routes: {e}")
# --- Basic routes ---
@app.get("/health")
async def health():
return {"status": "healthy", "service": "NLP Debater + Groq Voice"}
@app.get("/")
async def root():
return {
"message": "NLP Debater API with Groq Voice Support",
"docs": "/docs",
"voice_stt": "/api/v1/stt",
"voice_tts": "/api/v1/tts"
}
# --- Run server ---
if __name__ == "__main__":
uvicorn.run("main:app", host=HOST, port=PORT, reload=RELOAD)
|