from pathlib import Path

from fastapi import APIRouter, HTTPException, Request
from fastapi.responses import HTMLResponse
from fastapi.templating import Jinja2Templates

from app.services.engine_registry import get_engine_registry
from app.services.job_store import get_job, list_jobs

templates = Jinja2Templates(
    directory=str(Path(__file__).resolve().parent.parent / "templates")
)

router = APIRouter(prefix="", tags=["ui"])

UI_MODEL_LABELS = {
    "supertonic-2": "Supertonic",
    "kitten-tts-micro-0.8": "Kitten",
    "kokoro": "Kokoro",
    "melo-en": "Melo",
}


@router.get("/", response_class=HTMLResponse)
async def index_page(request: Request) -> HTMLResponse:
    registry = get_engine_registry()
    ui_models: list[dict[str, str]] = []
    voice_map: dict[str, list[dict]] = {}
    for model, label in UI_MODEL_LABELS.items():
        try:
            engine = registry.get_engine(model)
            ui_models.append({"id": model, "label": label})
            voice_map[model] = [
                {"alias": b["alias"], "canonical_name": b["canonical_name"]}
                for b in engine.list_voice_bindings()
                if b.get("source") != "private"
            ]
        except Exception:
            continue
    return templates.TemplateResponse(
        "index.html",
        {
            "request": request,
            "ui_models": ui_models,
            "voice_map": voice_map,
            "recent_jobs": list_jobs(limit=5),
        },
    )


@router.get("/jobs/{job_id}", response_class=HTMLResponse)
async def job_page(request: Request, job_id: str) -> HTMLResponse:
    job = get_job(job_id)
    if job is None:
        raise HTTPException(status_code=404, detail="Job not found")
    return templates.TemplateResponse("job.html", {"request": request, "job": job})