Update app.py

app.py

@@ -304,6 +304,11 @@ def get_user_info(access_token):
     else:
         raise Exception(f"Failed to fetch user info: {response.status_code} - {response.text}")
 
+def generate_state():
+    state = secrets.token_urlsafe(32)
+    st.session_state['auth_state'] = state
+    return state
+
 def initiate_auth_flow():
     client_instance = get_msal_app()
     auth_url = client_instance.get_authorization_request_url(
@@ -313,11 +318,6 @@ def initiate_auth_flow():
     )
     st.write('👋 Please [click here]({}) to log in and authorize the app.'.format(auth_url))
 
-def generate_state():
-    state = secrets.token_urlsafe(32)
-    st.session_state['auth_state'] = state
-    return state
-
 def is_token_valid(token):
     if not token:
         return False