Update app.py

app.py

@@ -172,28 +172,31 @@ def main():
         query_params = st.query_params
         query = (query_params.get('code'))
         if len(query) > 1: 
-
-        if 'code' in query_params:
-            #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  # 🔑 Save it for later
-            
-            st.rerun()  # Reload the app to clear the code from URL
-        else:
-            # 📢 Prompt user to log in
-            client_instance = get_msal_app()
-            authorization_url = client_instance.get_authorization_request_url(
-                scopes=SCOPES,
-                redirect_uri=REDIRECT_URI
-            )
-            st.write('👋 Please [click here]({}) to log in and authorize the app.'.format(authorization_url))
-            st.stop()
+            st.write('Parsing query ' + query_params )
+            if 'code' in query_params:
+                #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  # 🔑 Save it for later
+                
+                st.rerun()  # Reload the app to clear the code from URL
+            else:
+                # 📢 Prompt user to log in
+                client_instance = get_msal_app()
+                authorization_url = client_instance.get_authorization_request_url(
+                    scopes=SCOPES,
+                    redirect_uri=REDIRECT_URI
+                )
+                st.write('👋 Please [click here]({}) to log in and authorize the app.'.format(authorization_url))
+                st.stop()
+
+                
     else:
         # 🥳 User is authenticated, proceed with the app
         access_token = st.session_state['access_token']