convert to flask api

app.py

@@ -1,21 +1,33 @@
-import gradio as gr
+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."
 
-def assistant(question):
-    response = client.predict(
-        question,
-        [],  # pas d'historique
-        system_prompt,
-        api_name="/model_chat"
-    )
-    return response[1][-1][1]  # dernière réponse texte
+@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"]
 
-demo = gr.Interface(fn=assistant,
-                    inputs=gr.Textbox(label="Question"),
-                    outputs=gr.Textbox(label="Réponse"),
-                    allow_flagging="never")
+    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
 
-demo.launch()
+if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=7860)

requirements.txt

@@ -1,2 +1,4 @@
 gradio
 gradio_client
+request
+flask