Update app.py

app.py

@@ -1,5 +1,7 @@
 import streamlit as st
 import time
+import requests
+import json
 from streamlit.components.v1 import html
 
 # Custom CSS for professional look
@@ -99,7 +101,37 @@ def show_confetti():
     </script>
     """)
 
-# Game logic with fake data
+# Mistral AI API call
+def ask_mistral(conversation_history, category):
+    api_url = "https://api.mistral.ai/v1/chat/completions"
+    headers = {
+        "Authorization": "Bearer gz6lDXokxgR6cLY72oomALWcm7vhjzQ",
+        "Content-Type": "application/json"
+    }
+    
+    messages = [
+        {
+            "role": "system",
+            "content": f"You're playing 20 questions to guess a {category}. Ask strategic yes/no questions."
+        }
+    ] + conversation_history
+    
+    data = {
+        "model": "mistral-tiny",
+        "messages": messages,
+        "temperature": 0.7,
+        "max_tokens": 100
+    }
+    
+    try:
+        response = requests.post(api_url, headers=headers, json=data)
+        response.raise_for_status()
+        return response.json()["choices"][0]["message"]["content"]
+    except Exception as e:
+        st.error(f"Error calling Mistral API: {str(e)}")
+        return "Could not generate question"
+
+# Game logic
 def main():
     inject_custom_css()
     
@@ -108,28 +140,18 @@ def main():
     
     if 'game_state' not in st.session_state:
         st.session_state.game_state = "start"
-        st.session_state.questions = [
-            "Is it something you can hold in your hand?",
-            "Is it commonly found in households?",
-            "Can it be used as a tool?",
-            "Is it primarily made of metal?",
-            "Does it have moving parts?",
-            "Is it electronic?",
-            "Is it something you would gift to someone?",
-            "Can it be found in an office?",
-            "Does it require electricity to function?",
-            "Is it typically under $100?"
-        ]
+        st.session_state.questions = []
         st.session_state.current_q = 0
         st.session_state.answers = []
-        st.session_state.target = "smartphone"  # Fake target for testing
+        st.session_state.conversation_history = []
+        st.session_state.category = None
     
     # Start screen
     if st.session_state.game_state == "start":
         st.markdown("""
         <div class="question-box">
             <h3>Welcome to <span style='color:#6C63FF;'>KASOTI 🎯</span></h3>
-            <p>This is a fun guessing game! You’ll think of something, and I’ll ask maximum 20 Yes/No questions to guess what it is.</p>
+            <p>This is a fun guessing game! You'll think of something, and I'll ask maximum 20 Yes/No questions to guess what it is.</p>
             <p>You can choose from three categories:</p>
             <ul>
                 <li><strong>person</strong> – like a celebrity, fictional character, etc.</li>
@@ -145,40 +167,73 @@ def main():
 
             if st.form_submit_button("Start Game"):
                 if not category_input:
-                    st.error("Please fill out both fields!")
+                    st.error("Please enter a category!")
                 elif category_input not in ["person", "place", "object"]:
-                    st.error("Oops! Type either person, place or object to start the game!")
+                    st.error("Please enter either 'person', 'place', or 'object'!")
                 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)
+                    st.session_state.questions = [first_question]
+                    st.session_state.conversation_history = [
+                        {"role": "assistant", "content": first_question}
+                    ]
                     st.session_state.game_state = "gameplay"
                     st.rerun()
 
     # Gameplay screen
     elif st.session_state.game_state == "gameplay":
-        st.markdown(f'<div class="question-box">Question {st.session_state.current_q + 1}/10:<br><br>'
-                    f'<strong>{st.session_state.questions[st.session_state.current_q]}</strong></div>', 
+        current_question = st.session_state.questions[st.session_state.current_q]
+        st.markdown(f'<div class="question-box">Question {st.session_state.current_q + 1}/20:<br><br>'
+                    f'<strong>{current_question}</strong></div>', 
                     unsafe_allow_html=True)
         
         with st.form("answer_form"):
-            answer_input = st.text_input("Type your answer (yes / no):").strip().lower()
+            answer_input = st.text_input("Your answer (yes/no):").strip().lower()
     
-            if st.form_submit_button("Submit Answer"):
+            if st.form_submit_button("Submit"):
                 if answer_input not in ["yes", "no"]:
-                    st.error("Please type only 'yes' or 'no' (case doesn't matter)!")
+                    st.error("Please answer with 'yes' or 'no'!")
                 else:
-                    st.session_state.answers.append(answer_input == "yes")
-                    st.session_state.current_q += 1
-            
-                    if st.session_state.current_q >= len(st.session_state.questions):
+                    # Record answer
+                    st.session_state.answers.append(answer_input)
+                    st.session_state.conversation_history.append(
+                        {"role": "user", "content": answer_input}
+                    )
+                    
+                    # Check if we've reached max questions
+                    if st.session_state.current_q >= 19:  # 0-based index
                         st.session_state.game_state = "result"
+                    else:
+                        # Generate next question
+                        next_question = ask_mistral(
+                            st.session_state.conversation_history,
+                            st.session_state.category
+                        )
+                        st.session_state.questions.append(next_question)
+                        st.session_state.conversation_history.append(
+                            {"role": "assistant", "content": next_question}
+                        )
+                        st.session_state.current_q += 1
+                    
                     st.rerun()
 
     # Result screen
     elif st.session_state.game_state == "result":
+        # Generate final guess
+        final_guess = ask_mistral(
+            st.session_state.conversation_history + [
+                {"role": "user", "content": "Based on all my answers, what is your final guess? Just state the guess directly."}
+            ],
+            st.session_state.category
+        )
+        
         show_confetti()
-        st.markdown('<div class="final-reveal">🎉 I think it\'s a...</div>', unsafe_allow_html=True)
+        st.markdown('<div class="final-reveal">🎉 I think it\'s...</div>', unsafe_allow_html=True)
         time.sleep(1)
-        st.markdown(f'<div class="final-reveal" style="font-size:3.5rem;color:#6C63FF;">{f"something in the category of {st.session_state.category.upper()}"}</div>', 
+        st.markdown(f'<div class="final-reveal" style="font-size:3.5rem;color:#6C63FF;">{final_guess}</div>', 
                     unsafe_allow_html=True)
         
         if st.button("Play Again", key="play_again"):
@@ -186,5 +241,4 @@ def main():
             st.rerun()
 
 if __name__ == "__main__":
-    main()
-    
+    main()