Update app.py

app.py

@@ -477,40 +477,41 @@ def main():
             </div>
             """, unsafe_allow_html=True)
             
-            # Chat messages container
-            chat_container = st.container()
-            
-            # Input at bottom
-            with st.form("help_chat_form"):
-                cols = st.columns([4, 1])
-                help_query = cols[0].text_input("Type your question here:", key="help_query", 
-                                              placeholder="How to play? What are the rules?")
-                submitted = cols[1].form_submit_button("Send →")
+            # Display chat messages first
+            if st.session_state.help_conversation:
+                for msg in st.session_state.help_conversation:
+                    # User message
+                    st.markdown(f"""
+                    <div class="chat-message user-message">
+                        <div class="message-content">{msg['query']}</div>
+                        <div class="message-timestamp">You</div>
+                    </div>
+                    """, unsafe_allow_html=True)
+                    
+                    # Assistant message
+                    st.markdown(f"""
+                    <div class="chat-message assistant-message">
+                        <div class="message-content" style="color:#1e293b;">{msg['response']}</div>
+                        <div class="message-timestamp">Assistant</div>
+                    </div>
+                    """, unsafe_allow_html=True)
+
+            # Chat input form at bottom (separate form to ensure it works)
+            with st.form("help_chat_form", clear_on_submit=True):
+                help_query = st.text_input("Type your question here:", 
+                                        key="help_query", 
+                                        placeholder="How to play? What are the rules?",
+                                        label_visibility="collapsed")
+                submitted = st.form_submit_button("Send →")
                 
-            if submitted and help_query:
-                help_response = ask_help_agent(help_query)
-                st.session_state.help_conversation.append({"query": help_query, "response": help_response})
-                st.experimental_rerun()
-
-            # Display chat messages
-            with chat_container:
-                if st.session_state.help_conversation:
-                    for msg in st.session_state.help_conversation:
-                        # User message
-                        st.markdown(f"""
-                        <div class="chat-message user-message">
-                            <div class="message-content">{msg['query']}</div>
-                            <div class="message-timestamp">You</div>
-                        </div>
-                        """, unsafe_allow_html=True)
-                        
-                        # Assistant message
-                        st.markdown(f"""
-                        <div class="chat-message assistant-message">
-                            <div class="message-content" style="color:#1e293b;">{msg['response']}</div>
-                            <div class="message-timestamp">Assistant</div>
-                        </div>
-                        """, unsafe_allow_html=True)
+                if submitted and help_query:
+                    with st.spinner("Assistant is thinking..."):
+                        help_response = ask_help_agent(help_query)
+                        st.session_state.help_conversation.append({
+                            "query": help_query, 
+                            "response": help_response
+                        })
+                    st.experimental_rerun()
     # Guess confirmation screen using text input response
     elif st.session_state.game_state == "confirm_guess":
         st.markdown(f'''