Update app.py

app.py

@@ -193,7 +193,7 @@ def main():
         
         with st.form("answer_form"):
             answer_input = st.text_input("Your answer (yes/no):").strip().lower()
-    
+
             if st.form_submit_button("Submit"):
                 if answer_input not in ["yes", "no"]:
                     st.error("Please answer with 'yes' or 'no'!")
@@ -203,12 +203,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
                         st.session_state.game_state = "result"
                     else:
-                        # Generate next question
+                        # Generate next question only after answer is submitted
                         next_question = ask_llama(
                             st.session_state.conversation_history,
                             st.session_state.category
@@ -218,7 +218,6 @@ def main():
                             {"role": "assistant", "content": next_question}
                         )
                         st.session_state.current_q += 1
-                    
                     st.rerun()
 
     # Result screen