app.py

@@ -177,12 +177,14 @@ def main():
 # Function to handle OAuth callback
 def handle_callback():
     flow = create_flow()
-    code = st.experimental_get_query_params().get("code", [None])[0]
+    code = st.query_params.get("code")
     if code:
         flow.fetch_token(code=code)
-    credentials = flow.credentials
-    st.session_state['credentials'] = credentials.to_json()
-    return credentials
+        credentials = flow.credentials
+        st.session_state['credentials'] = credentials.to_json()
+        return credentials
+    else:
+        st.error("No authorization code found in the URL parameters.")
 
 if __name__ == '__main__':
     if 'code' in st.query_params: