from flask import Flask, request, jsonify
from gradio_client import Client

# Initialisation
app = Flask(__name__)
client = Client("Qwen/Qwen1.5-72B-Chat")

# Prompt système
system_prompt = "Tu es un assistant juridique spécialisé dans le droit du travail au Bénin. Donne des réponses claires, précises et utiles."

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

    if not data or "question" not in data:
        return jsonify({"error": "Champ 'question' manquant"}), 400

    question = data["question"]

    try:
        response = client.predict(
            question,
            [],  # pas d'historique
            system_prompt,
            api_name="/model_chat"
        )
        reponse_finale = response[1][-1][1]
        return jsonify({"reponse": reponse_finale})
    except Exception as e:
        return jsonify({"error": str(e)}), 500

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