from fastapi import APIRouter
from fastapi.responses import JSONResponse
from .model import Recommendations, RecommendType
from .controller import recycle_recommendation


router = APIRouter(
    prefix="/recommend",
    tags=["recommend"],
    default_response_class=JSONResponse,
)


@router.post("/", response_model=Recommendations)
async def recommend(data: RecommendType):
    return recycle_recommendation(data.recommend_type)