Update app.py

app.py

@@ -236,18 +236,37 @@ def ask_llama(conversation_history, category, is_final_guess=False):
 
 def ask_help_agent(query):
     try:
-        API_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.1"
-        headers = {"Authorization": "Bearer HF_TOKEN"}  # Get from huggingface.co/settings/tokens
+        import ollama
+        import requests
+
+        # (1) Check if Ollama server is running
+        try:
+            requests.get("http://localhost:11434", timeout=5)
+        except:
+            return "🛑 **Ollama is not running!**\n\nPlease:\n1. [Download Ollama](https://ollama.com)\n2. Run `ollama serve` in terminal\n3. Pull a model (`ollama pull llama3`)"
+
+        # (2) Build chat history
+        messages = [{"role": "system", "content": "You are a helpful AI assistant."}]
         
-        prompt = f"""<s>[INST] <<SYS>>
-        You are a helpful assistant. Answer concisely.
-        <</SYS>>
-        {query} [/INST]"""
+        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"]})
         
-        response = requests.post(API_URL, headers=headers, json={"inputs": prompt})
-        return response.json()[0]['generated_text'].split("[/INST]")[-1].strip()
+        messages.append({"role": "user", "content": query})
+
+        # (3) Get response
+        response = ollama.chat(
+            model="llama3",  # or "mistral" for lighter model
+            messages=messages,
+            options={"temperature": 0.7}
+        )
+        return response['message']['content']
+
     except Exception as e:
-        return f"Assistant unavailable: {str(e)}"
+        return f"⚠️ **Assistant Error**\n\n{str(e)}\n\nPlease ensure Ollama is installed and running."
 
 # Main game logic with enhanced UI
 def main():