Update app.py

app.py

@@ -116,7 +116,11 @@ def main():
     if "chat_history" not in st.session_state:
         st.session_state['chat_history'] = []
 
-    display_chat_history(st.session_state['chat_history'])
+def display_chat_history(chat_history):
+    for chat in chat_history:
+        background_color = "#FFA07A" if chat[2] == "new" else "#acf" if chat[0] == "User" else "#caf"
+        st.markdown(f"<div style='background-color: {background_color}; padding: 10px; border-radius: 10px; margin: 10px;'>{chat[0]}: {chat[1]}</div>", unsafe_allow_html=True)
+
 
     st.write("<!-- Start Spacer -->", unsafe_allow_html=True)
     st.write("<div style='flex: 1;'></div>", unsafe_allow_html=True)
@@ -166,6 +170,9 @@ def main():
 
             loading_message.empty()
 
+            display_chat_history(st.session_state['chat_history'])
+
+
             # Clear the input field by setting the query variable to an empty string
             query = ""