Update app.py

app.py

@@ -20,13 +20,18 @@ SCOPES = ['User.Read', 'Calendars.ReadWrite', 'Mail.ReadWrite', 'Notes.ReadWrite
 
 # New function to generate PKCE code verifier and challenge
 def generate_pkce_codes():
-    code_verifier = secrets.token_urlsafe(64)
+    code_verifier = secrets.token_urlsafe(128)[:128]
     code_challenge = base64.urlsafe_b64encode(hashlib.sha256(code_verifier.encode()).digest()).decode().rstrip('=')
     return code_verifier, code_challenge
 
-
+def get_msal_app(code_challenge=None):
+    return msal.PublicClientApplication(
+        client_id=APPLICATION_ID_KEY,
+        authority=AUTHORITY_URL
+    )
+    
 # 🛠️ Initialize the MSAL client
-def get_msal_app():
+def get_msal_app_old():
     return msal.ConfidentialClientApplication(
         client_id=APPLICATION_ID_KEY,
         client_credential=CLIENT_SECRET_KEY,
@@ -104,7 +109,6 @@ def process_query_params():
             st.write('🔑 Authorization Code Obtained:', code[:10] + '...')
             
             try:
-                # Retrieve the code verifier from session state
                 code_verifier = st.session_state.get('code_verifier')
                 if not code_verifier:
                     st.error("Code verifier not found in session state.")
@@ -113,6 +117,8 @@ def process_query_params():
                 access_token = get_access_token(code, code_verifier)
                 st.session_state['access_token'] = access_token
                 st.success("Access token acquired successfully!")
+                # Clear the code from the URL
+                st.experimental_set_query_params()
                 st.rerun()
             except Exception as e:
                 st.error(f"Error acquiring access token: {str(e)}")