Update app.py

app.py

@@ -6,16 +6,12 @@ from pydantic import BaseModel
 import uvicorn
 
 HF_TOKEN = os.environ.get("HF_TOKEN")
-
-# Correct endpoint for your model
 API_URL = "https://router.huggingface.co/hf-inference/models/mistralai/Mistral-7B-Instruct-v0.3"
 
-SYSTEM_PROMPT = "Tu es l'assistant SAV officiel du Centre Chery Tunisie. Tu réponds en français ou en arabe dialectal tunisien selon la langue du client."
+SYSTEM_PROMPT = "Tu es l'assistant SAV officiel du Centre Chery Tunisie. Tu réponds en français ou en arabe dialectal tunisien selon la langue du client. Réponds uniquement aux questions liées aux véhicules Chery, la maintenance, les pannes, et le service après-vente."
 
 def chat(message, history):
-    # Build prompt in Mistral instruct format
     prompt = f"[INST] <<SYS>>\n{SYSTEM_PROMPT}\n<</SYS>>\n\n"
-    
     for h in history:
         if isinstance(h, dict):
             if h["role"] == "user":
@@ -24,7 +20,6 @@ def chat(message, history):
                 prompt += f"{h['content']} </s><s>[INST] "
         else:
             prompt += f"{h[0]} [/INST] {h[1]} </s><s>[INST] "
-    
     prompt += f"{message} [/INST]"
 
     headers = {"Authorization": f"Bearer {HF_TOKEN}"}
@@ -38,9 +33,7 @@ def chat(message, history):
     }
 
     try:
-        response = requests.post(API_URL, headers=headers, json=payload, timeout=120)
-        print("Status:", response.status_code)
-        print("Body:", response.text[:300])
+        response = requests.post(API_URL, headers=headers, json=payload, timeout=60)
         result = response.json()
         if isinstance(result, list):
             return result[0].get("generated_text", "Pas de réponse")