Woocommerce Hide Featured Image from Specific Category

woocommerce-hide-featured-image

// Useful function to hide the product featured image for all products in specific categories
// The main image displayed when visiting products in these categories becomes the first gallery image
// Written by Mitchell Krog - mitchellkrog@gmail.com
// Replace categoryslug2 / categoryslug1 with your category slug (add to array as needed)

add_filter('woocommerce_single_product_image_thumbnail_html', 'remove_featured_image', 10, 2);
function remove_featured_image($html, $attachment_id ) {
    global $product;

if(  ! has_term( array( 'categoryslug1', 'categoryslug2' ) , 'product_cat', $product->get_id() ) ){
		return $html;
       }
       else
       {
       $featured_image = get_post_thumbnail_id( $post->ID );
        if ( $attachment_id == $featured_image )
		$html = '';
		return $html;
       }
}