Update app.py

app.py

@@ -58,6 +58,17 @@ if user_input:
     r, t = message["role"], message["parts"][0]["text"]
     st.markdown(f"**{r.title()}:** {t}")
 
-# Buttons to display, save, and reset chat   
+if st.button("Display History"):
+    c.execute("SELECT * FROM history")
+    rows = c.fetchall()
+
+    for row in rows:
+        st.markdown(f"**{row[0].title()}:** {row[1]}")
+        
+# Save chat history to database
+for message in chat_history:
+    c.execute("INSERT INTO history VALUES (?, ?)", 
+            (message["role"], message["parts"][0]["text"])) 
+    conn.commit() 
 
 conn.close()