Update app.py

app.py

@@ -49,6 +49,19 @@ if user_input:
     response_text = response.text
     
     chat_history.append({"role": "model", "parts": [{"text": response_text}]})
+
+      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()
     
   st.session_state["chat_history"] = chat_history