src/apis/language_translation_api.py

"""
Created By: ishwor subedi
Date: 2024-08-13
"""
from fastapi import HTTPException
from fastapi.routing import APIRouter
from src.models.models import LanguageTranslationRequest
from src.pipeline.language_translation_pipeline import LanguageTranslationPipeline

language_translation_router = APIRouter()

language_translation_pipeline = LanguageTranslationPipeline()


@language_translation_router.post("/translate")
async def translate_text(language_translation_info: LanguageTranslationRequest):
    try:
        translated_text = language_translation_pipeline.translate_text(
            text=language_translation_info.text,
            translator_backend_code=language_translation_info.translator_backend_code,
            target=language_translation_info.target
        )
        return {"translated_text": translated_text, 'status': 200}
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"An unexpected error occurred: {str(e)}")