routes.py

from fastapi import APIRouter, HTTPException
from app.predict import text_predictor
from app.schemas import TextRequest, PredictionResponse

router = APIRouter()

@router.post("/api/predict", response_model=PredictionResponse)
def predict_text(request: TextRequest):
    try:
        result = text_predictor.predict(request.text)
        return result
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))