Set image thumbnail for media image results on search page

searchwp-customisations.php

<?php
// Set image thumbnail for media image results on search page
add_filter( 'post_thumbnail_html', function( $html, $post_id ) {
    // Check if we're in a search context
    if (
		(
			is_search()
			|| doing_action( 'wp_ajax_searchwp_live_search' )
			|| doing_action( 'wp_ajax_nopriv_searchwp_live_search' )
			|| isset( $_REQUEST['swps'] )
		)
		&& 'attachment' === get_post_type( $post_id )
	) {
        $mime_type = get_post_mime_type( $post_id );

        // If it’s an image attachment, use the original thumbnail
        if ( strpos( $mime_type, "image" ) !== false ) {
            $html = wp_get_attachment_image( $post_id, 'thumbnail' );
        }
    }

    
    return $html;
}, 10, 2 );