<?php
namespace AskTamMCP\Tools;
defined('ABSPATH') || exit;

/**
 * MediaPromptEngine — AI-aware prompt enrichment for image and video generation.
 *
 * Inspired by MotionSites.ai design vocabulary:
 *   - cinematic dark heroes with video backgrounds and glass effects
 *   - left-aligned content composition with clear text overlay zones
 *   - industry-matched visual language (dark/SaaS vs warm/local-business)
 *
 * Called by: pexels_search_photos, pexels_search_videos,
 *            generate_gpt_image, generate_gemini_image
 */
class MediaPromptEngine {

    const CONTEXT_MAP = [
        'hero_background' => [
            'width' => 1920, 'height' => 1080, 'aspect' => '16:9',
            'orientation' => 'landscape', 'size' => '1536x1024', 'quality' => 'high',
            'composition' => 'wide angle, cinematic, left third of frame clear for text overlay, subject or scene on right, dark gradient on left edge',
            'format_note' => 'full-screen website hero background, 16:9 landscape',
        ],
        'section_background' => [
            'width' => 1920, 'height' => 700, 'aspect' => '16:6',
            'orientation' => 'landscape', 'size' => '1536x1024', 'quality' => 'medium',
            'composition' => 'wide panoramic, soft and low-contrast, content sits on top so avoid busy center, muted colors acceptable',
            'format_note' => 'website section background strip, panoramic landscape',
        ],
        'card_image' => [
            'width' => 800, 'height' => 600, 'aspect' => '4:3',
            'orientation' => 'landscape', 'size' => '1024x1024', 'quality' => 'medium',
            'composition' => 'subject centered, clean background, sharp focus, good contrast',
            'format_note' => 'website card or feature image, 4:3 ratio',
        ],
        'team_photo' => [
            'width' => 600, 'height' => 750, 'aspect' => '4:5',
            'orientation' => 'portrait', 'size' => '1024x1536', 'quality' => 'high',
            'composition' => 'professional headshot or half-body portrait, neutral or blurred background, face sharp and well-lit, friendly expression',
            'format_note' => 'team member or staff portrait, 4:5 portrait',
        ],
        'feature_icon' => [
            'width' => 500, 'height' => 500, 'aspect' => '1:1',
            'orientation' => 'square', 'size' => '1024x1024', 'quality' => 'medium',
            'composition' => 'flat design or minimal illustration, single subject, centered on white or transparent background, icon-style, clean lines',
            'format_note' => 'feature section icon, square 1:1',
        ],
        'gallery_image' => [
            'width' => 900, 'height' => 600, 'aspect' => '3:2',
            'orientation' => 'landscape', 'size' => '1536x1024', 'quality' => 'high',
            'composition' => 'editorial photography style, strong composition, vibrant or intentional color palette, high detail',
            'format_note' => 'gallery or portfolio item, 3:2 landscape',
        ],
        'cta_background' => [
            'width' => 1920, 'height' => 500, 'aspect' => '16:5',
            'orientation' => 'landscape', 'size' => '1536x1024', 'quality' => 'medium',
            'composition' => 'wide panoramic, very subtle and blurred, will have overlaid text and button — avoid any text or detail in center',
            'format_note' => 'CTA strip background, very wide panoramic',
        ],
        'blog_thumbnail' => [
            'width' => 1200, 'height' => 675, 'aspect' => '16:9',
            'orientation' => 'landscape', 'size' => '1536x1024', 'quality' => 'medium',
            'composition' => 'editorial, subject left or center, room for text overlay on right side',
            'format_note' => 'blog post thumbnail, 16:9',
        ],
        'logo_placeholder' => [
            'width' => 400, 'height' => 200, 'aspect' => '2:1',
            'orientation' => 'landscape', 'size' => '1024x1024', 'quality' => 'low',
            'composition' => 'minimal logo-style illustration, centered, white or transparent background, clean vector look',
            'format_note' => 'brand logo placeholder, 2:1',
        ],
        'video_hero' => [
            'width' => 1920, 'height' => 1080, 'aspect' => '16:9',
            'orientation' => 'landscape', 'duration' => [15, 30],
            'motion' => 'slow cinematic motion, subtle parallax, ambient loop, no jump cuts',
            'format_note' => 'full-screen hero video background loop, 16:9, 15-30 seconds',
        ],
        'video_section' => [
            'width' => 1920, 'height' => 1080, 'aspect' => '16:9',
            'orientation' => 'landscape', 'duration' => [5, 15],
            'motion' => 'ambient motion, slow pan or zoom, atmospheric, loops cleanly',
            'format_note' => 'section video accent, 16:9, 5-15 seconds',
        ],
    ];

    const INDUSTRY_PROFILES = [
        'saas' => [
            'style'    => 'dark cinematic, premium SaaS aesthetic, deep space or dark backgrounds, purple-black tones (#070612 style), glassmorphism accents, subtle glow effects',
            'lighting' => 'dramatic rim lighting, blue-purple ambience, volumetric light rays',
            'mood'     => 'futuristic, powerful, cutting-edge',
            'avoid'    => 'bright white backgrounds, cartoonish, stock-photo clichés, clip art',
        ],
        'tech' => [
            'style'    => 'dark high-tech, circuit board aesthetic, deep navy or charcoal backgrounds, electric blue accents, clean lines',
            'lighting' => 'cool blue-white lighting, neon highlights, screen glow',
            'mood'     => 'innovative, precise, powerful',
            'avoid'    => 'warm tones, soft textures, vintage, cartoonish',
        ],
        'agency' => [
            'style'    => 'premium agency aesthetic, bold typography composition, dark editorial, high contrast, cinematic film-like quality',
            'lighting' => 'directional cinematic lighting, strong shadows, dramatic contrast',
            'mood'     => 'creative, bold, premium',
            'avoid'    => 'generic stock, soft pastels, corporate clichés',
        ],
        'finance' => [
            'style'    => 'clean professional, dark navy or deep charcoal, gold accents, architectural minimalism',
            'lighting' => 'subtle warm-cool contrast, sophisticated ambient',
            'mood'     => 'trustworthy, serious, authoritative',
            'avoid'    => 'bright colors, playful, casual, cartoon',
        ],
        'web3' => [
            'style'    => 'crypto-native, dark backgrounds, neon gradients, holographic elements, abstract digital art',
            'lighting' => 'neon glow, holographic iridescence, particle effects',
            'mood'     => 'decentralized, innovative, futuristic',
            'avoid'    => 'corporate, traditional, stock photography',
        ],
        'medical' => [
            'style'    => 'clean medical professional, bright whites and soft blues, modern clinical interior, sterile but warm',
            'lighting' => 'bright natural or clinical lighting, soft fill, no harsh shadows',
            'mood'     => 'trustworthy, clean, professional, reassuring',
            'avoid'    => 'dark moody, dramatic, horror, industrial',
        ],
        'dental' => [
            'style'    => 'modern dental clinic interior, bright white, clean professional equipment visible, warm and welcoming',
            'lighting' => 'bright clinical with warm accents, soft fill light, clean',
            'mood'     => 'clean, professional, reassuring, modern',
            'avoid'    => 'dark, dramatic, blood or discomfort imagery, vintage',
        ],
        'health' => [
            'style'    => 'health and wellness aesthetic, fresh greens and whites, natural light, active lifestyle imagery',
            'lighting' => 'bright natural daylight, golden hour for outdoor, clean studio for indoor',
            'mood'     => 'energetic, healthy, positive, motivating',
            'avoid'    => 'dark, industrial, clinical coldness',
        ],
        'gym' => [
            'style'    => 'premium fitness aesthetic, high-contrast black and red or dark with neon, athletic energy, motion blur acceptable',
            'lighting' => 'dramatic spotlighting on subject, dark moody gym background',
            'mood'     => 'powerful, intense, motivating',
            'avoid'    => 'soft pastels, low energy, stock-photo clichés',
        ],
        'restaurant' => [
            'style'    => 'warm restaurant atmosphere, rich food photography colors, ambient candlelit or warm Edison bulb tones, appetizing',
            'lighting' => 'warm golden ambient, soft directional on food, bokeh background',
            'mood'     => 'inviting, delicious, warm, intimate',
            'avoid'    => 'cold blue tones, clinical, unappealing food presentation',
        ],
        'food' => [
            'style'    => 'premium food photography, hero angle or overhead, rich saturated colors, texture-forward, appetizing styling',
            'lighting' => 'natural side light or soft studio, warm highlights on food surfaces',
            'mood'     => 'appetizing, fresh, premium',
            'avoid'    => 'dark moody tones for food itself, flat lighting',
        ],
        'real_estate' => [
            'style'    => 'architectural real estate photography, wide-angle interior or exterior, golden hour exterior shots, clean and aspirational',
            'lighting' => 'golden hour exterior, balanced interior with window light, HDR-like clarity',
            'mood'     => 'aspirational, spacious, prestigious',
            'avoid'    => 'cluttered, dark, unflattering angles',
        ],
        'legal' => [
            'style'    => 'professional law firm aesthetic, dark wood and leather, serious and authoritative, classical architecture elements',
            'lighting' => 'warm professional lighting, wood-tone ambient',
            'mood'     => 'authoritative, trustworthy, serious',
            'avoid'    => 'playful, casual, bright pop colors',
        ],
        'education' => [
            'style'    => 'modern educational environment, bright and inspiring, diverse and inclusive imagery, clean classroom or campus',
            'lighting' => 'bright natural daylight, warm and inviting',
            'mood'     => 'inspiring, inclusive, forward-thinking',
            'avoid'    => 'dark, institutional, boring',
        ],
        'beauty' => [
            'style'    => 'luxury beauty and cosmetics, soft pink or gold tones, glamorous editorial photography, smooth textures',
            'lighting' => 'soft beauty lighting, catchlights, ethereal glow',
            'mood'     => 'luxurious, feminine, aspirational',
            'avoid'    => 'harsh industrial, dark moody, masculine aesthetics',
        ],
        'fashion' => [
            'style'    => 'high-fashion editorial, full-bleed model photography, bold art direction, strong silhouettes',
            'lighting' => 'dramatic directional, editorial quality, high contrast',
            'mood'     => 'aspirational, bold, editorial',
            'avoid'    => 'generic stock, catalog style, weak composition',
        ],
        'travel' => [
            'style'    => 'destination travel photography, golden hour landscapes, immersive wide vistas, vibrant local color',
            'lighting' => 'golden hour, magic hour, dramatic natural light',
            'mood'     => 'wanderlust, adventurous, escapist',
            'avoid'    => 'tourist trap clichés, overprocessed HDR',
        ],
        'construction' => [
            'style'    => 'industrial construction professional, bold and powerful, concrete and steel, strong workers or architecture',
            'lighting' => 'dramatic natural light, industrial shadows',
            'mood'     => 'strong, reliable, professional',
            'avoid'    => 'soft or delicate, overly corporate, stock-photo generic',
        ],
        'automotive' => [
            'style'    => 'automotive photography, dramatic studio or location, motion blur or static hero shot, premium finish on vehicle',
            'lighting' => 'studio rim lighting or golden-hour outdoor, reflections on bodywork',
            'mood'     => 'powerful, luxury, dynamic',
            'avoid'    => 'flat lighting, cluttered backgrounds, generic',
        ],
        'general' => [
            'style'    => 'photorealistic professional photography, clean and modern aesthetic, neutral tones with one accent color',
            'lighting' => 'soft natural or studio light, balanced exposure',
            'mood'     => 'professional, trustworthy, approachable',
            'avoid'    => 'cartoonish, clip art, overly dramatic, generic stock',
        ],
    ];

    const UNIVERSAL_IMAGE_SUFFIX = 'No text overlays. No watermarks. No logos. No borders. Photorealistic unless otherwise specified. Award-winning photography quality. Ultra-detailed.';
    const UNIVERSAL_VIDEO_SUFFIX = 'No text overlays. No logos. No watermarks. Seamlessly loopable. Cinematic quality. 4K resolution preferred. Smooth motion, no jump cuts.';

    public static function build(
        string $raw_prompt,
        string $usage_context = 'hero_background',
        string $industry      = 'general',
        array  $overrides     = []
    ): array {
        $ctx          = self::CONTEXT_MAP[$usage_context] ?? self::CONTEXT_MAP['hero_background'];
        $industry_key = self::resolveIndustry($industry);
        $profile      = self::INDUSTRY_PROFILES[$industry_key];
        $admin_profile = self::getAdminStyleProfile();
        $is_video     = str_starts_with($usage_context, 'video_');

        $enriched = $is_video
            ? self::buildVideoPrompt($raw_prompt, $ctx, $profile, $admin_profile, $overrides)
            : self::buildImagePrompt($raw_prompt, $ctx, $profile, $admin_profile, $overrides);

        $pexels_query = self::buildPexelsQuery($raw_prompt, $usage_context, $industry_key);

        $result = [
            'enriched_prompt'   => $enriched,
            'pexels_query'      => $pexels_query,
            'width'             => $ctx['width'],
            'height'            => $ctx['height'],
            'size'              => $ctx['size']        ?? '1536x1024',
            'orientation'       => $ctx['orientation'] ?? 'landscape',
            'quality'           => $ctx['quality']     ?? 'medium',
            'aspect'            => $ctx['aspect']      ?? '16:9',
            'format_note'       => $ctx['format_note'] ?? '',
            'is_video'          => $is_video,
            'industry_detected' => $industry_key,
            'usage_context'     => $usage_context,
        ];

        if ($is_video) {
            $result['video_duration_min'] = $ctx['duration'][0] ?? 10;
            $result['video_duration_max'] = $ctx['duration'][1] ?? 30;
            $result['video_motion']       = $ctx['motion']      ?? 'slow cinematic motion';
        }

        return $result;
    }

    private static function buildImagePrompt(string $raw, array $ctx, array $profile, string $admin, array $ov): string {
        $parts = [
            ucfirst(trim($raw)),
            $ctx['composition'],
            'Visual style: ' . ($ov['style_override'] ?? $profile['style']),
            'Lighting: ' . $profile['lighting'],
            'Mood: ' . $profile['mood'],
            'Format: ' . $ctx['format_note'],
            $admin ? "Brand profile: {$admin}" : null,
            isset($ov['color_hint']) ? "Color palette: {$ov['color_hint']}" : null,
            self::UNIVERSAL_IMAGE_SUFFIX,
            'Do NOT include: ' . $profile['avoid'],
        ];
        return implode('. ', array_filter($parts));
    }

    private static function buildVideoPrompt(string $raw, array $ctx, array $profile, string $admin, array $ov): string {
        $parts = [
            ucfirst(trim($raw)),
            $ctx['motion'] ?? 'slow ambient motion',
            'Visual style: ' . ($ov['style_override'] ?? $profile['style']),
            'Duration: ' . ($ctx['duration'][0] ?? 10) . '-' . ($ctx['duration'][1] ?? 30) . ' seconds',
            'Format: ' . $ctx['format_note'],
            $admin ? "Brand profile: {$admin}" : null,
            self::UNIVERSAL_VIDEO_SUFFIX,
        ];
        return implode('. ', array_filter($parts));
    }

    private static function buildPexelsQuery(string $raw, string $context, string $industry): string {
        $context_suffix = match(true) {
            $context === 'hero_background'       => 'wide cinematic',
            $context === 'section_background'    => 'wide background',
            $context === 'team_photo'            => 'professional portrait',
            str_starts_with($context, 'video_') => 'cinematic video',
            default                              => '',
        };
        $industry_suffix = match($industry) {
            'saas', 'tech'       => 'technology dark',
            'medical', 'dental'  => 'medical professional clean',
            'restaurant', 'food' => 'restaurant food warm',
            'gym', 'health'      => 'fitness active',
            'real_estate'        => 'architecture interior',
            'legal'              => 'professional office',
            default              => '',
        };
        $parts = array_filter([trim($raw), $context_suffix, $industry_suffix]);
        $query = implode(' ', $parts);
        if (str_word_count($query) > 6) {
            $query = implode(' ', array_slice(explode(' ', $query), 0, 5));
        }
        return $query;
    }

    public static function resolveIndustry(string $industry): string {
        $industry = strtolower(trim($industry));
        $map = [
            'saas' => 'saas', 'software' => 'saas', 'ai' => 'saas', 'startup' => 'saas', 'app' => 'saas',
            'tech' => 'tech', 'technology' => 'tech', 'cybersecurity' => 'tech', 'data' => 'tech',
            'agency' => 'agency', 'design' => 'agency', 'creative' => 'agency', 'marketing' => 'agency', 'portfolio' => 'agency', 'studio' => 'agency',
            'finance' => 'finance', 'fintech' => 'finance', 'banking' => 'finance', 'insurance' => 'finance', 'investment' => 'finance',
            'web3' => 'web3', 'crypto' => 'web3', 'blockchain' => 'web3', 'defi' => 'web3', 'nft' => 'web3',
            'medical' => 'medical', 'dental clinic' => 'dental', 'dental center' => 'dental', 'clinic' => 'medical', 'hospital' => 'medical', 'healthcare' => 'medical', 'pharmacy' => 'medical',
            'dental' => 'dental', 'dentist' => 'dental', 'orthodontic' => 'dental',
            'health' => 'health', 'wellness' => 'health', 'yoga' => 'health', 'spa' => 'health', 'therapy' => 'health',
            'gym' => 'gym', 'fitness' => 'gym', 'crossfit' => 'gym', 'boot camp' => 'gym', 'sport' => 'gym', 'training' => 'gym',
            'restaurant' => 'restaurant', 'cafe' => 'restaurant', 'coffee' => 'restaurant', 'bar' => 'restaurant', 'bakery' => 'restaurant',
            'food' => 'food', 'catering' => 'food',
            'real_estate' => 'real_estate', 'real estate' => 'real_estate', 'property' => 'real_estate', 'architecture' => 'real_estate', 'interior' => 'real_estate',
            'legal' => 'legal', 'law' => 'legal', 'lawyer' => 'legal', 'attorney' => 'legal',
            'education' => 'education', 'school' => 'education', 'university' => 'education', 'course' => 'education', 'academy' => 'education', 'elearning' => 'education',
            'hair salon' => 'beauty', 'nail salon' => 'beauty', 'beauty' => 'beauty', 'salon' => 'beauty', 'cosmetics' => 'beauty', 'skincare' => 'beauty', 'hair' => 'beauty', 'barbershop' => 'beauty',
            'fashion' => 'fashion', 'clothing' => 'fashion', 'apparel' => 'fashion',
            'travel' => 'travel', 'hotel' => 'travel', 'tourism' => 'travel', 'resort' => 'travel',
            'construction' => 'construction', 'contractor' => 'construction', 'building' => 'construction',
            'automotive' => 'automotive', 'car' => 'automotive', 'vehicle' => 'automotive',
        ];
        if (isset($map[$industry])) return $map[$industry];
        foreach ($map as $keyword => $profile) {
            if (strlen($keyword) < 3) continue;
            if (str_contains($industry, $keyword)) return $profile;
        }
        return 'general';
    }

    public static function getAdminStyleProfile(): string {
        $profiles = get_option('asktam_mcp_media_style_profiles', []);
        if (!is_array($profiles)) return '';
        $active = $profiles['active'] ?? '';
        if (!$active) return '';
        return $profiles['profiles'][$active]['description'] ?? '';
    }

    public static function imagePrompt(string $raw, string $context = 'hero_background', string $industry = 'general'): string {
        return self::build($raw, $context, $industry)['enriched_prompt'];
    }

    public static function pexelsQuery(string $raw, string $context = 'hero_background', string $industry = 'general'): string {
        return self::build($raw, $context, $industry)['pexels_query'];
    }
}