main.py

import sys
from pathlib import Path
import logging
from contextlib import asynccontextmanager

logging.basicConfig(
    level=logging.INFO,
    format="%(asctime)s - %(name)s - %(levelname)s - %(message)s"
)
logger = logging.getLogger(__name__)

# Add app directory
app_dir = Path(__file__).parent
sys.path.insert(0, str(app_dir))

from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
import uvicorn

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
)

@asynccontextmanager
async def lifespan(app: FastAPI):
    logger.info("="*60)
    logger.info("🚀 API STARTUP - Loading HuggingFace models...")
    logger.info("="*60)

    if PRELOAD_MODELS_ON_STARTUP:

        # Load stance model
        if LOAD_STANCE_MODEL:
            try:
                from services.stance_model_manager import load_model as load_stance
                load_stance(HUGGINGFACE_STANCE_MODEL_ID, HUGGINGFACE_API_KEY)
                logger.info("✓ Stance model loaded")
            except Exception as e:
                logger.error(f"✗ Failed loading stance model: {e}")

        # Load KPA model
        if LOAD_KPA_MODEL:
            try:
                from services.label_model_manager import load_model as load_kpa
                load_kpa(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 ============

# STT route (Groq Whisper)
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}")

# TTS route (Groq PlayAI TTS)
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}")

# Main NLP system routes
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"
    }

if __name__ == "__main__":
    uvicorn.run("main:app", host=HOST, port=PORT, reload=RELOAD)