from fastapi import APIRouter

from app.models.dto.text_request import TextRequest
from app.models.response.moderation_response import ModerationResponse
from app.services.text_moderation_service import TextModerationService

router = APIRouter(
    prefix="/api/v1/moderation",
    tags=["Moderation"]
)

service = TextModerationService()


@router.post("/text", response_model=ModerationResponse)
def moderate_text(request: TextRequest):

  prediction = service.moderate(request.text)

  return ModerationResponse(
      approved=prediction.approved,
      score=prediction.score,
      reason=prediction.reason
  )