24 lines
617 B
Python
24 lines
617 B
Python
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
|
|
) |