import { getContext, renderExtensionTemplateAsync, extension_settings as st_extension_settings } from '../../../extensions.js';
import { eventSource, event_types, substituteParams, chat, generateRaw, saveSettingsDebounced, chat_metadata, saveChatDebounced, user_avatar, getThumbnailUrl, characters, this_chid, extension_prompt_types, extension_prompt_roles, setExtensionPrompt, reloadCurrentChat, Generate, getRequestHeaders } from '../../../../script.js';
import { selected_group, getGroupMembers } from '../../../group-chats.js';
import { power_user } from '../../../power-user.js';

// Core modules
import { extensionName, extensionFolderPath } from './src/core/config.js';
import { i18n } from './src/core/i18n.js';
import {
    extensionSettings,
    lastGeneratedData,
    committedTrackerData,
    lastActionWasSwipe,
    isGenerating,
    isPlotProgression,
    pendingDiceRoll,
    FALLBACK_AVATAR_DATA_URI,
    $panelContainer,
    $userStatsContainer,
    $infoBoxContainer,
    $thoughtsContainer,
    $inventoryContainer,
    $questsContainer,
    setExtensionSettings,
    updateExtensionSettings,
    setLastGeneratedData,
    updateLastGeneratedData,
    setCommittedTrackerData,
    updateCommittedTrackerData,
    setLastActionWasSwipe,
    setIsGenerating,
    setIsPlotProgression,
    setPendingDiceRoll,
    setPanelContainer,
    setUserStatsContainer,
    setInfoBoxContainer,
    setThoughtsContainer,
    setInventoryContainer,
    setQuestsContainer,
    clearSessionAvatarPrompts
} from './src/core/state.js';
import { loadSettings, saveSettings, saveChatData, loadChatData, updateMessageSwipeData } from './src/core/persistence.js';
import { registerAllEvents } from './src/core/events.js';

// Generation & Parsing modules
import {
    generateTrackerExample,
    generateTrackerInstructions,
    generateContextualSummary,
    generateRPGPromptText,
    generateSeparateUpdatePrompt
} from './src/systems/generation/promptBuilder.js';
import { parseResponse, parseUserStats } from './src/systems/generation/parser.js';
import { updateRPGData, testExternalAPIConnection } from './src/systems/generation/apiClient.js';
import { onGenerationStarted } from './src/systems/generation/injector.js';

// Rendering modules
import { getSafeThumbnailUrl } from './src/utils/avatars.js';
import { renderUserStats } from './src/systems/rendering/userStats.js';
import { renderInfoBox, updateInfoBoxField } from './src/systems/rendering/infoBox.js';
import {
    renderThoughts,
    updateCharacterField,
    updateChatThoughts,
    createThoughtPanel
} from './src/systems/rendering/thoughts.js';
import { renderInventory } from './src/systems/rendering/inventory.js';
import { renderQuests } from './src/systems/rendering/quests.js';

// Interaction modules
import { initInventoryEventListeners } from './src/systems/interaction/inventoryActions.js';

// UI Systems modules
import {
    applyTheme,
    applyCustomTheme,
    toggleCustomColors,
    toggleAnimations,
    updateSettingsPopupTheme,
    applyCustomThemeToSettingsPopup
} from './src/systems/ui/theme.js';
import {
    DiceModal,
    SettingsModal,
    setupDiceRoller,
    setupSettingsPopup,
    updateDiceDisplay,
    addDiceQuickReply,
    getSettingsModal
} from './src/systems/ui/modals.js';
import {
    initTrackerEditor
} from './src/systems/ui/trackerEditor.js';
import {
    initChapterCheckpointUI,
    injectCheckpointButton,
    updateAllCheckpointIndicators,
    cleanupCheckpointUI
} from './src/systems/ui/checkpointUI.js';
import { restoreCheckpointOnLoad } from './src/systems/features/chapterCheckpoint.js';
import {
    togglePlotButtons,
    updateCollapseToggleIcon,
    setupCollapseToggle,
    updatePanelVisibility,
    updateSectionVisibility,
    applyPanelPosition,
    updateGenerationModeUI
} from './src/systems/ui/layout.js';
import {
    setupMobileToggle,
    constrainFabToViewport,
    setupMobileTabs,
    removeMobileTabs,
    setupMobileKeyboardHandling,
    setupContentEditableScrolling,
    updateMobileTabLabels
} from './src/systems/ui/mobile.js';
import {
    setupDesktopTabs,
    removeDesktopTabs
} from './src/systems/ui/desktop.js';

// Feature modules
import { setupPlotButtons, sendPlotProgression } from './src/systems/features/plotProgression.js';
import { setupClassicStatsButtons } from './src/systems/features/classicStats.js';
import { ensureHtmlCleaningRegex, detectConflictingRegexScripts } from './src/systems/features/htmlCleaning.js';
import { setupMemoryRecollectionButton, updateMemoryRecollectionButton } from './src/systems/features/memoryRecollection.js';
import { initLorebookLimiter } from './src/systems/features/lorebookLimiter.js';
import { DEFAULT_HTML_PROMPT } from './src/systems/generation/promptBuilder.js';
import { openEncounterModal } from './src/systems/ui/encounterUI.js';

// Integration modules
import {
    commitTrackerData,
    onMessageSent,
    onMessageReceived,
    onCharacterChanged,
    onMessageSwiped,
    updatePersonaAvatar,
    clearExtensionPrompts,
    onGenerationEnded
} from './src/systems/integration/sillytavern.js';

// Old state variable declarations removed - now imported from core modules
// (extensionSettings, lastGeneratedData, committedTrackerData, etc. are now in src/core/state.js)

// Utility functions removed - now imported from src/utils/avatars.js
// (getSafeThumbnailUrl)

// Persistence functions removed - now imported from src/core/persistence.js
// (loadSettings, saveSettings, saveChatData, loadChatData, updateMessageSwipeData)

// Theme functions removed - now imported from src/systems/ui/theme.js
// (applyTheme, applyCustomTheme, toggleCustomColors, toggleAnimations,
//  updateSettingsPopupTheme, applyCustomThemeToSettingsPopup)

// Layout functions removed - now imported from src/systems/ui/layout.js
// (togglePlotButtons, updateCollapseToggleIcon, setupCollapseToggle,
//  updatePanelVisibility, updateSectionVisibility, applyPanelPosition)
// Note: closeMobilePanelWithAnimation is only used internally by mobile.js

// Mobile UI functions removed - now imported from src/systems/ui/mobile.js
// (setupMobileToggle, constrainFabToViewport, setupMobileTabs, removeMobileTabs,
//  setupMobileKeyboardHandling, setupContentEditableScrolling)

/**
 * Updates UI elements that are dynamically generated and not covered by data-i18n-key.
 */
function updateDynamicLabels() {
    // Update "Refresh RPG Info" button, but only if it's not disabled
    const refreshBtn = document.getElementById('rpg-manual-update');
    if (refreshBtn && !refreshBtn.disabled) {
        const refreshText = i18n.getTranslation('template.mainPanel.refreshRpgInfo') || 'Refresh RPG Info';
        refreshBtn.innerHTML = `<i class="fa-solid fa-sync"></i> ${refreshText}`;
    }

    // Update "Last Roll" label
    updateDiceDisplay();

    // Update mobile tab labels
    updateMobileTabLabels();
}

/**
 * Adds the extension settings to the Extensions tab.
 */
async function addExtensionSettings() {
    // Load the HTML template for the settings
    const settingsHtml = await renderExtensionTemplateAsync(extensionName, 'settings');
    $('#extensions_settings2').append(settingsHtml);

    // Set up the enable/disable toggle
    $('#rpg-extension-enabled').prop('checked', extensionSettings.enabled).on('change', async function() {
        const wasEnabled = extensionSettings.enabled;
        extensionSettings.enabled = $(this).prop('checked');
        saveSettings();

        if (!extensionSettings.enabled && wasEnabled) {
            // Disabling extension - remove UI elements
            clearExtensionPrompts();
            updateChatThoughts(); // Remove thought bubbles
            cleanupCheckpointUI(); // Remove checkpoint buttons and indicators

            // Remove panel and toggle buttons
            $('#rpg-companion-panel').remove();
            $('#rpg-mobile-toggle').remove();
            $('#rpg-collapse-toggle').remove();
            $('#rpg-debug-toggle').remove();
            $('#rpg-debug-panel').remove();
            $('#rpg-plot-buttons').remove(); // Remove plot buttons
        } else if (extensionSettings.enabled && !wasEnabled) {
            // Enabling extension - initialize UI
            await initUI();
            loadChatData(); // Load chat data for current chat
            updateChatThoughts(); // Create thought bubbles if data exists
            injectCheckpointButton(); // Re-add checkpoint buttons
            updateAllCheckpointIndicators(); // Update button states
        }

        // Update Memory Recollection button visibility
        updateMemoryRecollectionButton();
    });

    // Set up language selector
    const langSelect = $('#rpg-companion-language-select');
    if (langSelect.length) {
        langSelect.val(i18n.currentLanguage);
        langSelect.on('change', async function() {
            const selectedLanguage = $(this).val();
            await i18n.setLanguage(selectedLanguage);
            // We need to re-apply translations to the settings panel specifically
            i18n.applyTranslations(document.getElementById('extensions_settings2'));
        });
    }
}

/**
 * Initializes the UI for the extension.
 */
async function initUI() {
    // Initialize i18n
    await i18n.init();

    // Only initialize UI if extension is enabled
    if (!extensionSettings.enabled) {
        console.log('[RPG Companion] Extension disabled - skipping UI initialization');
        return;
    }

    // Load the HTML template using SillyTavern's template system
    const templateHtml = await renderExtensionTemplateAsync(extensionName, 'template');

    // Append panel to body - positioning handled by CSS
    $('body').append(templateHtml);

    // Add mobile toggle button (FAB - Floating Action Button)
    const mobileToggleHtml = `
        <button id="rpg-mobile-toggle" class="rpg-mobile-toggle" title="Toggle RPG Panel">
            <i class="fa-solid fa-dice-d20"></i>
        </button>
    `;
    $('body').append(mobileToggleHtml);

    // Hide mobile toggle on desktop viewport (> 1000px)
    if (window.innerWidth > 1000) {
        $('#rpg-mobile-toggle').hide();
    }

    // Cache UI elements using state setters
    setPanelContainer($('#rpg-companion-panel'));
    setUserStatsContainer($('#rpg-user-stats'));
    setInfoBoxContainer($('#rpg-info-box'));
    setThoughtsContainer($('#rpg-thoughts'));
    setInventoryContainer($('#rpg-inventory'));
    setQuestsContainer($('#rpg-quests'));

    // Re-apply translations to the entire body to catch all new elements from the template
    i18n.applyTranslations(document.body);

    // Set up event listeners (enable/disable is handled in Extensions tab)
    $('#rpg-toggle-auto-update').on('change', function() {
        extensionSettings.autoUpdate = $(this).prop('checked');
        saveSettings();
    });

    $('#rpg-position-select').on('change', function() {
        extensionSettings.panelPosition = String($(this).val());
        saveSettings();
        applyPanelPosition();
        // Recreate thought bubbles to update their position
        updateChatThoughts();
    });

    $('#rpg-update-depth').on('change', function() {
        const value = $(this).val();
        extensionSettings.updateDepth = parseInt(String(value));
        saveSettings();
    });

    $('#rpg-memory-messages').on('change', function() {
        const value = $(this).val();
        extensionSettings.memoryMessagesToProcess = parseInt(String(value));
        saveSettings();
    });

    $('#rpg-generation-mode').on('change', function() {
        extensionSettings.generationMode = String($(this).val());
        saveSettings();
        updateGenerationModeUI();
    });

    $('#rpg-use-separate-preset').on('change', function() {
        extensionSettings.useSeparatePreset = $(this).prop('checked');
        saveSettings();
    });

    $('#rpg-toggle-user-stats').on('change', function() {
        extensionSettings.showUserStats = $(this).prop('checked');
        saveSettings();
        updateSectionVisibility();
    });

    $('#rpg-toggle-info-box').on('change', function() {
        extensionSettings.showInfoBox = $(this).prop('checked');
        saveSettings();
        updateSectionVisibility();
    });

    $('#rpg-toggle-thoughts').on('change', function() {
        extensionSettings.showCharacterThoughts = $(this).prop('checked');
        saveSettings();
        updateSectionVisibility();
    });

    $('#rpg-toggle-narrator-mode').on('change', function() {
        extensionSettings.narratorMode = $(this).prop('checked');
        saveSettings();
    });

    $('#rpg-toggle-inventory').on('change', function() {
        extensionSettings.showInventory = $(this).prop('checked');
        saveSettings();
        updateSectionVisibility();
    });

    $('#rpg-toggle-quests').on('change', function() {
        extensionSettings.showQuests = $(this).prop('checked');
        saveSettings();
        updateSectionVisibility();
    });

    $('#rpg-toggle-thoughts-in-chat').on('change', function() {
        extensionSettings.showThoughtsInChat = $(this).prop('checked');
        // console.log('[RPG Companion] Toggle showThoughtsInChat changed to:', extensionSettings.showThoughtsInChat);
        saveSettings();
        updateChatThoughts();
    });

    $('#rpg-toggle-always-show-bubble').on('change', function() {
        extensionSettings.alwaysShowThoughtBubble = $(this).prop('checked');
        saveSettings();
        // Force immediate save to ensure setting is persisted before any other code runs
        const context = getContext();
        const extension_settings = context.extension_settings || context.extensionSettings;
        extension_settings[extensionName] = extensionSettings;
        // Re-render thoughts to apply the setting
        updateChatThoughts();
    });

    $('#rpg-toggle-html-prompt').on('change', function() {
        extensionSettings.enableHtmlPrompt = $(this).prop('checked');
        // console.log('[RPG Companion] Toggle enableHtmlPrompt changed to:', extensionSettings.enableHtmlPrompt);
        saveSettings();
    });

    $('#rpg-custom-html-prompt').on('input', function() {
        extensionSettings.customHtmlPrompt = $(this).val().trim();
        saveSettings();
    });

    $('#rpg-restore-default-html-prompt').on('click', function() {
        extensionSettings.customHtmlPrompt = '';
        $('#rpg-custom-html-prompt').val('');
        saveSettings();
        toastr.success('HTML prompt restored to default');
    });

    $('#rpg-skip-guided-mode').on('change', function() {
        extensionSettings.skipInjectionsForGuided = String($(this).val());
        saveSettings();
    });

    $('#rpg-save-tracker-history').on('change', function() {
        extensionSettings.saveTrackerHistory = $(this).prop('checked');
        saveSettings();
    });

    $('#rpg-toggle-plot-buttons').on('change', function() {
        extensionSettings.enablePlotButtons = $(this).prop('checked');
        // console.log('[RPG Companion] Toggle enablePlotButtons changed to:', extensionSettings.enablePlotButtons);
        saveSettings();
        togglePlotButtons();
    });

    $('#rpg-toggle-encounters').on('change', function() {
        if (!extensionSettings.encounterSettings) {
            extensionSettings.encounterSettings = { enabled: true, historyDepth: 8, autoSaveLogs: true };
        }
        extensionSettings.encounterSettings.enabled = $(this).prop('checked');
        saveSettings();
        togglePlotButtons(); // This also controls encounter button visibility
    });

    $('#rpg-encounter-history-depth').on('change', function() {
        if (!extensionSettings.encounterSettings) {
            extensionSettings.encounterSettings = { enabled: true, historyDepth: 8, autoSaveLogs: true };
        }
        const value = $(this).val();
        extensionSettings.encounterSettings.historyDepth = parseInt(String(value));
        saveSettings();
    });

    $('#rpg-toggle-autosave-logs').on('change', function() {
        if (!extensionSettings.encounterSettings) {
            extensionSettings.encounterSettings = { enabled: true, historyDepth: 8, autoSaveLogs: true };
        }
        extensionSettings.encounterSettings.autoSaveLogs = $(this).prop('checked');
        saveSettings();
    });

    // Combat narrative style settings
    $('#rpg-combat-tense').on('change', function() {
        if (!extensionSettings.encounterSettings) {
            extensionSettings.encounterSettings = {};
        }
        if (!extensionSettings.encounterSettings.combatNarrative) {
            extensionSettings.encounterSettings.combatNarrative = {};
        }
        extensionSettings.encounterSettings.combatNarrative.tense = $(this).val();
        saveSettings();
    });

    $('#rpg-combat-person').on('change', function() {
        if (!extensionSettings.encounterSettings) {
            extensionSettings.encounterSettings = {};
        }
        if (!extensionSettings.encounterSettings.combatNarrative) {
            extensionSettings.encounterSettings.combatNarrative = {};
        }
        extensionSettings.encounterSettings.combatNarrative.person = $(this).val();
        saveSettings();
    });

    $('#rpg-combat-narration').on('change', function() {
        if (!extensionSettings.encounterSettings) {
            extensionSettings.encounterSettings = {};
        }
        if (!extensionSettings.encounterSettings.combatNarrative) {
            extensionSettings.encounterSettings.combatNarrative = {};
        }
        extensionSettings.encounterSettings.combatNarrative.narration = $(this).val();
        saveSettings();
    });

    $('#rpg-combat-pov').on('change', function() {
        if (!extensionSettings.encounterSettings) {
            extensionSettings.encounterSettings = {};
        }
        if (!extensionSettings.encounterSettings.combatNarrative) {
            extensionSettings.encounterSettings.combatNarrative = {};
        }
        extensionSettings.encounterSettings.combatNarrative.pov = $(this).val();
        saveSettings();
    });

    // Summary narrative style settings
    $('#rpg-summary-tense').on('change', function() {
        if (!extensionSettings.encounterSettings) {
            extensionSettings.encounterSettings = {};
        }
        if (!extensionSettings.encounterSettings.summaryNarrative) {
            extensionSettings.encounterSettings.summaryNarrative = {};
        }
        extensionSettings.encounterSettings.summaryNarrative.tense = $(this).val();
        saveSettings();
    });

    $('#rpg-summary-person').on('change', function() {
        if (!extensionSettings.encounterSettings) {
            extensionSettings.encounterSettings = {};
        }
        if (!extensionSettings.encounterSettings.summaryNarrative) {
            extensionSettings.encounterSettings.summaryNarrative = {};
        }
        extensionSettings.encounterSettings.summaryNarrative.person = $(this).val();
        saveSettings();
    });

    $('#rpg-summary-narration').on('change', function() {
        if (!extensionSettings.encounterSettings) {
            extensionSettings.encounterSettings = {};
        }
        if (!extensionSettings.encounterSettings.summaryNarrative) {
            extensionSettings.encounterSettings.summaryNarrative = {};
        }
        extensionSettings.encounterSettings.summaryNarrative.narration = $(this).val();
        saveSettings();
    });

    $('#rpg-summary-pov').on('change', function() {
        if (!extensionSettings.encounterSettings) {
            extensionSettings.encounterSettings = {};
        }
        if (!extensionSettings.encounterSettings.summaryNarrative) {
            extensionSettings.encounterSettings.summaryNarrative = {};
        }
        extensionSettings.encounterSettings.summaryNarrative.pov = $(this).val();
        saveSettings();
    });

    $('#rpg-toggle-animations').on('change', function() {
        extensionSettings.enableAnimations = $(this).prop('checked');
        saveSettings();
        toggleAnimations();
    });

    // Auto avatar generation settings
    $('#rpg-toggle-auto-avatars').on('change', function() {
        extensionSettings.autoGenerateAvatars = $(this).prop('checked');
        saveSettings();

        // Show/hide avatar options based on toggle
        const $options = $('#rpg-avatar-options');
        if (extensionSettings.autoGenerateAvatars) {
            $options.slideDown(200);
        } else {
            $options.slideUp(200);
        }
    });

    $('#rpg-avatar-llm-instruction').on('input', function() {
        extensionSettings.avatarLLMCustomInstruction = $(this).val().trim();
        saveSettings();
    });

    $('#rpg-toggle-dice-display').on('change', function() {
        extensionSettings.showDiceDisplay = $(this).prop('checked');
        saveSettings();
        updateDiceDisplay();
    });

    $('#rpg-manual-update').on('click', async function() {
        if (!extensionSettings.enabled) {
            // console.log('[RPG Companion] Extension is disabled. Please enable it in the Extensions tab.');
            return;
        }
        await updateRPGData(renderUserStats, renderInfoBox, renderThoughts, renderInventory);
    });

    $('#rpg-stat-bar-color-low').on('change', function() {
        extensionSettings.statBarColorLow = String($(this).val());
        saveSettings();
        renderUserStats(); // Re-render with new colors
    });

    $('#rpg-stat-bar-color-high').on('change', function() {
        extensionSettings.statBarColorHigh = String($(this).val());
        saveSettings();
        renderUserStats(); // Re-render with new colors
    });

    // Theme selection
    $('#rpg-theme-select').on('change', function() {
        extensionSettings.theme = String($(this).val());
        saveSettings();
        applyTheme();
        toggleCustomColors();
        updateSettingsPopupTheme(getSettingsModal()); // Update popup theme instantly
        updateChatThoughts(); // Recreate thought bubbles with new theme
    });

    // Custom color pickers
    $('#rpg-custom-bg').on('change', function() {
        extensionSettings.customColors.bg = String($(this).val());
        saveSettings();
        if (extensionSettings.theme === 'custom') {
            applyCustomTheme();
            updateSettingsPopupTheme(getSettingsModal()); // Update popup theme instantly
            updateChatThoughts(); // Update thought bubbles
        }
    });

    $('#rpg-custom-accent').on('change', function() {
        extensionSettings.customColors.accent = String($(this).val());
        saveSettings();
        if (extensionSettings.theme === 'custom') {
            applyCustomTheme();
            updateSettingsPopupTheme(getSettingsModal()); // Update popup theme instantly
            updateChatThoughts(); // Update thought bubbles
        }
    });

    $('#rpg-custom-text').on('change', function() {
        extensionSettings.customColors.text = String($(this).val());
        saveSettings();
        if (extensionSettings.theme === 'custom') {
            applyCustomTheme();
            updateSettingsPopupTheme(getSettingsModal()); // Update popup theme instantly
            updateChatThoughts(); // Update thought bubbles
        }
    });

    $('#rpg-custom-highlight').on('change', function() {
        extensionSettings.customColors.highlight = String($(this).val());
        saveSettings();
        if (extensionSettings.theme === 'custom') {
            applyCustomTheme();
            updateSettingsPopupTheme(getSettingsModal()); // Update popup theme instantly
            updateChatThoughts(); // Update thought bubbles
        }
    });

    // External API settings event handlers
    $('#rpg-external-base-url').on('change', function() {
        if (!extensionSettings.externalApiSettings) {
            extensionSettings.externalApiSettings = {
                baseUrl: '', apiKey: '', model: '', maxTokens: 8192, temperature: 0.7
            };
        }
        extensionSettings.externalApiSettings.baseUrl = String($(this).val()).trim();
        saveSettings();
    });

    $('#rpg-external-api-key').on('change', function() {
        // Securely store API key in localStorage instead of shared extension settings
        const apiKey = String($(this).val()).trim();
        localStorage.setItem('rpg_companion_external_api_key', apiKey);
        
        // Ensure the externalApiSettings object exists, but don't store the key in it
        if (!extensionSettings.externalApiSettings) {
            extensionSettings.externalApiSettings = {
                baseUrl: '', model: '', maxTokens: 8192, temperature: 0.7
            };
            saveSettings();
        }
    });

    $('#rpg-external-model').on('change', function() {
        if (!extensionSettings.externalApiSettings) {
            extensionSettings.externalApiSettings = {
                baseUrl: '', apiKey: '', model: '', maxTokens: 8192, temperature: 0.7
            };
        }
        extensionSettings.externalApiSettings.model = String($(this).val()).trim();
        saveSettings();
    });

    $('#rpg-external-max-tokens').on('change', function() {
        if (!extensionSettings.externalApiSettings) {
            extensionSettings.externalApiSettings = {
                baseUrl: '', apiKey: '', model: '', maxTokens: 8192, temperature: 0.7
            };
        }
        extensionSettings.externalApiSettings.maxTokens = parseInt(String($(this).val()));
        saveSettings();
    });

    $('#rpg-external-temperature').on('change', function() {
        if (!extensionSettings.externalApiSettings) {
            extensionSettings.externalApiSettings = {
                baseUrl: '', apiKey: '', model: '', maxTokens: 8192, temperature: 0.7
            };
        }
        extensionSettings.externalApiSettings.temperature = parseFloat(String($(this).val()));
        saveSettings();
    });

    $('#rpg-toggle-api-key-visibility').on('click', function() {
        const $input = $('#rpg-external-api-key');
        const type = $input.attr('type') === 'password' ? 'text' : 'password';
        $input.attr('type', type);
        $(this).find('i').toggleClass('fa-eye fa-eye-slash');
    });

    $('#rpg-test-external-api').on('click', async function() {
        const $result = $('#rpg-external-api-test-result');
        const $btn = $(this);
        const originalText = $btn.html();
        
        $btn.html('<i class="fa-solid fa-spinner fa-spin"></i> Testing...').prop('disabled', true);
        $result.hide().removeClass('rpg-success-message rpg-error-message');
        
        try {
            const result = await testExternalAPIConnection();
            
            if (result.success) {
                $result.addClass('rpg-success-message')
                    .html(`<i class="fa-solid fa-check-circle"></i> ${result.message}`)
                    .slideDown();
                toastr.success(result.message);
            } else {
                $result.addClass('rpg-error-message')
                    .html(`<i class="fa-solid fa-exclamation-circle"></i> ${result.message}`)
                    .slideDown();
                toastr.error(result.message);
            }
        } catch (error) {
            $result.addClass('rpg-error-message')
                .html(`<i class="fa-solid fa-exclamation-circle"></i> Error: ${error.message}`)
                .slideDown();
        } finally {
            $btn.html(originalText).prop('disabled', false);
        }
    });

    // Initialize UI state (enable/disable is in Extensions tab)
    $('#rpg-toggle-auto-update').prop('checked', extensionSettings.autoUpdate);
    $('#rpg-position-select').val(extensionSettings.panelPosition);
    $('#rpg-update-depth').val(extensionSettings.updateDepth);
    $('#rpg-memory-messages').val(extensionSettings.memoryMessagesToProcess || 16);
    $('#rpg-use-separate-preset').prop('checked', extensionSettings.useSeparatePreset);
    $('#rpg-toggle-user-stats').prop('checked', extensionSettings.showUserStats);
    $('#rpg-toggle-info-box').prop('checked', extensionSettings.showInfoBox);
    $('#rpg-toggle-thoughts').prop('checked', extensionSettings.showCharacterThoughts);
    $('#rpg-toggle-narrator-mode').prop('checked', extensionSettings.narratorMode);
    $('#rpg-toggle-inventory').prop('checked', extensionSettings.showInventory);
    $('#rpg-toggle-quests').prop('checked', extensionSettings.showQuests);
    $('#rpg-toggle-thoughts-in-chat').prop('checked', extensionSettings.showThoughtsInChat);
    $('#rpg-toggle-always-show-bubble').prop('checked', extensionSettings.alwaysShowThoughtBubble);
    $('#rpg-toggle-html-prompt').prop('checked', extensionSettings.enableHtmlPrompt);

    // Set default HTML prompt as actual text if no custom prompt exists
    $('#rpg-custom-html-prompt').val(extensionSettings.customHtmlPrompt || DEFAULT_HTML_PROMPT);

    $('#rpg-toggle-plot-buttons').prop('checked', extensionSettings.enablePlotButtons);
    $('#rpg-toggle-encounters').prop('checked', extensionSettings.encounterSettings?.enabled ?? true);
    $('#rpg-encounter-history-depth').val(extensionSettings.encounterSettings?.historyDepth ?? 8);
    $('#rpg-toggle-autosave-logs').prop('checked', extensionSettings.encounterSettings?.autoSaveLogs ?? true);

    // Combat narrative style
    $('#rpg-combat-tense').val(extensionSettings.encounterSettings?.combatNarrative?.tense ?? 'present');
    $('#rpg-combat-person').val(extensionSettings.encounterSettings?.combatNarrative?.person ?? 'third');
    $('#rpg-combat-narration').val(extensionSettings.encounterSettings?.combatNarrative?.narration ?? 'omniscient');
    $('#rpg-combat-pov').val(extensionSettings.encounterSettings?.combatNarrative?.pov ?? 'narrator');

    // Summary narrative style
    $('#rpg-summary-tense').val(extensionSettings.encounterSettings?.summaryNarrative?.tense ?? 'past');
    $('#rpg-summary-person').val(extensionSettings.encounterSettings?.summaryNarrative?.person ?? 'third');
    $('#rpg-summary-narration').val(extensionSettings.encounterSettings?.summaryNarrative?.narration ?? 'omniscient');
    $('#rpg-summary-pov').val(extensionSettings.encounterSettings?.summaryNarrative?.pov ?? 'narrator');

    $('#rpg-toggle-animations').prop('checked', extensionSettings.enableAnimations);

    // Initialize avatar options
    $('#rpg-toggle-auto-avatars').prop('checked', extensionSettings.autoGenerateAvatars || false);
    $('#rpg-avatar-llm-instruction').val(extensionSettings.avatarLLMCustomInstruction || '');

    // Initialize avatar options visibility
    if (extensionSettings.autoGenerateAvatars) {
        $('#rpg-avatar-options').show();
    } else {
        $('#rpg-avatar-options').hide();
    }

    $('#rpg-toggle-dice-display').prop('checked', extensionSettings.showDiceDisplay);
    $('#rpg-stat-bar-color-low').val(extensionSettings.statBarColorLow);
    $('#rpg-stat-bar-color-high').val(extensionSettings.statBarColorHigh);
    $('#rpg-theme-select').val(extensionSettings.theme);
    $('#rpg-custom-bg').val(extensionSettings.customColors.bg);
    $('#rpg-custom-accent').val(extensionSettings.customColors.accent);
    $('#rpg-custom-text').val(extensionSettings.customColors.text);
    $('#rpg-custom-highlight').val(extensionSettings.customColors.highlight);

    // Initialize External API settings values
    if (extensionSettings.externalApiSettings) {
        $('#rpg-external-base-url').val(extensionSettings.externalApiSettings.baseUrl || '');
        
        // Load API Key from secure localStorage
        const storedApiKey = localStorage.getItem('rpg_companion_external_api_key') || '';
        $('#rpg-external-api-key').val(storedApiKey);
        
        $('#rpg-external-model').val(extensionSettings.externalApiSettings.model || '');
        $('#rpg-external-max-tokens').val(extensionSettings.externalApiSettings.maxTokens || 8192);
        $('#rpg-external-temperature').val(extensionSettings.externalApiSettings.temperature ?? 0.7);
    }

    $('#rpg-generation-mode').val(extensionSettings.generationMode);
    $('#rpg-skip-guided-mode').val(extensionSettings.skipInjectionsForGuided);
    $('#rpg-save-tracker-history').prop('checked', extensionSettings.saveTrackerHistory);

    updatePanelVisibility();
    updateSectionVisibility();
    updateGenerationModeUI();
    applyTheme();
    applyPanelPosition();
    toggleCustomColors();
    toggleAnimations();

    // Setup mobile toggle button
    setupMobileToggle();

    // Setup desktop tabs (only on desktop viewport)
    if (window.innerWidth > 1000) {
        setupDesktopTabs();
    }

    // Setup collapse/expand toggle button
    setupCollapseToggle();

    // Render initial data if available
    renderUserStats();
    renderInfoBox();
    renderThoughts();
    renderInventory();
    renderQuests();
    updateDiceDisplay();
    setupDiceRoller();
    setupClassicStatsButtons();
    setupSettingsPopup();
    initTrackerEditor();
    addDiceQuickReply();
    setupPlotButtons(sendPlotProgression, openEncounterModal);
    setupMobileKeyboardHandling();
    setupContentEditableScrolling();
    initInventoryEventListeners();

    // Initialize chapter checkpoint UI
    initChapterCheckpointUI();
    injectCheckpointButton();

    // Setup Memory Recollection button in World Info
    setupMemoryRecollectionButton();

    // Initialize Lorebook Limiter
    initLorebookLimiter();
}





// Rendering functions removed - now imported from src/systems/rendering/*
// (renderUserStats, renderInfoBox, renderThoughts, updateInfoBoxField,
//  updateCharacterField, updateChatThoughts, createThoughtPanel)

// Event handlers removed - now imported from src/systems/integration/sillytavern.js
// (commitTrackerData, onMessageSent, onMessageReceived, onCharacterChanged,
//  onMessageSwiped, updatePersonaAvatar, clearExtensionPrompts)

/**
 * Ensures the "RPG Companion Trackers" preset exists in the user's OpenAI Settings.
 * Imports the preset file from the extension folder if it doesn't exist.
 */
async function ensureTrackerPresetExists() {
    try {
        const presetName = 'RPG Companion Trackers';

        // Check if preset already exists by fetching settings
        const checkResponse = await fetch('/api/settings/get', {
            method: 'POST',
            headers: getRequestHeaders()
        });

        if (checkResponse.ok) {
            const settings = await checkResponse.json();
            // openai_setting_names is an array of preset names
            if (settings.openai_setting_names && settings.openai_setting_names.includes(presetName)) {
                console.log(`[RPG Companion] Preset "${presetName}" already exists`);
                return;
            }
        }

        // Preset doesn't exist - import it from extension folder
        console.log(`[RPG Companion] Importing preset "${presetName}"...`);

        // Load preset from extension folder
        const extensionPresetPath = `${extensionFolderPath}/${presetName}.json`;
        const presetResponse = await fetch(`/${extensionPresetPath}`);

        if (!presetResponse.ok) {
            console.warn(`[RPG Companion] Could not load preset template from ${extensionPresetPath}`);
            return;
        }

        const presetData = await presetResponse.json();

        // Save preset to user's OpenAI Settings folder using SillyTavern's API
        const saveResponse = await fetch('/api/presets/save', {
            method: 'POST',
            headers: getRequestHeaders(),
            body: JSON.stringify({
                apiId: 'openai',
                name: presetName,
                preset: presetData
            })
        });

        if (saveResponse.ok) {
            console.log(`[RPG Companion] ✅ Successfully imported preset "${presetName}"`);
            toastr.success(
                `The "RPG Companion Trackers" preset has been imported to your OpenAI Settings.`,
                'RPG Companion',
                { timeOut: 5000 }
            );
        } else {
            console.warn(`[RPG Companion] Failed to save preset: ${saveResponse.statusText}`);
        }
    } catch (error) {
        console.error('[RPG Companion] Error importing tracker preset:', error);
        // Non-critical - users can manually import if needed
    }
}

/**
 * Main initialization function.
 */
jQuery(async () => {
    try {
        console.log('[RPG Companion] Starting initialization...');

        // Load settings with validation
        try {
            loadSettings();
        } catch (error) {
            console.error('[RPG Companion] Settings load failed, continuing with defaults:', error);
        }

        // Initialize i18n early for the settings panel
        await i18n.init();

        // Set up a central listener for language changes to update dynamic UI parts
        i18n.addEventListener('languageChanged', updateDynamicLabels);

        // Add extension settings to Extensions tab
        try {
            await addExtensionSettings();
        } catch (error) {
            console.error('[RPG Companion] Failed to add extension settings tab:', error);
            // Don't throw - extension can still work without settings tab
        }

        // Initialize UI
        try {
            await initUI();
        } catch (error) {
            console.error('[RPG Companion] UI initialization failed:', error);
            throw error; // This is critical - can't continue without UI
        }

        // Load chat-specific data for current chat
        try {
            loadChatData();
        } catch (error) {
            console.error('[RPG Companion] Chat data load failed, using defaults:', error);
        }

        // Import the HTML cleaning regex if needed
        try {
            await ensureHtmlCleaningRegex(st_extension_settings, saveSettingsDebounced);
        } catch (error) {
            console.error('[RPG Companion] HTML regex import failed:', error);
            // Non-critical - continue without it
        }

        // Import the RPG Companion Trackers preset if needed
        try {
            await ensureTrackerPresetExists();
        } catch (error) {
            console.error('[RPG Companion] Preset import failed:', error);
            // Non-critical - users can manually import if needed
        }

        // Detect conflicting regex scripts from old manual formatters
        try {
            const conflicts = detectConflictingRegexScripts(st_extension_settings);
            if (conflicts.length > 0) {
                console.log('[RPG Companion] ⚠️ Detected old manual formatting regex scripts that may conflict:');
                conflicts.forEach(name => console.log(`  - ${name}`));
                console.log('[RPG Companion] Consider disabling these regexes as the extension now handles formatting automatically.');

                // Show user-friendly warning (non-blocking)
                // toastr.warning(
                //     `Found ${conflicts.length} old RPG formatting regex script(s). These may conflict with the extension. Check console for details.`,
                //     'RPG Companion Warning',
                //     { timeOut: 8000 }
                // );
            }
        } catch (error) {
            console.error('[RPG Companion] Conflict detection failed:', error);
            // Non-critical - continue anyway
        }

        // Register all event listeners
        try {
            registerAllEvents({
                [event_types.MESSAGE_SENT]: onMessageSent,
                [event_types.GENERATION_STARTED]: onGenerationStarted,
                [event_types.MESSAGE_RECEIVED]: onMessageReceived,
                [event_types.GENERATION_STOPPED]: onGenerationEnded,
                [event_types.GENERATION_ENDED]: onGenerationEnded,
                [event_types.CHAT_CHANGED]: [onCharacterChanged, updatePersonaAvatar, restoreCheckpointOnLoad, clearSessionAvatarPrompts],
                [event_types.MESSAGE_SWIPED]: onMessageSwiped,
                [event_types.USER_MESSAGE_RENDERED]: updatePersonaAvatar,
                [event_types.SETTINGS_UPDATED]: updatePersonaAvatar
            });
        } catch (error) {
            console.error('[RPG Companion] Event registration failed:', error);
            throw error; // This is critical - can't continue without events
        }

        // Restore checkpoint state if one exists
        await restoreCheckpointOnLoad();

        console.log('[RPG Companion] ✅ Extension loaded successfully');
    } catch (error) {
        console.error('[RPG Companion] ❌ Critical initialization failure:', error);
        console.error('[RPG Companion] Error details:', error.message, error.stack);

        // Show user-friendly error message
        toastr.error(
            'RPG Companion failed to initialize. Check console for details. Please try refreshing the page or resetting extension settings.',
            'RPG Companion Error',
            { timeOut: 10000 }
        );
    }
});