added download feature

app.py

@@ -93,6 +93,7 @@ def generate_response(question):
 
     return "".join(preds)
 
+import uuid
 import streamlit as st
 from streamlit_chat import message
 
@@ -125,10 +126,18 @@ container = st.container()
 
 with container:
     with st.form(key='my_form', clear_on_submit=True):
-        user_input = st.text_input("You:", key='input',placeholder="Disclaimer: Be careful with punctuations like , ? . ! \" and IT WILL CUSS YOU")
+        user_input = st.text_input("You:", key='input',placeholder="Disclaimer: Be careful with punctuations like , ? . ! \"")
         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)
+    col1,col2=st.columns(2)
+    with col1:
+        clear_button = st.button("Clear Conversation", key="clear",use_container_width=True)
+    with col2:
+        save_button = st.button("Save Conversation", key="save",use_container_width=True)
+    down_id = st.text_input('Enter ID to download chat',placeholder="Message ID")
+    if down_id:
+        info=client['rohank']['table1']
+        data=info.find_one({'message_id':down_id})
+        down_button = st.download_button('Download chat', "\n".join(data['message']),file_name="sar_chat.txt")
     # reset everything
     if clear_button:
         st.session_state['generated'] = []
@@ -141,12 +150,18 @@ with container:
         # Send a ping to confirm a successful connection
         try:
             client.admin.command('ping')
-            st.write("Pinged your deployment. You successfully connected to MongoDB!")
+            st.success("Pinged your deployment. You successfully connected to MongoDB! Saved Successfully.")
+            info=client['rohank']['table1']
+            chats=list([])
+            for i in range(len(st.session_state['generated'])):
+                chats.append("You: "+st.session_state['past'][i])
+                chats.append("Bot: "+st.session_state['generated'][i])
+            id=uuid.uuid4()
+            info.insert_one({"message_id":str(id),"message":chats})
+            st.success("Copy this id "+str(id)+" for downloading saved chat anytime anywhere and then paste it down below!")
         except Exception as e:
-            st.write(e)
-        info=client['rohank']['table1']
-        info.insert_one({"message":"first time"})
-
+            st.error("Can't connect to MongoDB. Save Failed.")
+        
 
     if submit_button and user_input:
         output = generate_response(user_input)