chrome.runtime.onMessage.addListener((message, sender) => {
    if (message.action === 'updateBadge' && sender.tab) {
        const count = message.count > 0 ? message.count.toString() : '';
        
        chrome.action.setBadgeText({
            text: count,
            tabId: sender.tab.id
        });

        chrome.action.setBadgeBackgroundColor({
            color: '#486860',
            tabId: sender.tab.id
        });

        if (chrome.action.setBadgeTextColor) {
            chrome.action.setBadgeTextColor({
                color: '#FFFFFF',
                tabId: sender.tab.id
            });
        }
    }
});