app.py

@@ -180,13 +180,13 @@ def main():
 
         if st.button("Logout"):
             del st.session_state['credentials']
-            st.rerun()  # Changed from st.experimental_rerun() to st.rerun()
+            st.rerun()
 
 # Function to handle OAuth callback
 def handle_callback():
     flow = create_flow()
     try:
-        flow.fetch_token(code=st.experimental_get_query_params()["code"][0])
+        flow.fetch_token(code=st.query_params["code"])
         credentials = flow.credentials
         st.session_state['credentials'] = credentials.to_json()
         logging.debug("Token fetch successful")
@@ -198,7 +198,7 @@ def handle_callback():
 
 if __name__ == '__main__':
     logging.debug("Starting the application")
-    if 'code' in st.experimental_get_query_params():
+    if 'code' in st.query_params:
         logging.debug("Authorization code found in query parameters")
         handle_callback()
     main()