Update app.py

app.py

@@ -101,7 +101,6 @@ with chat_container:
 # User input at the bottom
 user_input = st.text_input("Type your message here:", key="user_message")
 
-# If user submits a message
 if user_input:
     # Save user message
     save_message(session_id, "user", user_input)
@@ -116,10 +115,10 @@ if user_input:
     st.session_state.messages.append(("assistant", response))
     
     # Display AI response
-    with chat_container:
-        with st.chat_message("assistant"):
-            st.markdown(response)
+    with st.chat_message("assistant"):
+        st.markdown(response)
 
-    #  Clear the input field after message submission
+    # ✅ Clear input field
     st.session_state.pop("user_message")
+    st.session_state["user_message"] = ""  
     st.rerun()  # Refresh the app