from fastapi import FastAPI
from pydantic import BaseModel
from fastapi.middleware.cors import CORSMiddleware
from deep_translator import GoogleTranslator

app = FastAPI()

app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

class TranslationRequest(BaseModel):
    text: str
    src_lang: str
    dest_lang: str

@app.post("/translate")
async def translate_text(req: TranslationRequest):
    try:
        translated = GoogleTranslator(source=req.src_lang, target=req.dest_lang).translate(req.text)
        return {"translated_text": translated}
    except Exception as e:
        return {"error": str(e)}