Update app.py

app.py

@@ -5,8 +5,10 @@ import numpy as np
 import tempfile
 import scipy.io.wavfile
 import requests
-import openai
-OPENAI_API_KEY = os.getenv("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")
+
+# Set your API keys here
+os.environ["MISTRAL_API_KEY"] = "R1ISnVkHrj7fSd5Dh6ZSZHqCJhhct0ZR"
+os.environ["GROQ_API_KEY"] = "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"
 
 state = {
     "started": False,
@@ -31,21 +33,46 @@ def convert_audio_to_text(audio, lang_choices):
                 return recog.recognize_google(recorded, language=lang).lower()
     except Exception as e:
         return f"[Error] Couldn't transcribe: {str(e)}"
+def query_llm(api, messages, model=None):
+    headers = {
+        "Authorization": f"Bearer {os.environ[f'{api}_API_KEY']}",
+        "Content-Type": "application/json"
+    }
+    payload = {
+        "messages": messages,
+        "model": model or ("llama3-70b-8192" if api == "GROQ" else "mistral-medium")
+    }
+    endpoint = {
+        "MISTRAL": "https://api.mistral.ai/v1/chat/completions",
+        "GROQ": "https://api.groq.com/openai/v1/chat/completions"
+    }[api]
+
+    response = requests.post(endpoint, headers=headers, json=payload)
 
-def ask_openai(prompt):
+    if response.status_code == 200:
+        return response.json()["choices"][0]["message"]["content"]
+    else:
+        return f"[{api} Error] {response.status_code} - {response.text}"
+
+def ask_mistral(prompt):
+    url = "https://api.groq.com/openai/v1/chat/completions"
     headers = {
-        "Authorization": f"Bearer {OPENAI_API_KEY}",
+        "Authorization": f"Bearer {os.environ['GROQ_API_KEY']}",
         "Content-Type": "application/json"
     }
+
     data = {
-        "model": "gpt-3.5-turbo",
-        "messages": [{"role": "user", "content": prompt}]
+        "model": "mistral-7b-8k",
+        "messages": [{"role": "user", "content": prompt}],
+        "temperature": 0.7
     }
 
-    response = requests.post("https://api.openai.com/v1/chat/completions", headers=headers, json=data)
+    response = requests.post(url, headers=headers, json=data)
+
     if response.status_code == 200:
         return response.json()["choices"][0]["message"]["content"]
-    return "[OpenAI Error]"
+    else:
+        return f"[Mistral Error] {response.status_code} - {response.text}"
 
 def begin_game():
     state["started"] = True
@@ -92,7 +119,7 @@ def get_next_question():
 {history_prompt}
 
 Your question:"""
-    return ask_openai(prompt).strip()
+    return ask_mistral(prompt).strip()
 
 def generate_final_guess():
     history = "\n".join([f"Q: {q}\nA: {a}" for q, a in state["history"]])
@@ -101,7 +128,7 @@ def generate_final_guess():
 {history}
 
 Your guess:"""
-    return ask_openai(prompt)
+    return ask_mistral(prompt)
 
 def hint_response():
     if not state["hint_mode"] or not state["started"]:
@@ -109,7 +136,7 @@ def hint_response():
     question = state["current_q"]
     history = "\n".join([f"{q} - {a}" for q, a in state["history"]])
     prompt = f"""The player seems confused by this question: "{question}". Previous Q&A were:\n{history}\n\nGive a helpful, simple hint."""
-    return ask_openai(prompt)
+    return ask_mistral(prompt)
 
 def toggle_hint_mode():
     state["hint_mode"] = not state["hint_mode"]