document.addEventListener('DOMContentLoaded', () => {
    const themeSwitch = document.getElementById('theme-switch');
    const currentTheme = localStorage.getItem('theme') ? localStorage.getItem('theme') : 'light';

    if (currentTheme) {
        document.body.setAttribute('data-bs-theme', currentTheme);
        if (themeSwitch) {
            themeSwitch.innerHTML = currentTheme === 'dark' ? '<i class="bi bi-sun-fill"></i>' : '<i class="bi bi-moon-fill"></i>';
        }
    }

    if(themeSwitch) {
        themeSwitch.addEventListener('click', () => {
            let theme = document.body.getAttribute('data-bs-theme');
            if (theme === 'dark') {
                document.body.setAttribute('data-bs-theme', 'light');
                localStorage.setItem('theme', 'light');
                themeSwitch.innerHTML = '<i class="bi bi-moon-fill"></i>';
            } else {
                document.body.setAttribute('data-bs-theme', 'dark');
                localStorage.setItem('theme', 'dark');
                themeSwitch.innerHTML = '<i class="bi bi-sun-fill"></i>';
            }
        });
    }
});