Update app.py

app.py

@@ -317,6 +317,34 @@ def main():
                                        placeholder="Type your answer here...").strip().lower()
             if st.form_submit_button("Submit Answer", use_container_width=True):
                 # Original answer handling logic remains unchanged
+                if answer_input not in ["yes", "no", "both"]:
+                    st.error("Please answer with 'yes', 'no', or 'both'!")
+                else:
+                    st.session_state.answers.append(answer_input)
+                    st.session_state.conversation_history.append(
+                        {"role": "user", "content": answer_input}
+                    )
+
+                    # Generate next response
+                    next_response = ask_llama(
+                        st.session_state.conversation_history,
+                        st.session_state.category
+                    )
+
+                    if "Final Guess:" in next_response:
+                        st.session_state.final_guess = next_response.split("Final Guess:")[1].strip()
+                        st.session_state.game_state = "confirm_guess"
+                    else:
+                        st.session_state.questions.append(next_response)
+                        st.session_state.conversation_history.append(
+                            {"role": "assistant", "content": next_response}
+                        )
+                        st.session_state.current_q += 1
+
+                        if st.session_state.current_q >= 20:
+                            st.session_state.game_state = "result"
+
+                    st.experimental_rerun()
 
     # Rest of the original game logic remains unchanged (confirm_guess, result states)
 
@@ -325,13 +353,16 @@ def main():
         st.markdown('<div class="help-chat">', unsafe_allow_html=True)
         help_query = st.text_input("Type your question:", key="help_query", 
                                  placeholder="How should I answer this?")
-        if st.button("Send", use_container_width=True, key="send_help"):
-            # Original help logic remains unchanged
-        if st.session_state.help_conversation:
-            for msg in st.session_state.help_conversation:
-                st.markdown(f'<div class="user-message">{msg["query"]}</div>', unsafe_allow_html=True)
-                st.markdown(f'<div class="bot-message">{msg["response"]}</div>', unsafe_allow_html=True)
-        st.markdown('</div>', unsafe_allow_html=True)
-
+            if st.button("Send", use_container_width=True, key="send_help"):
+                if help_query:
+                    help_response = ask_help_agent(help_query)
+                    st.session_state.help_conversation.append({"query": help_query, "response": help_response})
+                else:
+                    st.error("Please enter a query!")
+            if st.session_state.help_conversation:
+                for msg in st.session_state.help_conversation:
+                    st.markdown(f'<div class="user-message">{msg["query"]}</div>', unsafe_allow_html=True)
+                    st.markdown(f'<div class="bot-message">{msg["response"]}</div>', unsafe_allow_html=True)
+            st.markdown('</div>', unsafe_allow_html=True)
 if __name__ == "__main__":
     main()