feat: provisional solution to streamlit issue st.chat_input collapse after submit #12081

Author: yesidc

pages/ask_uos_chat.py

@@ -236,7 +236,10 @@ def handle_user_input(self):
         user_id = self.get_user_id()
         history = self.get_history(user_id)
 
-        if prompt := st.chat_input(placeholder=session_state["_"]("Message")):
+        if prompt := st.chat_input(
+            placeholder=session_state["_"]("Message"),
+            key=f"chat_input_{st.session_state.input_key_counter}",
+        ):
             if not session_state.feedback_saved:
                 self.log_feedback()
             st.session_state.feedback_saved = False
@@ -254,6 +257,10 @@ def handle_user_input(self):
             if history.messages[-1].type != ROLES[0]:  # "ai"
                 self.generate_response(prompt)
 
+            st.session_state.input_key_counter += 1
+
+            st.rerun()  # Rerun to update the chat messages and input field
+
     def get_agent(self):
         if st.session_state["agent"] is None:
             st.session_state["agent"] = CampusManagementOpenAIToolsAgent.run(

pages/utils.py

@@ -24,6 +24,7 @@ def initialize_session_sate() -> None:
         "agent_language": settings.language,
         "agent": None,
         "ask_uos_user_id": None,
+        "input_key_counter": 0,
     }
 
     for key, value in defaults.items():