Update app.py

app.py

@@ -3,7 +3,7 @@ import time
 import requests
 from streamlit.components.v1 import html
 
-# Custom CSS for professional look
+# Inject custom CSS
 def inject_custom_css():
     st.markdown("""
     <style>
@@ -62,6 +62,7 @@ def inject_custom_css():
     </style>
     """, unsafe_allow_html=True)
 
+# Show confetti effect
 def show_confetti():
     html("""
     <canvas id="confetti-canvas" class="confetti"></canvas>
@@ -78,6 +79,7 @@ def show_confetti():
     </script>
     """)
 
+# Call Groq API (LLama model)
 def ask_llama(messages, category, is_final=False):
     api_url = "https://api.groq.com/openai/v1/chat/completions"
     headers = {
@@ -85,148 +87,144 @@ def ask_llama(messages, category, is_final=False):
         "Content-Type": "application/json"
     }
     
-    prompt = (f"You're playing 20 questions to guess a {category}. " +
-              "Ask strategic yes/no questions one at a time." +
-              (" Based on the answers, what is your final guess? State only the guess." if is_final else ""))
-    
+    prompt = ("You're playing 20 questions to guess a " + category + 
+             ". Ask strategic yes/no questions one at a time." +
+             (" Based on the answers, what is your final guess? State only the guess." if is_final else ""))
+
     data = {
-        "model": "llama-3.3-70b-versatile",
+        "model": "llama-3-70b-8192",
         "messages": [{"role": "system", "content": prompt}] + messages,
         "temperature": 0.7,
         "max_tokens": 100
     }
-    
+
     try:
         with st.spinner("Thinking..."):
-            response = requests.post(api_url, headers=headers, json=data, timeout=10)
+            response = requests.post(api_url, headers=headers, json=data, timeout=15)
             response.raise_for_status()
             return response.json()["choices"][0]["message"]["content"]
     except Exception as e:
         st.error(f"API Error: {str(e)}")
         return None
 
+# Main Game Logic
 def main():
     inject_custom_css()
     st.markdown('<div class="title">KASOTI</div>', unsafe_allow_html=True)
     st.markdown('<div class="subtitle">The Ultimate Guessing Game</div>', unsafe_allow_html=True)
 
-    # Initialize game state
+    # Session State
     if 'game' not in st.session_state:
         st.session_state.game = {
             'state': 'start',
             'category': None,
-            'current_question': None,
+            'questions': [],
             'conversation': [],
-            'question_count': 0,
-            'waiting_for_answer': False
+            'input_buffer': ""
         }
 
-    # Start Screen
-    if st.session_state.game['state'] == 'start':
+    game = st.session_state.game
+
+    # Start screen
+    if game['state'] == 'start':
         st.markdown("""
         <div class="question-box">
             <h3>Welcome to <span style='color:#6C63FF;'>KASOTI 🎯</span></h3>
             <p>Think of something and I'll try to guess it with yes/no questions!</p>
             <p>Choose a category:</p>
+            <ul>
+                <li><strong>person</strong> (celebrity, fictional character)</li>
+                <li><strong>place</strong> (city, country, location)</li>
+                <li><strong>object</strong> (something you can touch)</li>
+            </ul>
         </div>
         """, unsafe_allow_html=True)
 
         col1, col2, col3 = st.columns(3)
         with col1:
             if st.button("Person", use_container_width=True):
-                st.session_state.game['category'] = 'person'
-                st.session_state.game['state'] = 'generate_question'
+                game['category'] = 'person'
+                game['state'] = 'playing'
                 st.rerun()
         with col2:
             if st.button("Place", use_container_width=True):
-                st.session_state.game['category'] = 'place'
-                st.session_state.game['state'] = 'generate_question'
+                game['category'] = 'place'
+                game['state'] = 'playing'
                 st.rerun()
         with col3:
             if st.button("Object", use_container_width=True):
-                st.session_state.game['category'] = 'object'
-                st.session_state.game['state'] = 'generate_question'
+                game['category'] = 'object'
+                game['state'] = 'playing'
                 st.rerun()
 
-    # Generate Question State
-    elif st.session_state.game['state'] == 'generate_question':
-        # Generate new question
-        question = ask_llama(
-            st.session_state.game['conversation'],
-            st.session_state.game['category']
-        )
-        
-        if question:
-            st.session_state.game['current_question'] = question
-            st.session_state.game['conversation'].append({"role": "assistant", "content": question})
-            st.session_state.game['question_count'] += 1
-            st.session_state.game['state'] = 'wait_for_answer'
-            st.session_state.game['waiting_for_answer'] = True
-            st.rerun()
-        else:
-            st.error("Failed to generate question. Please try again.")
-            st.session_state.game['state'] = 'start'
-            st.rerun()
+    # Game play screen
+    elif game['state'] == 'playing':
+        if not game['questions']:
+            first_q = ask_llama([{"role": "user", "content": "Ask your first yes/no question."}], game['category'])
+            if first_q:
+                game['questions'].append(first_q)
+                game['conversation'].append({"role": "assistant", "content": first_q})
+                st.rerun()
+            else:
+                st.error("Failed to generate question.")
+                game['state'] = 'start'
+                st.rerun()
+
+        current_q_index = len(game['questions']) - 1
+        current_q = game['questions'][current_q_index]
 
-    # Wait for Answer State
-    elif st.session_state.game['state'] == 'wait_for_answer':
         st.markdown(f"""
         <div class="question-box">
-            Question {st.session_state.game['question_count']}:<br><br>
-            <strong>{st.session_state.game['current_question']}</strong>
+            Question {current_q_index + 1}:<br><br>
+            <strong>{current_q}</strong>
         </div>
         """, unsafe_allow_html=True)
 
         with st.form("answer_form"):
-            answer = st.radio("Your answer:", ["Yes", "No"], index=None, horizontal=True)
+            user_input = st.text_input("Your answer (Yes, No, or more detailed response):")
             submitted = st.form_submit_button("Submit")
-            
-            if submitted and answer:
-                # Record answer
-                answer_lower = answer.lower()
-                st.session_state.game['conversation'].append({"role": "user", "content": answer_lower})
-                st.session_state.game['waiting_for_answer'] = False
-                
-                # Check if we should guess or ask another question
-                if st.session_state.game['question_count'] >= 5:  # Minimum 5 questions before checking
-                    ready = ask_llama(
-                        st.session_state.game['conversation'] + [
-                            {"role": "user", "content": "Can you guess now? Answer only 'yes' or 'no'."}
-                        ],
-                        st.session_state.game['category']
+
+            if submitted and user_input.strip() != "":
+                game['conversation'].append({"role": "user", "content": user_input.strip()})
+
+                # Ask if ready to guess after 5 questions
+                if current_q_index >= 4:
+                    can_guess = ask_llama(
+                        game['conversation'] + [{"role": "user", "content": "Can you guess now? Answer only yes or no."}],
+                        game['category']
                     )
-                    if ready and ready.strip().lower() == 'yes':
-                        st.session_state.game['state'] = 'result'
+                    if can_guess and can_guess.strip().lower() == 'yes':
+                        game['state'] = 'result'
                         st.rerun()
-                
-                # Continue with next question if under 20
-                if st.session_state.game['question_count'] < 20:
-                    st.session_state.game['state'] = 'generate_question'
-                    st.rerun()
+
+                # Generate next question
+                if current_q_index < 19:
+                    next_q = ask_llama(game['conversation'], game['category'])
+                    if next_q:
+                        game['questions'].append(next_q)
+                        game['conversation'].append({"role": "assistant", "content": next_q})
+                        st.rerun()
+                    else:
+                        st.error("Couldn't get next question.")
                 else:
-                    st.session_state.game['state'] = 'result'
+                    game['state'] = 'result'
                     st.rerun()
 
     # Result Screen
-    elif st.session_state.game['state'] == 'result':
-        guess = ask_llama(
-            st.session_state.game['conversation'],
-            st.session_state.game['category'],
-            is_final=True
-        )
-        
-        if guess:
+    elif game['state'] == 'result':
+        final_guess = ask_llama(game['conversation'], game['category'], is_final=True)
+
+        if final_guess:
             show_confetti()
             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;">{guess}</div>', 
-                        unsafe_allow_html=True)
+            st.markdown(f'<div class="final-reveal" style="font-size:3.5rem;color:#6C63FF;">{final_guess}</div>', unsafe_allow_html=True)
         else:
-            st.error("Sorry, I couldn't make a guess. Let's try again!")
-        
+            st.error("Sorry, I couldn't make a guess.")
+
         if st.button("Play Again", type="primary"):
             st.session_state.clear()
             st.rerun()
 
 if __name__ == "__main__":
-    main()
+    main()