from fastapi import APIRouter

from config import settings
from models.responses import HealthResponse

router = APIRouter()


@router.get("/health", response_model=HealthResponse)
async def health():
    from main import vectorstore_service

    if vectorstore_service is None:
        return HealthResponse(
            status="loading",
            chromadb="not_connected",
            embedding_model=settings.embedding_model,
            documents_count=0,
            chunks_count=0,
        )

    stats = vectorstore_service.get_stats()
    return HealthResponse(
        status="ok",
        chromadb="connected",
        embedding_model=settings.embedding_model,
        documents_count=stats["documents_count"],
        chunks_count=stats["chunks_count"],
    )