Update app.py

app.py

@@ -101,11 +101,11 @@ def show_confetti():
     </script>
     """)
 
-# Mistral AI API call
-def ask_mistral(conversation_history, category):
-    api_url = "https://api.mistral.ai/v1/chat/completions"
+# Groq Llama AI API call
+def ask_llama(conversation_history, category):
+    api_url = "https://api.groq.ai/v1/chat/completions"
     headers = {
-        "Authorization": "Bearer gz6lDXokxgR6cLY72oomALWcm7vhjzQ",
+        "Authorization": "Bearer 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",
         "Content-Type": "application/json"
     }
     
@@ -117,7 +117,7 @@ def ask_mistral(conversation_history, category):
     ] + conversation_history
     
     data = {
-        "model": "mistral-tiny",
+        "model": "llama-3.3-70b-versatile",
         "messages": messages,
         "temperature": 0.7,
         "max_tokens": 100
@@ -128,7 +128,7 @@ def ask_mistral(conversation_history, category):
         response.raise_for_status()
         return response.json()["choices"][0]["message"]["content"]
     except Exception as e:
-        st.error(f"Error calling Mistral API: {str(e)}")
+        st.error(f"Error calling Llama API: {str(e)}")
         return "Could not generate question"
 
 # Game logic
@@ -173,9 +173,10 @@ def main():
                 else:
                     st.session_state.category = category_input
                     # Generate first question
-                    first_question = ask_mistral([
-                        {"role": "user", "content": "Ask your first yes/no question."}
-                    ], category_input)
+                    first_question = ask_llama([{
+                        "role": "user", 
+                        "content": "Ask your first yes/no question."
+                    }], category_input)
                     st.session_state.questions = [first_question]
                     st.session_state.conversation_history = [
                         {"role": "assistant", "content": first_question}
@@ -208,7 +209,7 @@ def main():
                         st.session_state.game_state = "result"
                     else:
                         # Generate next question
-                        next_question = ask_mistral(
+                        next_question = ask_llama(
                             st.session_state.conversation_history,
                             st.session_state.category
                         )
@@ -223,7 +224,7 @@ def main():
     # Result screen
     elif st.session_state.game_state == "result":
         # Generate final guess
-        final_guess = ask_mistral(
+        final_guess = ask_llama(
             st.session_state.conversation_history + [
                 {"role": "user", "content": "Based on all my answers, what is your final guess? Just state the guess directly."}
             ],
@@ -241,4 +242,4 @@ def main():
             st.rerun()
 
 if __name__ == "__main__":
-    main()
+    main()