Update app.py

app.py

@@ -41,21 +41,21 @@ def generate_pkce_codes():
     code_challenge = base64.urlsafe_b64encode(hashlib.sha256(code_verifier.encode()).digest()).decode().rstrip('=')
     return code_verifier, code_challenge
 
-def get_msal_app():
+def get_msal_app(code_verifier=None):
     return msal.PublicClientApplication(
         client_id=APPLICATION_ID_KEY,
-        authority=AUTHORITY_URL
+        authority=AUTHORITY_URL,
+        token_cache=msal.SerializableTokenCache()
     )
 
-def get_access_token(code, code_verifier):
+def get_access_token(code):
     client_instance = get_msal_app()
     
     try:
         result = client_instance.acquire_token_by_authorization_code(
             code=code,
             scopes=st.session_state.get('request_scopes', BASE_SCOPES),
-            redirect_uri=REDIRECT_URI,
-            code_verifier=code_verifier
+            redirect_uri=REDIRECT_URI
         )
         
         if 'access_token' in result:
@@ -111,7 +111,7 @@ def main():
             st.write('🔑 Authorization Code Obtained:', code[:10] + '...')
             
             try:
-                access_token = get_access_token(code, code_verifier)
+                access_token = get_access_token(code)
                 st.session_state['access_token'] = access_token
                 st.success("Access token acquired successfully!")
                 st.rerun()