Spaces:
Runtime error
Runtime error
# app/routers/synonyms.py | |
import logging | |
from fastapi import APIRouter, Depends, HTTPException, status # Import HTTPException and status for validation | |
from app.schemas.base import TextOnlyRequest | |
from app.services.synonyms import SynonymSuggester # Import the service class | |
from app.core.security import verify_api_key # Assuming API key verification is still used | |
from app.core.config import APP_NAME # For logger naming | |
from app.core.exceptions import ServiceError # For re-raising internal errors | |
logger = logging.getLogger(f"{APP_NAME}.routers.synonyms") | |
router = APIRouter(prefix="/synonyms", tags=["Synonyms"]) | |
# Initialize service instance once per application lifecycle | |
synonym_suggester_service = SynonymSuggester() | |
# Added /suggest path for clarity | |
async def suggest_synonyms_endpoint(payload: TextOnlyRequest): | |
""" | |
Suggests synonyms for words in the provided text. | |
""" | |
text = payload.text.strip() | |
if not text: | |
raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail="Input text cannot be empty.") | |
logger.info(f"Received synonym suggestion request for text (first 50 chars): '{text[:50]}...'") | |
try: | |
# Directly call the async service method | |
# ModelNotDownloadedError will be raised here if model/data is missing, | |
# and caught by the global exception handler in app/main.py | |
result = await synonym_suggester_service.suggest(text) | |
logger.info(f"Synonym suggestion successful for text (first 50 chars): '{text[:50]}...'") | |
return {"synonyms": 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 synonym suggestion endpoint for text: '{text[:50]}...'") | |
raise ServiceError(status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, detail="An unexpected error occurred during synonym suggestion.") from e |