diff --git a/streamlit/interface_3.py b/streamlit/interface_3.py index 8c83d06..9bda8f6 100644 --- a/streamlit/interface_3.py +++ b/streamlit/interface_3.py @@ -44,9 +44,9 @@ with st.sidebar: # st.write("Doczy") _,c1= st.columns([5,1]) -util.setup_page(REDIRECT_URI) + try: - util.setup_page(REDIRECT_URI) + st.session_state['user_info'] = util.setup_page(REDIRECT_URI) except: st.session_state['user_info'] = {'mail': 'ahinge@aarete.com', 'displayName': 'Ankit Hinge'} c1.write(f"User: **{st.session_state.user_info['displayName']}**") diff --git a/streamlit/util.py b/streamlit/util.py index 9ae66f7..814ea23 100644 --- a/streamlit/util.py +++ b/streamlit/util.py @@ -17,7 +17,7 @@ def setup_page(REDIRECT_URI): if access_token: user_info = security.get_user_info(access_token) st.session_state['user_info'] = user_info - return True + return user_info else: st.write("Please sign-in to use this app.") auth_url = security.get_auth_url(REDIRECT_URI)