app/main.py

import logging
from contextlib import asynccontextmanager
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from app.config import settings
from app.services.transcription import transcription_service
from app.services.llm import gemini_service
from app.routers import health, session

logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)


@asynccontextmanager
async def lifespan(app: FastAPI):
    logger.info("Starting up...")
    transcription_service.load_model(settings.whisper_model)
    gemini_service.initialize()
    logger.info("Startup complete")
    yield
    logger.info("Shutting down...")


app = FastAPI(title="AI Checklist Agent", lifespan=lifespan)

app.add_middleware(
    CORSMiddleware,
    allow_origins=settings.cors_origins,
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

app.include_router(health.router)
app.include_router(session.router)