test2

streamlit_agent/mrkl_demo.py

@@ -97,11 +97,16 @@ with st.form(key="form"):
 
 output_container = st.empty()
 if with_clear_container(submit_clicked):
-    output_container = output_container.container()
-    output_container.chat_message("user").write(user_input)
-
-    answer_container = output_container.chat_message("assistant", avatar="🦜")
-    st_callback = StreamlitCallbackHandler(answer_container)
+    with output_container.container():
+        output_container.text("user")
+        st.write(user_input)
+    # output_container = output_container.container()
+    # output_container.chat_message("user").write(user_input)
+    with output_container.container():
+        output_container.text("assistant")
+        # st.write(answer_container)
+        answer_container = output_container.text("assistant")
+        st_callback = StreamlitCallbackHandler(answer_container)
 
     # If we've saved this question, play it back instead of actually running LangChain
     # (so that we don't exhaust our API calls unnecessarily)