Update app.py

app.py

@@ -1,6 +1,3 @@
-import threading
-import os
-from flask import Flask, request, jsonify
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
@@ -9,23 +6,23 @@ MODEL_NAME = "microsoft/phi-3-mini-4k-instruct"
 MAX_TOKENS = 300
 TEMPERATURE = 0.7
 
+# === Chargement du modèle ===
 print("Chargement du modèle...")
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 model = AutoModelForCausalLM.from_pretrained(
     MODEL_NAME,
-    torch_dtype=torch.float32,
+    torch_dtype=torch.float32,  # pas de GPU
 )
-print("Modèle chargé ✅")
+print("✅ Modèle chargé avec succès")
 
+# === Fonction principale du chatbot ===
 def chat_with_phi3(message, history=None):
-    """Conversation Gradio & Flask"""
     if history is None:
         history = []
 
     system_prompt = (
-        "Tu es FINANFA, un assistant médical bienveillant et compétent. "
-        "Tu réponds uniquement aux questions de santé ou de médecine, "
-        "avec précision mais de manière compréhensible pour tous. "
+        "Tu es FINANFA, un assistant médical intelligent et empathique. "
+        "Tu ne réponds qu'aux questions liées à la santé, aux maladies ou aux médicaments. "
         "Si une question n’est pas médicale, dis poliment que tu ne peux pas répondre."
     )
 
@@ -47,54 +44,46 @@ def chat_with_phi3(message, history=None):
     response = response.split("Assistant :")[-1].strip()
     return response
 
-def gradio_chat(message, history):
-    return chat_with_phi3(message, history)
 
-chat_interface = gr.ChatInterface(
-    fn=gradio_chat,
-    title="🩺 FINANFA — Chatbot Médical",
-    description=(
-        "Cet assistant répond uniquement aux questions médicales. "
-        "⚠️ Il ne remplace pas un médecin qualifié."
-    ),
-    examples=[
-        "Quels sont les symptômes du paludisme ?",
-        "Comment prévenir l’hypertension ?",
-        "Quelle est la cause de la fièvre typhoïde ?",
-        "Quels sont les effets secondaires du paracétamol ?",
-    ],
-)
+# === Fonction API simulée (Flutter peut l’appeler via POST /api) ===
+def api_endpoint(message: str, history_json: str = "[]"):
+    import json
+    try:
+        history = json.loads(history_json)
+    except Exception:
+        history = []
+    response = chat_with_phi3(message, history)
+    return {"response": response}
 
-app = Flask(__name__)
 
-@app.route("/chat", methods=["POST"])
-def chat_api():
-    data = request.get_json()
-    if not data or "message" not in data:
-        return jsonify({"error": "Le champ 'message' est requis."}), 400
-    message = data["message"]
-    history = data.get("history", [])
-    response = chat_with_phi3(message, history)
-    return jsonify({"response": response})
+# === Interface Gradio ===
+with gr.Blocks(title="FINANFA — Chatbot Médical") as demo:
+    gr.Markdown(
+        """
+        # 🏥 FINANFA — Assistant Médical
+        Cet assistant répond uniquement aux questions médicales.  
+        ⚠️ Il ne remplace pas un avis médical professionnel.
+        """
+    )
 
-@app.route("/")
-def home():
-    return jsonify({
-        "message": "Bienvenue sur l’API FINANFA (chatbot médical).",
-        "endpoints": {
-            "POST /chat": {"params": {"message": "string", "history": "liste optionnelle"}},
-        },
-        "info": "Ce modèle répond uniquement aux questions médicales."
-    })
+    chat = gr.ChatInterface(
+        fn=chat_with_phi3,
+        title="FINANFA - Assistant Médical",
+        description="Posez vos questions médicales en toute sécurité.",
+        examples=[
+            "Quels sont les symptômes du paludisme ?",
+            "Comment prévenir le diabète ?",
+            "Quels médicaments pour la fièvre ?"
+        ]
+    )
 
-def launch_gradio():
-    chat_interface.launch(server_name="0.0.0.0", server_port=7860, share=False)
+    with gr.Row():
+        gr.Markdown("### 🔗 API Flutter")
+    message_box = gr.Textbox(label="Message (POST /api)", placeholder="Votre texte ici...")
+    history_box = gr.Textbox(label="Historique JSON (optionnel)", value="[]")
+    api_button = gr.Button("Tester API /api (simulateur)")
+    api_output = gr.JSON(label="Réponse API")
 
-def launch_flask():
-    port = int(os.getenv("PORT", 5000))
-    app.run(host="0.0.0.0", port=port)
+    api_button.click(api_endpoint, inputs=[message_box, history_box], outputs=api_output)
 
-if __name__ == "__main__":
-    flask_thread = threading.Thread(target=launch_flask)
-    flask_thread.start()
-    launch_gradio()
+demo.launch(server_name="0.0.0.0", server_port=7860)