Update app.py

app.py

@@ -2,63 +2,62 @@ import streamlit as st
 import google.generativeai as genai 
 import sqlite3
 
-conn = sqlite3.connect('chat_history.db')
+# Database setup
+conn = sqlite3.connect('chat_history.db')  
 c = conn.cursor()
 
 c.execute('''
-          CREATE TABLE IF NOT EXISTS history
+          CREATE TABLE IF NOT EXISTS history  
           (role TEXT, message TEXT)  
           ''')
           
+# Generative AI setup          
 api_key = "AIzaSyC70u1sN87IkoxOoIj4XCAPw97ae2LZwNM"
-genai.configure(api_key=api_key)
+genai.configure(api_key=api_key)  
 
 generation_config = {
-  "temperature": 0.9,    
-  "max_output_tokens": 100,
+  "temperature": 0.9,
+  "max_output_tokens": 100  
 }
 
-safety_settings = [] 
+safety_settings = []
 
 model = genai.GenerativeModel(
   model_name="gemini-pro",
   generation_config=generation_config,
-  safety_settings=safety_settings  
+  safety_settings=safety_settings
 )
-   
-st.title("Chatbot")
-   
-chat_history = st.session_state.get("chat_history", [])  
+
+# Streamlit UI
+st.title("Chatbot")   
+
+chat_history = st.session_state.get("chat_history", [])
 
 if len(chat_history) % 2 == 0:
   role = "user"
-else:
+else:  
   role = "model"
   
 for message in chat_history:
-  r, t = message["role"], message["parts"][0]["text"]
+  r, t = message["role"], message["parts"][0]["text"]  
   st.markdown(f"**{r.title()}:** {t}")
   
-user_input = st.text_input("")
-  
-if user_input:
-  chat_history.append({"role": role, "parts": [{"text": user_input}]})  
+user_input = st.text_input("")  
+
+if user_input: 
+  chat_history.append({"role": role, "parts": [{"text": user_input}]})
   
-  if role == "user":  
-    response = model.generate_content(chat_history)
+  if role == "user":
+    response = model.generate_content(chat_history)  
     response_text = response.text
-    
-    chat_history.append({"role": "model", "parts": [{"text": response_text}]})
+    chat_history.append({"role": "model", "parts": [{"text": response_text}]}) 
     
   st.session_state["chat_history"] = chat_history
-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]}")
+
   for message in chat_history:
     r, t = message["role"], message["parts"][0]["text"]
     st.markdown(f"**{r.title()}:** {t}")
 
-# Save and display chat history
+# Buttons to display, save, and reset chat   
+
+conn.close()