"""
FastAPI app exposing /health and /recommend
Run: uvicorn src.app:app --reload
"""
from fastapi import FastAPI, HTTPException
from pydantic import BaseModel, AnyUrl
from typing import Optional, List
from recommender import Recommender

app = FastAPI(title="SHL Assessment Recommender")

rec = None

class RecommendRequest(BaseModel):
    query: Optional[str] = None
    url: Optional[AnyUrl] = None
    k: int = 10

class AssessmentItem(BaseModel):
    assessment_name: str
    url: AnyUrl
    test_type: Optional[str] = None
    assessment_length_min: Optional[int] = None
    languages: Optional[List[str]] = []

class RecommendResponse(BaseModel):
    results: List[AssessmentItem]

@app.on_event("startup")
def load_rec():
    global rec
    rec = Recommender()

@app.get("/health")
def health():
    return {"status": "ok"}

@app.post("/recommend", response_model=RecommendResponse)
def recommend(body: RecommendRequest):
    if not body.query and not body.url:
        raise HTTPException(status_code=400, detail="Provide either 'query' or 'url'")
    k = max(1, min(10, body.k))
    try:
        items = rec.recommend(query=body.query, jd_url=str(body.url) if body.url else None, k=k)
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))
    return {"results": items}