Update app.py

app.py

@@ -41,7 +41,7 @@ def inject_custom_css():
             padding: 2rem;
             margin: 1.5rem 0;
             box-shadow: 0 4px 6px rgba(0,0,0,0.1);
-            color: black !important;  /* Force black text */
+            color: black !important;
         }
 
         .answer-btn {
@@ -122,8 +122,8 @@ def ask_llama(conversation_history, category, is_final_guess=False):
 1. Ask strategic, non-repeating yes/no questions that narrow down possibilities
 2. Consider all previous answers carefully before asking next question
 3. If you're very confident (80%+ sure), respond with "Final Guess: [your guess]"
-4. For places: ask about continent, climate, famous landmarks, or population
-5. For people: ask about profession, gender, alive/dead, nationality, or fame
+4. For places: ask about continent, climate, famous landmarks, country, city or population
+5. For people: ask about fictional or real, profession, gender, alive/dead, nationality, or fame
 6. For objects: ask about size, color, usage, material, or where it's found
 7. Never repeat questions and always make progress toward guessing"""
 
@@ -279,35 +279,38 @@ def main():
                     st.markdown(f"**You:** {msg['query']}")
                     st.markdown(f"**Help Assistant:** {msg['response']}")
 
-    # Guess confirmation screen
+    # Guess confirmation screen using text input response
     elif st.session_state.game_state == "confirm_guess":
         st.markdown(f'<div class="question-box">🤖 My Final Guess:<br><br>'
                     f'<strong>Is it {st.session_state.final_guess}?</strong></div>',
                     unsafe_allow_html=True)
         
-        col1, col2 = st.columns(2)
-        with col1:
-            if st.button("Yes! You're correct!", key="correct_guess", use_container_width=True):
-                st.session_state.game_state = "result"
-                st.rerun()
-        with col2:
-            if st.button("No, try again", key="wrong_guess", use_container_width=True):
-                # Add negative response to history and continue
-                st.session_state.conversation_history.append(
-                    {"role": "user", "content": "no"}
-                )
-                st.session_state.game_state = "gameplay"
-                # Generate follow-up question
-                next_response = ask_llama(
-                    st.session_state.conversation_history,
-                    st.session_state.category
-                )
-                st.session_state.questions.append(next_response)
-                st.session_state.conversation_history.append(
-                    {"role": "assistant", "content": next_response}
-                )
-                st.session_state.current_q += 1
-                st.rerun()
+        with st.form("confirm_form"):
+            confirm_input = st.text_input("Type your answer (yes/no/both):", key="confirm_input").strip().lower()
+            if st.form_submit_button("Submit"):
+                if confirm_input not in ["yes", "no", "both"]:
+                    st.error("Please answer with 'yes', 'no', or 'both'!")
+                else:
+                    if confirm_input == "yes":
+                        st.session_state.game_state = "result"
+                        st.rerun()
+                        st.stop()  # Immediately halt further execution
+                    else:
+                        # Add negative response to history and continue gameplay
+                        st.session_state.conversation_history.append(
+                            {"role": "user", "content": "no"}
+                        )
+                        st.session_state.game_state = "gameplay"
+                        next_response = ask_llama(
+                            st.session_state.conversation_history,
+                            st.session_state.category
+                        )
+                        st.session_state.questions.append(next_response)
+                        st.session_state.conversation_history.append(
+                            {"role": "assistant", "content": next_response}
+                        )
+                        st.session_state.current_q += 1
+                        st.rerun()
 
     # Result screen
     elif st.session_state.game_state == "result":
@@ -338,4 +341,4 @@ def main():
             st.rerun()
 
 if __name__ == "__main__":
-    main()
+    main()