Update app.py

app.py

@@ -497,21 +497,18 @@ if __name__ == '__main__':
         },
     }
 
-    if 'selected' not in st.session_state:
-        selected = 'ChatBot'
-    else:
-        selected = st.session_state.selected
+    if 'menu_id' not in st.session_state:
+        st.session_state.menu_id = 'ChatBot'
     
-    menu_id = st_navbar(
+    st.session_state.menu_id = st_navbar(
         ['ChatBot', 'Documents'], 
-        selected,
+        st.session_state.menu_id,
         options={
             'hide_nav': False,
             'fix_shadow': False,
             'use_padding': False
         },
-        styles=styles,
-        key='selected'
+        styles=styles
     )
 
     st.title('Multipurpose AI Agent')
@@ -535,7 +532,7 @@ if __name__ == '__main__':
             
             for deleted_idx in st.session_state.key_data_editor['deleted_rows']:
                 name = st.session_state.df.loc[deleted_idx, 'document']
-                print(f'{name}')
+                print(f'WHAT IS THAT : {name}')
                 os.remove(os.path.join(embeddings_path, name + '_ids.npy'))
                 ids_values = retrieve_ids_value(conn, cursor, name)
                 
@@ -548,7 +545,7 @@ if __name__ == '__main__':
             st.toast('Deletion Completed !', icon='🎉')
             
     
-    if menu_id == 'Documents':       
+    if st.session_state.menu_id == 'Documents':       
         st.session_state.df = st.data_editor(
             st.session_state.df, 
             num_rows="dynamic",
@@ -604,15 +601,15 @@ if __name__ == '__main__':
     except:
         conversations = {}
 
-    if menu_id == 'ChatBot':  
+    if st.session_state.menu_id == 'ChatBot':  
         if 'id_chat' not in st.session_state:
-            st.session_state.id_chat = 'New Chat'
+            st.session_state.id_chat = 'New Conversation'
 
         def options_list(conversations: Dict[str, list]):
-            if st.session_state.id_chat == 'New Chat':
+            if st.session_state.id_chat == 'New Conversation':
                 return [st.session_state.id_chat] + list(conversations.keys())
             else:
-                return ['New Chat'] + list(conversations.keys())
+                return ['New Conversation'] + list(conversations.keys())
         
         with st.sidebar:
             st.session_state.id_chat = st.selectbox(
@@ -623,26 +620,26 @@ if __name__ == '__main__':
                 key='chat_id'
             )
 
-            st.session_state.messages = conversations[st.session_state.id_chat] if st.session_state.id_chat != 'New Chat' else []
+            st.session_state.messages = conversations[st.session_state.id_chat] if st.session_state.id_chat != 'New Conversation' else []
             
             def update_selectbox_remove(conversations_path, conversations):
                 conversations.pop(st.session_state.chat_id)
                 with open(conversations_path, 'wb') as fp:
                     packed_bytes = msgpack.packb(conversations, use_bin_type=True)
                     fp.write(packed_bytes)
-                st.session_state.chat_id = 'New Chat'
+                st.session_state.chat_id = 'New Conversation'
                     
             
             st.button(
-                'Delete Chat', 
+                'Delete Conversation', 
                 use_container_width=True, 
-                disabled=False if st.session_state.id_chat != 'New Chat' else True,
+                disabled=False if st.session_state.id_chat != 'New Conversation' else True,
                 on_click=update_selectbox_remove,
                 args=(conversations_path, conversations)
             )
 
         def generate_conv_title(llm: vllm.LLM, tokenizer: AutoTokenizer):
-            if st.session_state.chat_id == 'New Chat':
+            if st.session_state.chat_id == 'New Conversation':
                 messages = [
                     {"role": "system", "content": 'You are a helpful assistant.'},
                     {"role": "user", "content": f"""Understand the question of the user.