Adding more features

app.py

@@ -1,6 +1,19 @@
 import os
 import streamlit as st
 import google.generativeai as genai
+import sqlite3
+
+# Connect to SQLite database
+conn = sqlite3.connect('chat_history.db')
+c = conn.cursor()
+
+# Create table if it doesn't exist
+c.execute('''
+    CREATE TABLE IF NOT EXISTS history (
+        role TEXT,
+        message TEXT
+    )
+''')
 
 # API key
 api_key = "AIzaSyC70u1sN87IkoxOoIj4XCAPw97ae2LZwNM"
@@ -14,16 +27,18 @@ st.title("Gemini API Chatbot")
 # Get chat history from session state
 chat_history = st.session_state.get("chat_history", [])
 
+# Display previous messages
+for message in chat_history:
+    role, text = message
+    st.markdown(f"**{role}:** {text}")
+
 # Get user input from text box
 user_input = st.text_input("You")
 
 # Check if user input is not empty
 if user_input:
   # Add user message to chat history
-  chat_history.append(user_input)
-
-  # Display user message with markdown
-  st.markdown(f"**You:** {user_input}")
+  chat_history.append(("You", user_input))
 
   # Create model object
   model = genai.GenerativeModel(model_name="gemini-pro")
@@ -36,10 +51,31 @@ if user_input:
   response_text = response.text
 
   # Add response message to chat history
-  chat_history.append(response_text)
-
-  # Display response message with markdown
-  st.markdown(f"**Gemini Bot:** {response_text}")
+  chat_history.append(("Gemini Bot", response_text))
 
   # Update session state with chat history
   st.session_state["chat_history"] = chat_history
+
+# Add a button to reset chat
+if st.button("Reset Chat"):
+    st.session_state["chat_history"] = []
+
+# Add a button to display chat history from database
+if st.button("Display History"):
+    c.execute("SELECT * FROM history")
+    rows = c.fetchall()
+    for row in rows:
+        st.markdown(f"**{row[0]}:** {row[1]}")
+
+# Add a button to clear chat history from database
+if st.button("Clear History"):
+    c.execute("DELETE FROM history")
+    conn.commit()
+
+# Save chat history to database
+for message in chat_history:
+    c.execute("INSERT INTO history VALUES (?, ?)", message)
+conn.commit()
+
+# Close the connection
+conn.close()