app.py

import streamlit as st
import google.generativeai as genai
import sqlite3

conn = sqlite3.connect('chat_history.db')
c = conn.cursor()

c.execute('''
          CREATE TABLE IF NOT EXISTS history
          (role TEXT, message TEXT)  
          ''')
          
api_key = "AIzaSyC70u1sN87IkoxOoIj4XCAPw97ae2LZwNM" 
genai.configure(api_key=api_key)

generation_config = {
  "temperature": 0.9,
  "max_output_tokens": 2000  
}

safety_settings = []

model = genai.GenerativeModel(
  model_name="Gemini-Pro",
  generation_config=generation_config, 
  safety_settings=safety_settings
)

st.title("Chatbot")  

chat_history = st.session_state.get("chat_history", [])

displayed_msgs = []
for message in chat_history:
  r, t = message["role"], message["parts"][0]["text"]
  st.markdown(f"**{r.title()}:** {t}")
  displayed_msgs.append(message)
  
user_input = st.text_input("")  

if user_input: 

  if len(chat_history) % 2 == 0:
    role = "user"
  else:
    role = "model"
    
  chat_history.append({"role": role, "parts": [{"text": user_input}]})
  
  if role == "user":
    response = model.generate_content(chat_history)
    response_text = response.text  
    chat_history.append({"role": "model", "parts": [{"text": response_text}]})

  st.session_state["chat_history"] = chat_history
  
  displayed_msgs = []
  for message in chat_history:
    if message not in displayed_msgs:  
      r, t = message["role"], message["parts"][0]["text"]
      st.markdown(f"**{r.title()}:** {t}")
      displayed_msgs.append(message)  
        
  if st.button("Display History"):    
    displayed_msgs = [] 
    c.execute("SELECT * FROM history")
    rows = c.fetchall()

    for row in rows:
      if {"role": row[0], "parts": [{"text": row[1]}]} not in displayed_msgs:  
        st.markdown(f"**{row[0].title()}:** {row[1]}")
        displayed_msgs.append({"role": row[0], "parts": [{"text": row[1]}]})
              
for message in chat_history:
  if {"role": message["role"], "parts": [message["parts"][0]]} not in rows:   
    c.execute("INSERT INTO history VALUES (?, ?)",
              (message["role"], message["parts"][0]["text"])) 
    conn.commit()
    
conn.close()