Update app.py

app.py

@@ -169,12 +169,21 @@ def main():
     # 🔑 Authentication
     if 'access_token' not in st.session_state:
         # 🕵️‍♂️ Check for authorization code in query parameters
-        query_params = st.get_query_params()
+        query_params = st.query_params
+        query = (query_params.get('code'))
+        if len(query) > 1: 
+
         if 'code' in query_params:
-            code = query_params['code'][0]
-            st.write('🔑 Acquiring access token...')
+            #code = query_params['code'][0]
+            code = query_params.get('code')
+            
+            st.write('🔑Access Code Obtained from MS Graph Redirect🔑!:' + code)
+            
+            st.write('🔑 Acquiring access token from redirect URL for code parameter.')
             access_token = get_access_token(code)
-            st.session_state['access_token'] = access_token
+            
+            st.session_state['access_token'] = access_token  # 🔑 Save it for later
+            
             st.rerun()  # Reload the app to clear the code from URL
         else:
             # 📢 Prompt user to log in