src/api/fix_newlines.py

from fastapi import APIRouter, HTTPException, Request

from src.schemas.requests import FixNewlinesRequest
from src.schemas.responses import FixNewlinesResponse

router = APIRouter()


@router.post("/fix-newlines", response_model=FixNewlinesResponse)
def fix_newlines(request: Request, body: FixNewlinesRequest):
    pipeline = request.app.state.one_model_pipeline
    fixed = pipeline.predict(body.text)
    return FixNewlinesResponse(fixed_text=fixed, model_used=pipeline.model_name)