Update app.py

app.py

@@ -152,12 +152,15 @@ def main():
         # output container
         output_container = st.empty()
         if submit_clicked:
-            output_container = output_container.container()
-            output_container.chat_message("user").write(user_input)
-            with st.chat_message("assistant"):
-                st_callback = StreamlitCallbackHandler(st.container())
-                response = agent.run(user_input, callbacks=[st_callback])
-                st.write(response)
+            st_callback = StreamlitCallbackHandler(st.container())
+            response = agent_run(user_input),callbacks = [st_callback])
+            st.write(response)
+            # output_container = output_container.container()
+            # output_container.chat_message("user").write(user_input)
+            # with st.chat_message("assistant"):
+            #     st_callback = StreamlitCallbackHandler(st.container())
+            #     response = agent.run(user_input, callbacks=[st_callback])
+            #     st.write(response)
 
             # answer_container = output_container.chat_message("assistant", avatar="🦜")
             # st_callback = StreamlitCallbackHandler(answer_container,)