Update app.py

app.py

@@ -2,10 +2,11 @@ import gradio as gr
 import random
 import os
 import json
-from huggingface_hub import InferenceClient
+import openai
 
-# Use Zephyr for general AI chatbot
-client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
+# 🔁 SWITCHED TO OPENROUTER + MISTRAL MODEL
+openai.api_base = "https://openrouter.ai/api/v1"
+openai.api_key = "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"
 
 # 🧠 MEMORY SETUP
 MEMORY_FILE = "hoopmind_memory.json"
@@ -32,13 +33,20 @@ def update_memory(user_input):
     save_memory(memory)
     return memory
 
-# 💬 Normal chatbot fallback
+# 💬 General fallback chat using Mistral
 def alpha_chatbot_response(user_input):
     try:
-        system_prompt = "You are a helpful, smart, and friendly AI assistant."
-        prompt = f"<|system|>\n{system_prompt}\n<|user|>\n{user_input}\n<|assistant|>"
-        response = client.text_generation(prompt, max_new_tokens=200, temperature=0.7)
-        return response.strip()
+        messages = [
+            {"role": "system", "content": "You are a helpful, smart, and friendly AI assistant."},
+            {"role": "user", "content": user_input}
+        ]
+        response = openai.ChatCompletion.create(
+            model="mistral/mistral-7b-instruct",  # or "mistral/mixtral-8x7b-instruct"
+            messages=messages,
+            temperature=0.7,
+            max_tokens=200
+        )
+        return response.choices[0].message["content"].strip()
     except Exception as e:
         return "Oops 😓 I had a lil error trying to respond. Try again in a bit!"
 
@@ -109,7 +117,7 @@ def hoopmind_chat(user_input, history=[]):
             "I’m your personal trainer + hype man + homie rolled in one 🧠🔥"
         ])
     
-    # Fallback: use Zephyr for normal chat
+    # Fallback: use Mistral
     return alpha_chatbot_response(user_input)
 
 # 🚀 CHAT INTERFACE
@@ -136,3 +144,4 @@ Drop a question or say what's up 👇
 chat.launch()
 
 
+