from fastapi import APIRouter, HTTPException

from app.modules.search.jobs import get_job
from app.modules.search.schemas import SearchStatusResponse

router = APIRouter(prefix="/search", tags=["search"])


@router.get("/status/{job_id}", response_model=SearchStatusResponse)
def search_status(job_id: str):
    """Poll this until status is 'done' (or 'error'). On 'done',
    clustered_papers is the final payload the UI renders."""
    job = get_job(job_id)
    if job is None:
        raise HTTPException(status_code=404, detail="Unknown job_id.")
    return SearchStatusResponse(**job)