From 4dd55599f794062cd46cb7b80cf166fd020365f8 Mon Sep 17 00:00:00 2001 From: Mayank Aamseek Date: Tue, 2 Apr 2024 22:16:19 +0530 Subject: [PATCH] SSO bug fixed --- streamlit/security.py | 2 +- streamlit/util.py | 3 +-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/streamlit/security.py b/streamlit/security.py index 7fc266c..0701852 100644 --- a/streamlit/security.py +++ b/streamlit/security.py @@ -19,7 +19,7 @@ def get_auth_url(REDIRECT_URI): auth_url = app.get_authorization_request_url(SCOPE, redirect_uri=REDIRECT_URI) return auth_url - +@st.cache_data def get_token_from_code(auth_code, REDIRECT_URI): app = msal.ConfidentialClientApplication(CLIENT_ID, authority=AUTHORITY, client_credential=CLIENT_SECRET) result = app.acquire_token_by_authorization_code(auth_code, scopes=SCOPE, redirect_uri=REDIRECT_URI) diff --git a/streamlit/util.py b/streamlit/util.py index 814ea23..7e4565a 100644 --- a/streamlit/util.py +++ b/streamlit/util.py @@ -2,7 +2,6 @@ import streamlit as st import security -@st.cache_data def setup_page(REDIRECT_URI): # st.set_page_config( # page_title=page_title, @@ -17,7 +16,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 user_info + return True else: st.write("Please sign-in to use this app.") auth_url = security.get_auth_url(REDIRECT_URI)