from fastapi import APIRouter, HTTPException
from app.schemas.chat import ChatRequest, ChatResponse
from app.services.qa import get_answer
import time

router = APIRouter()

@router.post("/", response_model=ChatResponse)
async def chat(request: ChatRequest):
    try:
        start_time = time.time()
        answer = get_answer(request.message)
        latency_ms = int((time.time() - start_time) * 1000)
        return ChatResponse(answer=answer, latency_ms=latency_ms)
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))