mongodb testing

app.py

@@ -96,6 +96,15 @@ def generate_response(question):
 import streamlit as st
 from streamlit_chat import message
 
+
+from pymongo.mongo_client import MongoClient
+from pymongo.server_api import ServerApi
+
+uri = "mongodb+srv://rohank587:zaqxswcdevfr@rkcluster.e3fpzja.mongodb.net/?retryWrites=true&w=majority"
+
+# Create a new client and connect to the server
+client = MongoClient(uri, server_api=ServerApi('1'))
+
 st.title(":red[_Sarcastic_] Chatbot")
 
 if 'generated' not in st.session_state:
@@ -119,6 +128,7 @@ with container:
         user_input = st.text_input("You:", key='input',placeholder="Disclaimer: Be careful with punctuations like , ? . ! \" and IT WILL CUSS YOU")
         submit_button = st.form_submit_button(label='Send',use_container_width=True)
     clear_button = st.button("Clear Conversation", key="clear",use_container_width=True)
+    save_button = st.button("Save Conversation", key="save",use_container_width=True)
     # reset everything
     if clear_button:
         st.session_state['generated'] = []
@@ -127,6 +137,17 @@ with container:
             {"role": "system", "content": "You are a helpful assistant."}
         ]
 
+    if save_button:
+        # Send a ping to confirm a successful connection
+        try:
+            client.admin.command('ping')
+            st.write("Pinged your deployment. You successfully connected to MongoDB!")
+        except Exception as e:
+            st.write(e)
+        info=client['rohank']['table1']
+        info.insert_one({"message":"first time"})
+
+
     if submit_button and user_input:
         output = generate_response(user_input)
         st.session_state['past'].append(user_input)