Update app.py

app.py

@@ -4,6 +4,13 @@ from transformers import pipeline
 from bardapi import Bard 
 import os 
 from getvalues import getValues
+from pymongo import MongoClient
+
+uri = os.environ.get("MONGO_CONNECTION_STRING")
+conn = MongoClient(uri, tlsCertificateFileKey="database/cert.pem")
+
+db = conn["myapp"]
+col = db["reminders"]
 
 bardkey = os.environ.get("BARD_API_KEY")
 
@@ -22,6 +29,7 @@ def chatbot():
             values = getValues(query)
             with st.chat_message("assistant"):
                 st.write(values)
+                col.insert_one(values)
         elif ans["labels"][0] == "General Conversation":
             umsg = bard.get_answer(message)["content"]
             with st.chat_message("assistant"):