from fastapi import APIRouter, Request
from fastapi.templating import Jinja2Templates

router = APIRouter(prefix="/ui", tags=["ui"])
templates = Jinja2Templates(directory="app/templates")


@router.get("/predict")
async def predict_page(request: Request):
    return templates.TemplateResponse("predict.html", {"request": request})


@router.get("/health")
async def health_page(request: Request):
    return templates.TemplateResponse("health.html", {"request": request})