Update app.py

app.py

@@ -28,36 +28,25 @@ def cs_body():
     with chat:
         st.title('ChatOps')
 
-        # Display the messages in the session state
-        #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"])
-
         # Chat input
         if prompt := st.chat_input("How Can I Help?"):
-            # st.session_state.messages.append({"role": "user", "content": prompt, "type": "text"})
             with chat.chat_message("user"):
                 st.markdown(prompt)
-        # Assistant working
+        # Assistant Response
             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"})
-
+        # Chat History
             for message in reversed(st.session_state.messages):
                 with st.chat_message(message["role"]):
-                    # if type is text
+                    # Text Response
                     if message["type"] == "text":
                         st.markdown(message["content"])
-                    # if type is dataframe
+                    # Data Response
                     if message["type"] == "dataframe":
                         st.dataframe(message["content"])
-            
+
+            # Add input and response to chat history
             st.session_state.messages.append({"role": "assistant", "content": "I am working..", "type": "text"})
             st.session_state.messages.append({"role": "user", "content": prompt, "type": "text"})