from fastapi import FastAPI
from pydantic import BaseModel
from openai import OpenAI
import os
import logging
# Logging Configuration
logging.basicConfig(
    level=logging.INFO,
    format="%(asctime)s %(levelname)s %(name)s : %(message)s"
)
logger = logging.getLogger(__name__)
# OpenAI Client
client = OpenAI(
    api_key=os.getenv("OPENAI_API_KEY")
)
# FastAPI App
app = FastAPI()
# Request Model
class TranslationRequest(BaseModel):
    text: str
    targetLanguage: str
# Locale Mapping
LOCALE_MAP = {
    "es-PR": "Puerto Rico Spanish",
    "es-MX": "Mexican Spanish",
    "es-ES": "Spain Spanish",
    "es-US": "United States Spanish",
    "fr-CA": "Canadian French",
    "pt-BR": "Brazilian Portuguese"
}
# Translation API
@app.post("/api/translations")
def translate(request: TranslationRequest):
    locale_name = LOCALE_MAP.get(
        request.targetLanguage,
        request.targetLanguage
    )
    prompt = f"""
Translate the following text into natural {locale_name}.
Rules:
- Preserve tone and meaning
- Keep placeholders unchanged
- Keep HTML tags unchanged
- Avoid robotic translation
- Return only the translated text
Text:
{request.text}
"""
    # Logging
    logger.info("======================================")
    logger.info("Translation Request Received")
    logger.info("Target Language: %s", request.targetLanguage)
    logger.info("Locale Name: %s", locale_name)
    logger.info("Original Text: %s", request.text)
    logger.info("Prompt:\n%s", prompt)
    try:
        response = client.chat.completions.create(
            model="gpt-5",
            messages=[
                {
                    "role": "system",
                    "content": "You are a professional localization translator."
                },
                {
                    "role": "user",
                    "content": prompt
                }
            ]
        )
        translated_text = response.choices[0].message.content
        logger.info("Translated Text: %s", translated_text)
        logger.info("======================================")
        return {
            "translatedText": translated_text,
            "targetLanguage": request.targetLanguage
        }
    except Exception as e:
        logger.exception("Translation Error Occurred")
        return {
            "error": str(e)
        }