Update app.py

app.py

@@ -40,25 +40,26 @@ def cs_body():
 
         # Chat input
         if prompt := st.chat_input("How Can I Help?"):
-            st.session_state.messages.append({"role": "user", "content": prompt, "type": "text"})
+            # st.session_state.messages.append({"role": "user", "content": prompt, "type": "text"})
             with chat.chat_message("user"):
                 st.markdown(prompt)
         # Assistant working
             with st.chat_message("assistant"):
                 message_placeholder = st.empty()
                 message_placeholder.write("I am working..")
-                st.session_state.messages.append({"role": "assistant", "content": "I am working..", "type": "text"})
-
-        for message in st.session_state.messages:
-            with st.chat_message(message["role"]):
-                # if type is text
-                if message["type"] == "text":
-                    st.markdown(message["content"])
-                # if type is dataframe
-                if message["type"] == "dataframe":
-                    st.dataframe(message["content"])
+            #    st.session_state.messages.append({"role": "assistant", "content": "I am working..", "type": "text"})
 
+            for message in st.session_state.messages:
+                with st.chat_message(message["role"]):
+                    # if type is text
+                    if message["type"] == "text":
+                        st.markdown(message["content"])
+                    # if type is dataframe
+                    if message["type"] == "dataframe":
+                        st.dataframe(message["content"])
 
+            st.session_state.messages.append({"role": "assistant", "content": "I am working..", "type": "text"})
+            st.session_state.messages.append({"role": "user", "content": prompt, "type": "text"})
 
     env.subheader('Enviornment:')
     return None