Update app.py

app.py

@@ -220,3 +220,17 @@ def process_input():
 
         # Reset the user_input in session state
         st.session_state['user_input'] = ''
+
+# Display the conversation history
+for message in st.session_state.history:
+    if message['role'] == 'user':
+        st.markdown(f"**User:** {message['content']}")
+    elif message['role'] == 'assistant':
+        if isinstance(message['content'], pd.DataFrame):
+            st.markdown("**Assistant:** Query Results:")
+            st.dataframe(message['content'])
+        else:
+            st.markdown(f"**Assistant:** {message['content']}")
+
+# Place the input field at the bottom with the callback
+st.text_input("Enter your message:", key='user_input', on_change=process_input)