File size: 2,068 Bytes
73a6a7e
 
 
 
 
 
 
 
 
 
1cfeb58
71192d1
73a6a7e
 
 
 
22c7fb1
b7b6eaf
73a6a7e
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
import logging
from fastapi import APIRouter, Depends, HTTPException, status 

from app.schemas.base import TranslateRequest 
from app.services.translation import Translator 
from app.core.security import verify_api_key 
from app.core.config import APP_NAME 
from app.core.exceptions import ServiceError 

logger = logging.getLogger(f"{APP_NAME}.routers.translate")

router = APIRouter(prefix="/translate", tags=["Translate"])


translator_service = Translator()


@router.post("/", dependencies=[Depends(verify_api_key)])
async def translate_text_endpoint(payload: TranslateRequest):
    """
    Translates the provided text to a specified target language.
    """
    text = payload.text.strip()
    target_lang = payload.target_lang.strip()

    if not text:
        raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail="Input text cannot be empty.")
    if not target_lang:
        raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail="Target language cannot be empty.")

    logger.info(f"Received translation request for text (first 50 chars): '{text[:50]}...' to '{target_lang}'")

    try:
        # Directly call the async service method
        # ModelNotDownloadedError will be raised here if model is missing,
        # and caught by the global exception handler in app/main.py
        result = await translator_service.translate(text, target_lang)

        logger.info(f"Translation successful for text (first 50 chars): '{text[:50]}...' to '{target_lang}'")
        return {"translation": result} # Consistent key for response

    except ServiceError as e:
        # Re-raise ServiceError. It will be caught by the global exception handler.
        raise e
    except Exception as e:
        # Catch any unexpected exceptions and re-raise as a generic ServiceError
        logger.exception(f"Unhandled error in translation endpoint for text: '{text[:50]}...' to '{target_lang}'")
        raise ServiceError(status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, detail="An unexpected error occurred during translation.") from e