app.py

@@ -177,7 +177,9 @@ def main():
 # Function to handle OAuth callback
 def handle_callback():
     flow = create_flow()
-    flow.fetch_token(code=st.query_params["code"])
+    code = st.experimental_get_query_params().get("code", [None])[0]
+    if code:
+        flow.fetch_token(code=code)
     credentials = flow.credentials
     st.session_state['credentials'] = credentials.to_json()
     return credentials