Update app.py

app.py

@@ -344,20 +344,27 @@ def main():
     if 'auth_flow' not in st.session_state or 'auth_url' not in st.session_state:
         st.session_state.auth_flow, st.session_state.auth_url = google_auth(client_config)
 
-    query_params = st.query_params.to_dict()
-    # Print the query parameters to the console
-    print(query_params)
+    # Retrieve query parameters
+    query_params = st.experimental_get_query_params()
+    print("Query Parameters:", query_params)
     
-    # Display the query parameters in the Streamlit app
-    st.write("Query Parameters:", query_params['code'][0])
+    # Display the query parameters in the Streamlit app for debugging
+    st.write("Query Parameters:", query_params)
     
     # Extract the authorization code from query parameters
-    auth_code = query_params['code'][0] if 'code' in query_params else None
-    #auth_code = query_params.code if 'code' in query_params else None
-
+    auth_code = query_params.get('code', [None])[0]
+    print("Authorization Code:", auth_code)
+    
+    # Display the authorization code in the Streamlit app for debugging
+    st.write("Authorization Code:", auth_code)
+    
     if auth_code and 'credentials' not in st.session_state:
-        st.session_state.auth_flow.fetch_token(code=auth_code)
-        st.session_state.credentials = st.session_state.auth_flow.credentials
+        try:
+            st.session_state.auth_flow.fetch_token(code=auth_code)
+            st.session_state.credentials = st.session_state.auth_flow.credentials
+        except Exception as e:
+            st.error(f"Error fetching token: {e}")
+            print(f"Error fetching token: {e}")
 
     if 'credentials' not in st.session_state:
         show_google_sign_in(st.session_state.auth_url)
@@ -391,7 +398,6 @@ def main():
                 download_csv_link(st.session_state.report_data)
             elif st.session_state.report_data is not None:
                 st.warning("No data found for the selected criteria.")
-
                 
 if __name__ == "__main__":
-    main()
+    main()