import chatgptIcon from '../../assets/logos/chatgpt-logo.png';
import claudeIcon from '../../assets/logos/claude-color.svg';
import googleIcon from '../../assets/logos/g.png';
import perplexityIcon from '../../assets/logos/perplexity-ai-icon.svg';
import geminiIcon from '../../assets/logos/gemini-color.svg';
import bingIcon from '../../assets/logos/bing.png';

export const AI_BRAND_MAP: Record<string, { label: string; color: string; icon?: string }> = {
    gpt: { label: 'ChatGPT', color: '#10a37f', icon: chatgptIcon },
    claude: { label: 'Claude', color: '#a855f7', icon: claudeIcon },
    google: { label: 'Google', color: '#ea4335', icon: googleIcon },
    'google-extended': { label: 'Google', color: '#ea4335', icon: googleIcon },
    perplexity: { label: 'Perplexity', color: '#20b2aa', icon: perplexityIcon },
    gemini: { label: 'Gemini', color: '#4285f4', icon: geminiIcon },
    bing: { label: 'Bing', color: '#008ad8', icon: bingIcon },
    meta: { label: 'Meta AI', color: '#0668E1' },
    bytedance: { label: 'ByteDance', color: '#fe2c55' },
    apple: { label: 'Applebot', color: '#a2aaad' },
    amazon: { label: 'Amazon', color: '#ff9900' },
    xai: { label: 'Grok', color: '#1d9bf0' },
    deepseek: { label: 'DeepSeek', color: '#4b6bfb' },
    mistral: { label: 'Mistral', color: '#f7d046' },
    cohere: { label: 'Cohere', color: '#39594d' },
    ahrefs: { label: 'Ahrefs', color: '#e05d44' },
    semrush: { label: 'Semrush', color: '#ff642d' },
};

export const AGENT_COLORS: Record<string, string> = {
    normal_user: '#3b82f6',
    gpt: '#10a37f',
    claude: '#a855f7',
    perplexity: '#20b2aa',
    google: '#ea4335',
    bing: '#008ad8',
    meta: '#0668E1',
    bytedance: '#fe2c55',
    apple: '#a2aaad',
    amazon: '#ff9900',
    xai: '#1d9bf0',
    deepseek: '#4b6bfb',
    mistral: '#f7d046',
    cohere: '#39594d',
    ahrefs: '#e05d44',
    semrush: '#ff642d',
    bot: '#06b6d4',
};