from flask import Flask, request, jsonify
import requests
import os


app = Flask(__name__)
HF_SPACE_URL = "https://R-TA-NLLB-200-AI.hf.space"

def translate_text(text, source_lang="eng_Latn", target_lang="urd_Arab"):
    try:
        response = requests.post(
            f"{HF_SPACE_URL}/translate",
            json={
                "inputs": {
                    "text": text,
                    "source_lang": source_lang,
                    "target_lang": target_lang
                }
            },
            timeout=30
        )
        return response.json().get("translation", "")
    except Exception as e:
        return str(e)

@app.route("/")
def home():
    return jsonify({"status": "✅ NLLB Translator is running!"})

@app.route("/translate", methods=["POST"])
def translate():
    data = request.get_json()
    translation = translate_text(
        data.get("text", ""),
        data.get("source_lang", "eng_Latn"),
        data.get("target_lang", "urd_Arab")
    )
    return jsonify({"translation": translation})

if __name__ == "__main__":
    app.run(host="0.0.0.0", port=8080)