app.py

@@ -108,6 +108,7 @@ def main():
             authorization_url, state = flow.authorization_url(prompt="consent")
             st.session_state['oauth_state'] = state
             st.session_state['oauth_state_time'] = time.time()
+            st.experimental_set_query_params(state=state)
             st.markdown(f"[Login with Google]({authorization_url})")
     else:
         st.title('📚 Student Grade Lookup')
@@ -188,8 +189,9 @@ def main():
 # Function to handle OAuth callback
 def handle_callback():
     flow = create_flow()
-    code = st.query_params.get("code")
-    state = st.query_params.get("state")
+    query_params = st.experimental_get_query_params()
+    code = query_params.get("code", [None])[0]
+    state = query_params.get("state", [None])[0]
     
     if code and state:
         if 'oauth_state' not in st.session_state or 'oauth_state_time' not in st.session_state:
@@ -215,6 +217,7 @@ def handle_callback():
             logging.debug("Token fetch successful")
             del st.session_state['oauth_state']
             del st.session_state['oauth_state_time']
+            st.experimental_set_query_params()  # Clear query params
             return credentials
         except InvalidGrantError as e:
             logging.error(f"InvalidGrantError: {str(e)}")
@@ -230,7 +233,8 @@ def handle_callback():
 
 if __name__ == '__main__':
     logging.debug("Starting the application")
-    if 'code' in st.query_params and 'state' in st.query_params:
+    query_params = st.experimental_get_query_params()
+    if 'code' in query_params and 'state' in query_params:
         logging.debug("Authorization code and state found in query parameters")
         handle_callback()
         st.rerun()