Update app.py

app.py

@@ -27,7 +27,6 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-HF_TOKEN = os.getenv("REACT_APP_HF_TOKEN")
 
 @app.get("/invert")
 async def invert(text: str):
@@ -36,25 +35,41 @@ async def invert(text: str):
         "inverted": text[::-1],
     }
 
+HF_TOKEN = os.getenv("REACT_APP_HF_TOKEN")
+if not HF_TOKEN:
+    raise RuntimeError("Le token Hugging Face (HF_TOKEN) n'est pas défini dans les variables d'environnement.")
+
+# Initialisation du client HF
+hf_client = InferenceClient(
+    provider="novita",
+    api_key=HF_TOKEN,
+)
+
 @app.post("/api/chat")
 async def chat(request: Request):
     data = await request.json()
-    user_message = data.get("message", "")
-    headers = {"Authorization": f"Bearer {HF_TOKEN}"}
-    payload = {
-        "inputs": {
-            "past_user_inputs": [],
-            "generated_responses": [],
-            "text": user_message
-        }
-    }
-    response = requests.post(
-        "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.3",
-        headers=headers,
-        json=payload
-    )
-    result = response.json()
-    return {"response": result}
+    user_message = data.get("message", "").strip()
+    if not user_message:
+        raise HTTPException(status_code=400, detail="Le champ 'message' est requis.")
+
+    try:
+        # Appel au modèle en mode chat
+        completion = hf_client.chat.completions.create(
+            model="mistralai/Mistral-7B-Instruct-v0.3",
+            messages=[
+                {"role": "system", "content": "Tu es un assistant médical spécialisé en schizophrénie."},
+                {"role": "user",   "content": user_message}
+            ],
+            max_tokens=512,
+            temperature=0.7,
+        )
+
+        bot_msg = completion.choices[0].message.content
+        return {"response": bot_msg}
+
+    except Exception as e:
+        # En cas d'erreur d'inférence
+        raise HTTPException(status_code=502, detail=f"Erreur d'inférence HF : {e}")
 
 
 @app.get("/data")