function media_display_shortcode($atts) {
    $atts = shortcode_atts(
        array(
            'tag'    => '',
            'limit'  => -1,
            'type'   => 'all',      // video, image, all (پیش‌فرض all)
            'layout' => 'default'   // default, carousel, grid
        ),
        $atts,
        'media_display'
    );

    $tags = array();

    if (trim($atts['tag']) === 'me' && function_exists('is_product') && is_product()) {
        global $product;
        if ($product) {
            $sku = $product->get_sku();
            $model = $product->get_attribute('pa_model');
            if (!empty($sku)) {
                $tags[] = strval($sku);
            }
            if (!empty($model)) {
                $tags[] = strval($model);
            }
        }
    } elseif (strpos($atts['tag'], 'me') !== false) {
        return '';
    } else {
        $tags = explode(',', $atts['tag']);
        $tags = array_map('trim', $tags);
    }

    $meta_query = array('relation' => 'OR');
    foreach ($tags as $tag) {
        $meta_query[] = array(
            'key'     => '_media_tags',
            'value'   => $tag,
            'compare' => 'LIKE'
        );
    }

    $args = array(
        'post_type'      => 'attachment',
        'post_status'    => 'inherit',
        'posts_per_page' => intval($atts['limit']),
        'meta_query'     => $meta_query
    );

    $query = new WP_Query($args);

    // آرایه ها برای دسته بندی
    $videos = [];
    $audios = [];
    $images = [];
    $seen_ids = [];

    while ($query->have_posts()) {
        $query->the_post();
        $post_id = get_the_ID();

        if (in_array($post_id, $seen_ids)) continue;
        $seen_ids[] = $post_id;

        $mime_type = get_post_mime_type($post_id);
        $media_url = wp_get_attachment_url($post_id);
        $thumbnail_html = wp_get_attachment_image($post_id, 'large');
        $media_item = '<div class="media-item">';

        // فیلتر کردن نوع مدیا بر اساس پارامتر type
        if ($atts['type'] === 'video' && strpos($mime_type, 'video') === false) continue;
        if ($atts['type'] === 'image' && strpos($mime_type, 'image') === false) continue;

        if (strpos($mime_type, 'video') !== false) {
            $poster = esc_url(get_the_post_thumbnail_url($post_id, 'large'));
            $media_item .= '<a class="video-thumb" data-fancybox="gallery" href="' . esc_url($media_url) . '" data-caption="' . esc_attr(get_the_title($post_id)) . '">';
            $media_item .= '<img src="' . $poster . '" alt="' . esc_attr(get_the_title($post_id)) . '" />';
            $media_item .= '<span class="play-icon">&#9658;</span>';
            $media_item .= '</a>';
            $media_item .= '<span>'. esc_html(get_post_field('post_content', $post_id)).'</span>';
            $media_item .= '</div>';
            $videos[] = $media_item;
        } elseif (strpos($mime_type, 'audio') !== false) {
            $media_item .= '<audio preload="metadata" controls><source src="' . esc_url($media_url) . '" type="' . esc_attr($mime_type) . '"></audio>';
            $media_item .= '<span>'. esc_html(get_post_field('post_content', $post_id)).'</span>';
            $media_item .= '</div>';
            $audios[] = $media_item;
        } elseif (strpos($mime_type, 'image') !== false) {
            $image_id = get_post_thumbnail_id($post_id);
            $image_url = wp_get_attachment_url($image_id);
            $media_item .= '<a href="' . esc_url($media_url) . '" data-fancybox="gallery" data-caption="' . esc_attr(get_the_title($post_id)) . '">'.$thumbnail_html.'</a>';
            $media_item .= '<span>'. esc_html(get_post_field('post_content', $post_id)).'</span>';
            $media_item .= '</div>';
            $images[] = $media_item;
        }
    }

    wp_reset_postdata();

    // ترکیب آرایه‌ها بر اساس نوع درخواست
    $output_items = [];
    if ($atts['type'] === 'video') {
        $output_items = $videos;
    } elseif ($atts['type'] === 'image') {
        $output_items = $images;
    } else {
        $output_items = array_merge($videos, $images);
    }

    if (empty($output_items)) return '';

    // خروجی HTML با توجه به layout
    ob_start();

    // لود CSS و JS فنس‌باکس فقط یکبار و اگر لازم بود
    echo '
    <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/@fancyapps/ui/dist/fancybox.css" />
    <script src="https://cdn.jsdelivr.net/npm/@fancyapps/ui/dist/fancybox.umd.js"></script>
    ';

    // استایل پایه (برای grid و default)
    echo '<style>
        .media-display-wrapper { margin: 20px 0; }
        .media-gallery.default, .media-gallery.grid {
            display: flex;
            flex-wrap: wrap;
            gap: 15px;
        }
        .media-gallery.grid .media-item {
            width: calc(33.33% - 10px);
        }
        .media-gallery.default .media-item {
            flex: 1 1 100%;
            margin-bottom: 15px;
        }
        .media-item img {
            max-width: 100%;
            height: auto;
            display: block;
            border-radius: 6px;
        }
        .video-thumb {
            position: relative;
            display: inline-block;
        }
        .video-thumb .play-icon {
            position: absolute;
            top: 3px;
            right: 0;
            font-size: 20px;
            color: white;
            border-radius: 50%;
            width: 30px;
            height: 30px;
            text-align: center;
            line-height: 30px;
            pointer-events: none;
            z-index: 2;
        }
    </style>';

    // اگر حالت carousel خواستی اینجا لود کن (مثلاً Slick Carousel)
    if ($atts['layout'] === 'carousel') {
        // لود CSS و JS اسلایدر (مثال: Slick)
        echo '
        <link rel="stylesheet" type="text/css" href="https://cdn.jsdelivr.net/npm/slick-carousel@1.8.1/slick/slick.css"/>
        <script src="https://cdn.jsdelivr.net/npm/jquery@3.6.0/dist/jquery.min.js"></script>
        <script src="https://cdn.jsdelivr.net/npm/slick-carousel@1.8.1/slick/slick.min.js"></script>
        <style>
            .media-gallery.carousel .media-item {
                margin: 0 10px;
            }
        </style>
        <script>
            jQuery(document).ready(function($){
                $(".media-gallery.carousel").slick({
                    slidesToShow: 3,
                    slidesToScroll: 1,
                    arrows: true,
                    dots: true,
                    infinite: false,
                    adaptiveHeight: true,
                    responsive: [
                        { breakpoint: 768, settings: { slidesToShow: 1 } }
                    ]
                });
            });
        </script>
        ';
    }

    // کلاس wrapper بر اساس layout
    $gallery_class = 'media-gallery';
    if ($atts['layout'] === 'carousel') {
        $gallery_class .= ' carousel';
    } elseif ($atts['layout'] === 'grid') {
        $gallery_class .= ' grid';
    } else {
        $gallery_class .= ' default';
    }

    echo '<div id="media-display" class="media-display-wrapper">';
    echo '<div class="' . esc_attr($gallery_class) . '">';
    echo implode('', $output_items);
    echo '</div>';
    echo '</div>';

    // فنس‌باکس JS
    echo '<script>
        document.addEventListener("DOMContentLoaded", function () {
            Fancybox.bind("[data-fancybox=\'gallery\']", {
                on: {
                    reveal: (fancybox, slide) => {
                        const videos = document.querySelectorAll("video");
                        videos.forEach(v => v.pause && v.pause());
                    }
                },
                trapFocus: true,
                autoFocus: false,
                dragToClose: false
            });
        });
    </script>';

    return ob_get_clean();
}
add_shortcode('media_display', 'media_display_shortcode');