aiData/static/js/query.js

const { createApp, ref, computed, watch, nextTick, onMounted } = Vue;

createApp({
    setup() {
        const apiBase = ref(window.location.origin || "http://localhost:8000");
        
        // Query State
        const userQuery = ref('');
        const queryLoading = ref(false);
        const queryResult = ref('');
        const eventSource = ref(null);
        const rowsAcc = ref([]);
        const bufferRow = ref(null);
        const examples = [
            "查询12月份充电量TOP 10场站的充电情况",
            "查询净月商贸城站12月的充电量",
            "查询12月企业充电量排名的TOP 10",
            "查询所有场站的近3个月的充电情况，找出变化最大的前10名"
        ];

        // Helpers
        const isPlainObject = (obj) => Object.prototype.toString.call(obj) === '[object Object]';
        const colSynonyms = {
            name: ['name', 'station', 'station_name', 'site', '场站', '场站名称', '站点', '站名'],
            total: ['total', 'total_kwh', 'sum_kwh', 'energy_total', '总电量', '总充电量', '总度', '总度数'],
            peak: ['peak', 'peak_kwh', 'energy_peak', '峰', '峰时电量'],
            flat: ['flat', 'flat_kwh', 'energy_flat', '平', '平时电量'],
            valley: ['valley', 'valley_kwh', 'energy_valley', '谷', '谷时电量'],
        };
        const pickKey = (obj, keys) => keys.find(k => Object.prototype.hasOwnProperty.call(obj, k));
        const pickVal = (obj, keys) => {
            const k = pickKey(obj, keys);
            return k ? obj[k] : undefined;
        };
        
        const formatNumber = (n) => {
            if (n === null || n === undefined || n === '') return '';
            const v = parseFloat(String(n).replace(/,/g, ''));
            if (isNaN(v)) return String(n);
            return v.toLocaleString('zh-CN', { maximumFractionDigits: 2 });
        };

        const pushRow = (row) => {
            if (!row || !row.name) return;
            rowsAcc.value.push(row);
            const r = row;
            const line = `| ${r.name ?? ''} | ${formatNumber(r.total)} | ${formatNumber(r.peak)} | ${formatNumber(r.flat)} | ${formatNumber(r.valley)} |`;
            
            if (rowsAcc.value.length === 1) {
                 const header = `| 场站名称 | 总电量 | 峰 | 平 | 谷 |\n| --- | ---: | ---: | ---: | ---: |`;
                 if (queryResult.value && !queryResult.value.endsWith('\n')) {
                     queryResult.value += '\n\n';
                 } else if (queryResult.value) {
                     queryResult.value += '\n';
                 }
                 queryResult.value += header + '\n';
            }
            queryResult.value += line + '\n';
        };

        const handleDegreeSearch = () => {
            if (!userQuery.value.trim()) {
                if (typeof ElementPlus !== 'undefined') ElementPlus.ElMessage.warning('请输入您的问题');
                else alert('请输入您的问题');
                return;
            }

            if (queryLoading.value) {
                if (eventSource.value) {
                    eventSource.value.close();
                }
            }

            queryLoading.value = true;
            queryResult.value = '';
            rowsAcc.value = [];
            bufferRow.value = null;
            
            const url = `${apiBase.value}/degree/chat?q=${encodeURIComponent(userQuery.value)}`;

            try {
                const es = new EventSource(url);
                eventSource.value = es;

                es.onmessage = (event) => {
                    const eventData = event.data.trim();
                    if (eventData === '[DONE]') {
                        queryLoading.value = false;
                        es.close();
                        renderCharts();
                        return;
                    }

                    try {
                        const data = JSON.parse(event.data);
                        
                        let chunk = '';
                        if (data.content) chunk = data.content;
                        else if (data.markdown) chunk = data.markdown;
                        else if (data.chunk) chunk = data.chunk;
                        else if (typeof data === 'string') chunk = data;
                        
                        if (chunk) {
                            queryResult.value += chunk;
                        }
                    } catch (e) {
                        if (eventData && eventData !== '[DONE]') {
                            queryResult.value += event.data;
                        }
                    }
                };

                es.onerror = (err) => {
                    console.error('SSE Error:', err);
                    queryLoading.value = false;
                    es.close();
                    if (!queryResult.value) {
                        queryResult.value = '查询过程出现异常，未能获取到数据';
                    }
                };
            } catch (err) {
                console.error('Failed to create EventSource:', err);
                queryLoading.value = false;
                if (typeof ElementPlus !== 'undefined') ElementPlus.ElMessage.error('无法建立连接');
                else alert('无法建立连接');
            }
        };

        const stopDegreeGeneration = () => {
            if (eventSource.value) {
                eventSource.value.close();
                eventSource.value = null;
            }
            queryLoading.value = false;
        };

        const setExample = (text) => {
            userQuery.value = text;
            handleDegreeSearch();
        };

        const renderCharts = () => {
            if (typeof echarts === 'undefined') return;
            nextTick(() => {
                const containers = document.querySelectorAll('.echarts-container');
                containers.forEach(container => {
                    if (container.getAttribute('data-rendered') === 'true') return;
                    try {
                        const configStr = decodeURIComponent(container.getAttribute('data-config')).trim();
                        if (!configStr.startsWith('{') || !configStr.endsWith('}')) return;
                        const config = JSON.parse(configStr);
                        let chart = echarts.getInstanceByDom(container);
                        if (!chart) {
                            chart = echarts.init(container, 'dark');
                            window.addEventListener('resize', () => chart.resize());
                        }
                        chart.setOption(config);
                        container.setAttribute('data-rendered', 'true');
                        container.style.opacity = '1';
                    } catch (e) {
                        if (queryLoading.value) return; 
                        console.error('ECharts rendering error:', e);
                        container.innerHTML = `<p style="color: #ef4444; padding: 10px;">图表渲染失败: ${e.message}</p>`;
                    }
                });
            });
        };

        // Configure Marked
        if (typeof marked !== 'undefined') {
            const renderer = new marked.Renderer();
            const oldCode = renderer.code.bind(renderer);
            renderer.code = function(code, language) {
                if (language === 'echarts') {
                    const id = 'chart-' + Math.random().toString(36).substr(2, 9);
                    return `<div class="echarts-container" id="${id}" style="width: 100%; height: 400px; margin: 20px 0; background: rgba(30, 41, 59, 0.5); border: 1px solid rgba(148, 163, 184, 0.1); border-radius: 12px; padding: 20px;" data-config="${encodeURIComponent(code)}"></div>`;
                }
                return oldCode(code, language);
            };
            marked.setOptions({
                renderer: renderer,
                gfm: true,
                breaks: true
            });
        }

        // 降级用的简易 Markdown 解析器
        const simpleMarkdown = (text) => {
            if (!text) return '';
            let lines = text.split('\n');
            let html = '';
            let inList = false;
            const parseInline = (str) => {
                return str
                    .replace(/\*\*(.*?)\*\*/g, '<strong>$1</strong>')
                    .replace(/`(.*?)`/g, '<code style="background:rgba(148, 163, 184, 0.1); padding:2px 4px; border-radius:4px;">$1</code>');
            };
            for (let line of lines) {
                let trimmed = line.trim();
                if (!trimmed) {
                    if (inList) { html += '</ul>'; inList = false; }
                    html += '<br>';
                    continue;
                }
                if (trimmed.startsWith('### ')) {
                    if (inList) { html += '</ul>'; inList = false; }
                    html += `<h3>${parseInline(trimmed.substring(4))}</h3>`;
                } else if (trimmed.startsWith('- ') || /^\d+\./.test(trimmed)) {
                    if (!inList) { html += '<ul>'; inList = true; }
                    let content = trimmed.replace(/^(- |\d+\. )/, '');
                    html += `<li>${parseInline(content)}</li>`;
                } else {
                    if (inList) { html += '</ul>'; inList = false; }
                    html += `<p>${parseInline(trimmed)}</p>`;
                }
            }
            if (inList) html += '</ul>';
            return html;
        };

        // 增强的 Markdown & LaTeX 解析器
        const renderMarkdownAndLatex = (text) => {
            if (!text) return '';
            
            try {
                // 1. 处理 LaTeX (简单替换，先处理 $$ 再处理 $)
                let processedText = text;
                
                // 处理块级 LaTeX: $$ ... $$
                processedText = processedText.replace(/\$\$\s*([\s\S]*?)\s*\$\$/g, (match, formula) => {
                    try {
                        if (typeof katex !== 'undefined') {
                            return '<div class="katex-block">' + katex.renderToString(formula, { displayMode: true, throwOnError: false }) + '</div>';
                        }
                        return match;
                    } catch (e) {
                        return match;
                    }
                });
                
                // 处理行内 LaTeX: $ ... $
                processedText = processedText.replace(/\$([^\$\n]+?)\$/g, (match, formula) => {
                    try {
                        if (typeof katex !== 'undefined') {
                            return katex.renderToString(formula, { displayMode: false, throwOnError: false });
                        }
                        return match;
                    } catch (e) {
                        return match;
                    }
                });

                // 2. 使用 marked 解析 Markdown
                if (typeof marked !== 'undefined') {
                    // 直接解析处理后的文本，marked 会处理 Markdown 转义
                    return marked.parse(processedText);
                } else {
                    // 降级使用之前的 simpleMarkdown
                    return simpleMarkdown(processedText);
                }
            } catch (e) {
                console.error('Markdown/LaTeX rendering error:', e);
                return text;
            }
        };

        const renderedResult = computed(() => {
            return renderMarkdownAndLatex(queryResult.value);
        });

        watch(queryResult, () => {
             renderCharts();
        });

        return {
            userQuery,
            queryLoading,
            queryResult,
            examples,
            handleDegreeSearch,
            stopDegreeGeneration,
            setExample,
            renderedResult
        };
    }
}).use(ElementPlus).mount('#app');