Update app.py

app.py

@@ -22,7 +22,28 @@ PRODUCT_SCOPES = {
 
 BASE_SCOPES = ['User.Read']
 
-# ... (keep the get_msal_app and get_access_token functions as they were) ...
+def get_msal_app():
+    return msal.ConfidentialClientApplication(
+        client_id=APPLICATION_ID_KEY,
+        client_credential=CLIENT_SECRET_KEY,
+        authority=AUTHORITY_URL
+    )
+
+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
+        )
+        if 'access_token' in result:
+            return result['access_token']
+        else:
+            raise Exception(f"Error acquiring token: {result.get('error_description')}")
+    except Exception as e:
+        st.error(f"Exception in get_access_token: {str(e)}")
+        raise
 
 def make_api_call(access_token, endpoint, method='GET', data=None):
     headers = {'Authorization': f'Bearer {access_token}', 'Content-Type': 'application/json'}