Update app.py

app.py

@@ -235,11 +235,11 @@ def load_models_and_documents():
             
             docs_texts, docs_metadatas, dense_embeddings, sparse_embeddings = chunk_documents(texts, metadatas, dense_model, sparse_model)
 
-            with open(texts_path, mode="wb") as outfile_texts:
+            with open(texts_path, "wb") as outfile_texts:
                 packed_texts = msgpack.packb(docs_texts, use_bin_type=True)
                 outfile_texts.write(packed_texts)
 
-            with open(metadatas_path, mode="wb") as outfile_metadatas:
+            with open(metadatas_path, "wb") as outfile_metadatas:
                 packed_metadatas = msgpack.packb(docs_metadatas, use_bin_type=True)
                 outfile_metadatas.write(packed_metadatas)
             
@@ -260,10 +260,10 @@ def load_models_and_documents():
             combined_sparse_matrix = vstack(sparse_matrices)
             save_npz(sparse_path, combined_sparse_matrix)
         else:
-            with open(texts_path, mode="rb") as data_file_texts:
+            with open(texts_path, "rb") as data_file_texts:
                 decompressed_texts = data_file_texts.read()
                 
-            with open(metadatas_path, mode="rb") as data_file_metadatas:
+            with open(metadatas_path, "rb") as data_file_metadatas:
                 decompressed_metadatas = data_file_metadatas.read()
                 
             docs_texts = msgpack.unpackb(decompressed_texts, raw=False)
@@ -414,7 +414,9 @@ if __name__ == '__main__':
     conversations_path = os.path.join(os.getenv('HF_HOME'), 'conversations')
 
     try:
-        conversations: dict = joblib.load(conversations_path)
+        with open(conversations_path, 'rb') as fp:
+            packed_bytes = fp.read()
+        conversations: dict = msgpack.unpackb(packed_bytes, raw=False)
     except:
         conversations = {}
 
@@ -425,7 +427,7 @@ if __name__ == '__main__':
         with st.sidebar:
             st.session_state.chat_id = st.selectbox(
                 label='Choose a conversation',
-                options=[st.session_state.chat_id] + list(conversations.keys()),
+                options=([st.session_state.chat_id] if st.session_state.chat_id != None else []) + list(conversations.keys()),
                 index=0,
                 placeholder='_'
             )
@@ -437,8 +439,11 @@ if __name__ == '__main__':
                     st.session_state.chat_id = 'New Chat'
             with col2:
                 if st.button('Delete Chat', use_container_width=True, disabled=False if st.session_state.chat_id != 'New Chat' else True):
-                    del conversations[st.session_state.chat_id]
-                    joblib.dump(conversations, conversations_path)
+                    conversations.pop(st.session_state.chat_id)
+                    st.session_state.chat_id = None
+                    with open(conversations_path, 'wb') as fp:
+                        packed_bytes = msgpack.packb(conversations, use_bin_type=True)
+                        fp.write(packed_bytes)
 
         if st.session_state.chat_id == 'New Chat':
             st.session_state.messages = []
@@ -463,7 +468,9 @@ if __name__ == '__main__':
                 )
                 st.session_state.chat_id = outputs[0].outputs[0].text
                 conversations.update({st.session_state.chat_id: st.session_state.messages})
-                joblib.dump(conversations, conversations_path)
+                with open(conversations_path, 'wb') as fp:
+                    packed_bytes = msgpack.packb(conversations, use_bin_type=True)
+                    fp.write(packed_bytes)
             st.chat_message("user").markdown(prompt)
             st.session_state.messages.append({"role": "user", "content": prompt})
             print(f'PROMPT: {prompt}')
@@ -479,7 +486,9 @@ if __name__ == '__main__':
                                           
             st.session_state.messages.append({"role": "assistant", "content": full_response})
             conversations.update({st.session_state.chat_id: st.session_state.messages})
-            joblib.dump(conversations, conversations_path)
+            with open(conversations_path, 'wb') as fp:
+                packed_bytes = msgpack.packb(conversations, use_bin_type=True)
+                fp.write(packed_bytes)
 
 
         with st.sidebar: