Update app.py

app.py

@@ -422,20 +422,13 @@ if __name__ == '__main__':
         conversations = {}
 
     if menu_id == 'ChatBot':
+        if 'chat_id' not in st.session_state:
+            st.session_state.chat_id = 'New Chat'
+            
         with st.sidebar:
-            if 'options' not in st.session_state:
-                st.session_state.options = 'New Chat'
-    
-            if 'chat_id' not in st.session_state:
-                st.session_state.chat_id = st.selectbox(
-                    label='Choose a conversation',
-                    options=st.session_state.options,
-                    placeholder='_',
-                )
-            else:
                 st.session_state.chat_id = st.selectbox(
                     label='Choose a conversation',
-                    options=[st.session_state.options] + conversations.keys(),
+                    options=[st.session_state.chat_id] + conversations.keys(),
                     index=0,
                     placeholder='_',
                 )
@@ -443,7 +436,7 @@ if __name__ == '__main__':
             col1, col2 = st.columns([1,1])
             with col1:
                 if st.button('Add New Chat', use_container_width=True, disabled=False if st.session_state.chat_id != 'New Chat' else True):
-                    st.session_state.options = 'New Chat'
+                    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]
@@ -453,10 +446,10 @@ if __name__ == '__main__':
             st.session_state.messages = []
         else:
             st.session_state.messages = conversations[st.session_state.chat_id]
-        
-        for message in st.session_state.messages:
-            with st.chat_message(message["role"]):
-                st.markdown(message["content"])
+            
+            for message in st.session_state.messages:
+                with st.chat_message(message["role"]):
+                    st.markdown(message["content"])
                 
         if prompt := st.chat_input("Message Video Game Assistant"):
             if st.session_state.chat_id == 'New Chat':