Update app.py

app.py

@@ -233,31 +233,50 @@ def ask_llama(conversation_history, category, is_final_guess=False):
         return "Could not generate question"
 
 # New function for the help AI assistant using the Hugging Face InferenceClient
-
+MISTRAL_API_KEY = "wm5eLl09b9I9cOxR3E9n5rrRr1CRQQjn"
 def ask_help_agent(query):
     try:
-        API_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.1"
-        headers = {"Authorization": "Bearer hf_wm5eLl09b9I9cOxR3E9n5rrRr1CRQQjn"}  # ⚠️ Replace with your token
-        
-        prompt = f"""<s>[INST] <<SYS>>
-        You are a helpful assistant. Keep answers short.
-        <</SYS>>
-        {query} [/INST]"""
+        # Prepare Mistral API request
+        url = "https://api.mistral.ai/v1/chat/completions"
+        headers = {
+            "Authorization": f"Bearer {MISTRAL_API_KEY}",
+            "Content-Type": "application/json"
+        }
+
+        system_message = "You are a friendly Chatbot."
         
-        response = requests.post(
-            API_URL,
-            headers=headers,
-            json={"inputs": prompt},
-            timeout=10
-        )
+        # Build message history
+        messages = [{"role": "system", "content": system_message}]
+        if "help_conversation" in st.session_state:
+            for msg in st.session_state.help_conversation:
+                if msg.get("query"):
+                    messages.append({"role": "user", "content": msg["query"]})
+                if msg.get("response"):
+                    messages.append({"role": "assistant", "content": msg["response"]})
         
+        # Add current user query
+        messages.append({"role": "user", "content": query})
+
+        # API payload
+        payload = {
+            "model": "mistral-tiny",
+            "messages": messages,
+            "temperature": 0.7,
+            "top_p": 0.95
+        }
+
+        # Send POST request
+        response = requests.post(url, headers=headers, json=payload)
+
         if response.status_code == 200:
-            return response.json()[0]['generated_text'].split("[/INST]")[-1].strip()
+            result = response.json()
+            return result["choices"][0]["message"]["content"]
         else:
-            return f"Error: {response.text}"
-            
+            return f"API Error {response.status_code}: {response.text}"
+
     except Exception as e:
-        return f"Assistant busy. Try later. ({str(e)})"
+        return f"Error in help agent: {str(e)}"
+
 
 # Main game logic with enhanced UI
 def main():