import logging
from fastapi import FastAPI
from pydantic import BaseModel
from transformers import pipeline

logging.basicConfig(level=logging.INFO)

app = FastAPI()

logging.info("Loading translation model...")

translator = pipeline(
    "translation",
    model="Helsinki-NLP/opus-mt-synthetic-en-gd"
)

logging.info("Model loaded successfully")

class TranslateRequest(BaseModel):
    text: str

@app.get("/")
def health():
    logging.info("Health check called")
    return {"status": "ok"}

@app.post("/translate")
def translate(req: TranslateRequest):
    logging.info(f"Received text: {req.text}")

    result = translator(req.text, max_length=140)

    logging.info(f"Raw model output: {result}")

    return {
        "translatedText": result[0]["translation_text"],
        "phonetics": ""
    }