Update app.py

app.py

@@ -16,7 +16,15 @@ import os
 def on_button_click(query):
     st.session_state['button_clicked'] = query
 
-import streamlit as st
+
+
+def display_chat_history(chat_history):
+    for sender, msg, _ in chat_history:
+        background_color = "#FFA07A" if sender == "User" else "#caf"
+        st.markdown(f"<div style='background-color: {background_color}; padding: 10px; border-radius: 10px; margin: 10px;'>{sender}: {msg}</div>", unsafe_allow_html=True)
+
+
+
 
 def cloud_button(label, query, key=None, color=None, overlap=30):
     button_id = f"cloud-button-{key or label}"