[candidate_photo] shortcode to output candidate photo (as thumbnail, etc)

functions.php

<?php
function display_candidate_photo_shortcode($atts) {
    // Check if the function exists before calling it
    if (function_exists('the_candidate_photo')) {
        ob_start();
        // Options are full, thumbnail, medium, large
        the_candidate_photo( 'thumbnail' );
        return ob_get_clean();
    } else {
        return 'Candidate photo function not available.';
    }
}

add_shortcode('candidate_photo', 'display_candidate_photo_shortcode');