diff --git a/streamlit_authenticator/models/cookie_model.py b/streamlit_authenticator/models/cookie_model.py index 56e1c8b..9f199e0 100644 --- a/streamlit_authenticator/models/cookie_model.py +++ b/streamlit_authenticator/models/cookie_model.py @@ -76,12 +76,12 @@ def get_cookie(self) -> Optional[Dict[str, Any]]: If valid, returns a dictionary containing the cookie's data. Returns None if the cookie is expired or invalid. """ - if st.session_state['logout']: - return False - # self.token = self.cookie_manager.get(self.cookie_name) + if st.session_state.get('logout'): + return None self.token = st.context.cookies[self.cookie_name] if self.cookie_name in \ st.context.cookies else None if self.token is not None: + self.cookie_manager.set(self.cookie_name, self.token) self.token = self._token_decode() if (self.token is not False and 'username' in self.token and self.token['exp_date'] > datetime.now().timestamp()):