from fastapi import APIRouter

from backend.database import db
from backend.schemas import DashboardResponse, RecommendationRequest
from services.recommendation_service import score_recommendation

router = APIRouter(prefix='/api/v1/stats', tags=['stats'])


@router.get('/summary', response_model=DashboardResponse)
def stats_summary() -> dict:
    return db.stats()


@router.get('/heatmap')
def stats_heatmap() -> dict:
    return {'items': db.heatmap()}


@router.post('/recommendation')
def stats_recommendation(payload: RecommendationRequest) -> dict:
    result = score_recommendation(
        age=payload.age,
        elapsed_hours=payload.elapsed_hours,
        terrain=payload.terrain_primary,
        weather=payload.weather,
        health_flags=payload.health_flags,
    )
    return {
        'recommendation': result['recommendation'],
        'score': result['score'],
    }