Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-import google.generativeai as genai
+import google.generativeai as genai 
 import sqlite3
 
 conn = sqlite3.connect('chat_history.db')
@@ -10,71 +10,55 @@ c.execute('''
           (role TEXT, message TEXT)  
           ''')
           
-api_key = "AIzaSyC70u1sN87IkoxOoIj4XCAPw97ae2LZwNM" 
+api_key = "AIzaSyC70u1sN87IkoxOoIj4XCAPw97ae2LZwNM"
 genai.configure(api_key=api_key)
 
 generation_config = {
-  "temperature": 0.9,
-  "max_output_tokens": 2000  
+  "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
+  generation_config=generation_config,
+  safety_settings=safety_settings  
 )
+   
+st.title("Chatbot")
+   
+chat_history = st.session_state.get("chat_history", [])  
 
-st.title("Chatbot")  
-
-chat_history = st.session_state.get("chat_history", [])
-
-displayed_msgs = []
+if len(chat_history) % 2 == 0:
+  role = "user"
+else:
+  role = "model"
+  
 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}]})
+user_input = st.text_input("")
   
-  if role == "user":
+if user_input:
+  chat_history.append({"role": role, "parts": [{"text": user_input}]})  
+  
+  if role == "user":  
     response = model.generate_content(chat_history)
-    response_text = response.text  
+    response_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()
   
-  displayed_msgs = []
+  for row in rows:
+    st.markdown(f"**{row[0].title()}:** {row[1]}")
   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()
+    r, t = message["role"], message["parts"][0]["text"]
+    st.markdown(f"**{r.title()}:** {t}")
 
-    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()
+# Save and display chat history