Update app.py

app.py

@@ -1,38 +1,23 @@
 import gradio as gr
 import requests
 import os
-from fastapi import FastAPI
-from pydantic import BaseModel
 
-# 🔐 Token
 HF_TOKEN = os.environ.get("HF_TOKEN")
+API_URL = "https://router.huggingface.co/hf-inference/models/mistralai/Mistral-7B-Instruct-v0.3"
 
-# ✅ Use router endpoint (good choice)
-API_URL = "https://api-inference.huggingface.co/models/dali4444444/chery-sav-chatbot"
-# 🧠 System prompt
 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."""
+Réponds uniquement aux questions liées aux véhicules Chery."""
 
-# 💬 Chat function
 def chat(message, history):
     prompt = f"[INST] <<SYS>>\n{SYSTEM_PROMPT}\n<</SYS>>\n\n"
 
     for h in history:
-        if isinstance(h, dict):
-            if h.get("role") == "user":
-                prompt += f"{h.get('content')} [/INST] "
-            else:
-                prompt += f"{h.get('content')} </s><s>[INST] "
-        else:
-            prompt += f"{h[0]} [/INST] {h[1]} </s><s>[INST] "
+        prompt += f"{h[0]} [/INST] {h[1]} </s><s>[INST] "
 
     prompt += f"{message} [/INST]"
 
-    headers = {
-        "Authorization": f"Bearer {HF_TOKEN}",
-        "Content-Type": "application/json"
-    }
+    headers = {"Authorization": f"Bearer {HF_TOKEN}"}
 
     payload = {
         "inputs": prompt,
@@ -47,50 +32,21 @@ def chat(message, history):
 
     try:
         response = requests.post(API_URL, headers=headers, json=payload, timeout=60)
-
-        # ✅ Check HTTP error
-        if response.status_code != 200:
-            return f"Erreur HTTP {response.status_code}: {response.text}"
-
         result = response.json()
 
-        # ✅ Handle response formats
         if isinstance(result, list):
             return result[0].get("generated_text", "Pas de réponse")
 
-        if isinstance(result, dict):
-            if "error" in result:
-                if "loading" in result["error"].lower():
-                    return "⏳ Le modèle est en cours de chargement, réessayez dans quelques secondes."
-                return f"Erreur API: {result['error']}"
+        if "error" in result:
+            return f"Erreur: {result['error']}"
 
         return str(result)
 
-    except requests.exceptions.Timeout:
-        return "⏱️ Timeout: le modèle met trop de temps à répondre."
     except Exception as e:
         return f"Erreur: {str(e)}"
 
 
-# 🚀 FastAPI
-app = FastAPI()
-
-class ChatRequest(BaseModel):
-    message: str
-    history: list = []
-
-
-@app.post("/api/chat")
-async def api_chat(req: ChatRequest):
-    return {"reply": chat(req.message, req.history)}
-
-
-# 🎨 Gradio UI
-demo = gr.ChatInterface(
-    fn=chat,
-    title="🚗 Chery SAV Assistant",
-    description="Assistant intelligent pour le service après-vente Chery Tunisie"
-)
+demo = gr.ChatInterface(fn=chat, title="🚗 Chery SAV Assistant")
 
-# 🔗 Mount Gradio inside FastAPI
-app = gr.mount_gradio_app(app, demo, path="/")
+# 🔥 THIS LINE IS THE KEY
+demo.launch(server_name="0.0.0.0", server_port=7860)