Update app.py

app.py

@@ -127,7 +127,8 @@ def ask_llama(conversation_history, category, is_final_guess=False):
 6. For objects: ask about size, color, usage, material, or where it's found.
 7. Never repeat questions and always make progress toward guessing.
 8. If the guess is generic (e.g. "bed") and the player confirms it with "yes", end the game.
-9. If the guess is generic but the answer is "no", ask additional questions to refine the guess."""
+9. If the guess is generic but the answer is "no", ask additional questions to refine the guess.
+10. After a rejected final guess, continue asking questions instead of making new guesses immediately."""
 
     if is_final_guess:
         prompt = f"""Based on these answers about a {category}, provide ONLY your final guess with no extra text:
@@ -262,7 +263,7 @@ def main():
                         st.session_state.current_q += 1
 
                         # Stop after 20 questions max
-                        if st.session_state.current_q >= 20:
+                        if st.session_state.current_q >= 19:
                             st.session_state.game_state = "result"
                     
                     st.rerun()
@@ -311,6 +312,8 @@ def main():
                             {"role": "assistant", "content": next_response}
                         )
                         st.session_state.current_q += 1
+                        if st.session_state.current_q >= 19:
+                            st.session_state.game_state = "result"
                         st.rerun()
 
     # Result screen
@@ -342,4 +345,4 @@ def main():
             st.rerun()
 
 if __name__ == "__main__":
-    main()
+    main()