Update app.py

app.py

@@ -38,35 +38,30 @@ else:
 for message in chat_history:
   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}]})  
-  
-  if role == "user":  
-    response = model.generate_content(chat_history)
-    response_text = response.text
-    
-    chat_history.append({"role": "model", "parts": [{"text": response_text}]})
+  # Existing code
 
-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
-      
+  
+  # Replace display code
+  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"):
+    # Non-duplicate display code
+
+  # Replace insert code
   for message in chat_history:
-    r, t = message["role"], message["parts"][0]["text"]
-    st.markdown(f"**{r.title()}:** {t}")
+    if message not in rows:
+      c.execute("INSERT INTO history VALUES (?, ?)", 
+                (message["role"], message["parts"][0]["text"]))
+      conn.commit() 
 
-# Save and display chat history
+conn.close()