src/api/fix_newlines_all_models.py

from fastapi import APIRouter, HTTPException, Request

from src.schemas.requests import FixNewlinesAllModelsRequest
from src.schemas.responses import FixNewlinesAllModelsResponse, ModelResult

router = APIRouter()


@router.post("/fix-newlines-all-models", response_model=FixNewlinesAllModelsResponse)
def fix_newlines_all_models(request: Request, body: FixNewlinesAllModelsRequest):
    pipeline = request.app.state.all_models_pipeline
    results_dict = pipeline.predict(body.text)
    results = [
        ModelResult(model_name=name, fixed_text=text)
        for name, text in results_dict.items()
    ]
    return FixNewlinesAllModelsResponse(results=results)