Update app.py

app.py

@@ -102,21 +102,21 @@ def show_confetti():
 
 # Groq Llama AI API call
 def ask_llama(conversation_history, category):
-    api_url = "https://api.groq.com/openai/v1/chat/completions"  # Updated API URL
+    api_url = "https://api.groq.com/openai/v1/chat/completions"
     headers = {
-        "Authorization": "Bearer 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",  # Your new API key
+        "Authorization": "Bearer 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",
         "Content-Type": "application/json"
     }
     
     messages = [
         {
             "role": "system",
-            "content": f"You're playing 20 questions to guess a {category}. Ask strategic yes/no questions."
+            "content": f"You're playing 20 questions to guess a {category}. Ask strategic yes/no questions one at a time based on previous answers."
         }
     ] + conversation_history
     
     data = {
-        "model": "llama-3.3-70b-versatile",  # The model you are using
+        "model": "llama-3.3-70b-versatile",
         "messages": messages,
         "temperature": 0.7,
         "max_tokens": 100
@@ -144,6 +144,7 @@ def main():
         st.session_state.answers = []
         st.session_state.conversation_history = []
         st.session_state.category = None
+        st.session_state.waiting_for_answer = False
     
     # Start screen
     if st.session_state.game_state == "start":
@@ -181,15 +182,18 @@ def main():
                         {"role": "assistant", "content": first_question}
                     ]
                     st.session_state.game_state = "gameplay"
+                    st.session_state.waiting_for_answer = True
                     st.rerun()
 
     # Gameplay screen
     elif st.session_state.game_state == "gameplay":
+        # Show current question
         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)
         
+        # Answer form
         with st.form("answer_form"):
             answer_input = st.text_input("Your answer (yes/no):").strip().lower()
 
@@ -202,12 +206,12 @@ def main():
                     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
+                    if st.session_state.current_q >= 19:
                         st.session_state.game_state = "result"
                     else:
-                        # Generate next question only after answer is submitted
+                        # Generate next question based on all previous Q&A
                         next_question = ask_llama(
                             st.session_state.conversation_history,
                             st.session_state.category
@@ -217,8 +221,14 @@ def main():
                             {"role": "assistant", "content": next_question}
                         )
                         st.session_state.current_q += 1
+                    
                     st.rerun()
 
+        # Early guess button
+        if st.button("I think I can guess now!"):
+            st.session_state.game_state = "result"
+            st.rerun()
+
     # Result screen
     elif st.session_state.game_state == "result":
         # Generate final guess
@@ -235,9 +245,12 @@ def main():
         st.markdown(f'<div class="final-reveal" style="font-size:3.5rem;color:#6C63FF;">{final_guess}</div>', 
                     unsafe_allow_html=True)
         
+        st.markdown("<br><br>", unsafe_allow_html=True)
+        
+        # Play again button
         if st.button("Play Again", key="play_again"):
             st.session_state.clear()
             st.rerun()
 
 if __name__ == "__main__":
-    main()
+    main()