from flask import Flask, request, jsonify
from transformers import pipeline

app = Flask(__name__)

translator = pipeline("translation", model="Bildad/Swahili-English_Translation")


@app.route("/translate", methods=["POST"])
def translate():
    data = request.get_json()

    if "text" not in data:
        return jsonify({"error": "No text provided"}), 400

    text_to_translate = data["text"]
    translation = translator(text_to_translate)[0]
    translated_text = translation["translation_text"]

    return jsonify({"translated_text": translated_text})


if __name__ == "__main__":
    from waitress import serve

    serve(app, host="0.0.0.0", port=10000)