display Tax and vat conditionally

display Tax and vat conditionally

add_filter( 'woocommerce_countries_tax_or_vat', function( $label ) {
	$country = '';

	// Try to get customer's country from session
	if ( WC()->customer ) {
		$country = WC()->customer->get_taxable_address()[0];
	}

	// Fallback to store base country if no customer country yet (e.g., first visit)
	if ( empty( $country ) ) {
		$country = WC()->countries->get_base_country();
	}
 

	return strtoupper( $country ) === 'US' ? __( 'Tax', 'woocommerce' ) : $label;
} );