Update app.py

app.py

@@ -171,30 +171,30 @@ def main():
         # 🕵️‍♂️ Check for authorization code in query parameters
         query_params = st.query_params
         query = (query_params.get('code'))
-        if len(query) > 1: 
+        #if len(query) > 1: 
+        
+        st.write('Parsing query ' + query_params )
+        if 'code' in query_params:
+            #code = query_params['code'][0]
+            code = query_params.get('code')
             
-            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()
+            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: