++

2026-02-05 10:18:53 +08:00
parent 717f8c1747
commit e8a4b2d318
364 changed files with 453102 additions and 0 deletions
@@ -0,0 +1,465 @@
+/**
+ * Inspiration Master - Frontend Logic
+ */
+
+// Global state for inspiration feature
+let inspirationState = {
+    isLoggedIn: false,
+    llmConfig: {
+        baseUrl: "https://api.moonshot.cn/v1",
+        model: "kimi-k2-turbo-preview",
+        apiKey: ""
+    },
+    options: null,
+    selectedDataset: null
+};
+
+document.addEventListener('DOMContentLoaded', function() {
+    const inspirationBtn = document.getElementById('inspirationBtn');
+    if (inspirationBtn) {
+        inspirationBtn.addEventListener('click', openInspirationModal);
+    }
+
+    // Initialize event listeners for the modal
+    document.getElementById('inspire-region').addEventListener('change', updateUniversesAndDelays);
+    document.getElementById('inspire-search-dataset').addEventListener('click', searchDatasets);
+    document.getElementById('inspire-generate').addEventListener('click', generateAlphaTemplates);
+    document.getElementById('inspire-download').addEventListener('click', downloadInspirationResult);
+    document.getElementById('inspire-new-task').addEventListener('click', resetInspirationTask);
+    document.getElementById('inspire-close').addEventListener('click', closeInspirationModal);
+    
+    const testBtn = document.getElementById('inspire-test-llm');
+    if (testBtn) {
+        testBtn.addEventListener('click', testLLMConnection);
+    }
+    
+    // Initially disable generate button until tested
+    const genBtn = document.getElementById('inspire-generate');
+    if (genBtn) {
+        genBtn.disabled = true;
+        genBtn.title = "Please test LLM connection first";
+    }
+    
+    // Initially disable new task button
+    const newTaskBtn = document.getElementById('inspire-new-task');
+    if (newTaskBtn) {
+        newTaskBtn.disabled = true;
+        newTaskBtn.style.opacity = '0.5';
+        newTaskBtn.style.cursor = 'not-allowed';
+    }
+    
+    // Check login status periodically or on load to update button state
+    checkLoginAndUpdateButton();
+});
+
+function getHeaders() {
+    const headers = {'Content-Type': 'application/json'};
+    const sessionId = localStorage.getItem('brain_session_id');
+    if (sessionId) {
+        headers['Session-ID'] = sessionId;
+    }
+    return headers;
+}
+
+function checkLoginAndUpdateButton() {
+    fetch('/api/check_login', { headers: getHeaders() })
+        .then(response => response.json())
+        .then(data => {
+            const btn = document.getElementById('inspirationBtn');
+            if (data.logged_in) {
+                inspirationState.isLoggedIn = true;
+                if (btn) {
+                    btn.style.opacity = '1';
+                    btn.style.cursor = 'pointer';
+                    // Add a visual indicator of logged-in state if desired, e.g., change icon color
+                }
+            } else {
+                inspirationState.isLoggedIn = false;
+                if (btn) {
+                    // Keep it clickable but maybe visually distinct
+                    btn.style.opacity = '1'; 
+                    btn.style.cursor = 'pointer';
+                }
+            }
+            // Always ensure it's enabled
+            if (btn) btn.disabled = false;
+        })
+        .catch(err => console.error("Error checking login status:", err));
+}
+
+// Expose this function globally so other scripts (like brain.js) can call it after login
+window.updateInspirationButtonState = checkLoginAndUpdateButton;
+
+function openInspirationModal() {
+    if (!inspirationState.isLoggedIn) {
+        // Double check
+        fetch('/api/check_login', { headers: getHeaders() })
+            .then(response => response.json())
+            .then(data => {
+                if (data.logged_in) {
+                    inspirationState.isLoggedIn = true;
+                    document.getElementById('inspirationModal').style.display = 'block';
+                    loadInspirationOptions();
+                } else {
+                    // Trigger Brain Login Modal
+                    if (typeof openBrainLoginModal === 'function') {
+                        openBrainLoginModal();
+                    } else {
+                        alert("请先登录 BRAIN。");
+                    }
+                }
+            });
+        return;
+    }
+    document.getElementById('inspirationModal').style.display = 'block';
+    loadInspirationOptions();
+}
+
+function closeInspirationModal() {
+    document.getElementById('inspirationModal').style.display = 'none';
+}
+
+function loadInspirationOptions() {
+    if (inspirationState.options) return; // Already loaded
+
+    fetch('/api/inspiration/options', { headers: getHeaders() })
+        .then(res => res.json())
+        .then(data => {
+            inspirationState.options = data;
+            populateRegionDropdown(data);
+        })
+        .catch(err => console.error("Failed to load options:", err));
+}
+
+function populateRegionDropdown(data) {
+    const regionSelect = document.getElementById('inspire-region');
+    regionSelect.innerHTML = '<option value="">Select Region</option>';
+    
+    // Assuming data structure matches what we get from ace_lib
+    // Structure: { "EQUITY": { "USA": { ... }, "CHN": { ... } } }
+    // We'll focus on EQUITY for now or iterate all
+    
+    let regions = new Set();
+    if (data.EQUITY) {
+        Object.keys(data.EQUITY).forEach(r => regions.add(r));
+    }
+    
+    regions.forEach(r => {
+        const option = document.createElement('option');
+        option.value = r;
+        option.textContent = r;
+        regionSelect.appendChild(option);
+    });
+}
+
+function updateUniversesAndDelays() {
+    const region = document.getElementById('inspire-region').value;
+    const universeSelect = document.getElementById('inspire-universe');
+    const delaySelect = document.getElementById('inspire-delay');
+    
+    universeSelect.innerHTML = '<option value="">Select Universe</option>';
+    delaySelect.innerHTML = '<option value="">Select Delay</option>';
+    
+    if (!region || !inspirationState.options || !inspirationState.options.EQUITY || !inspirationState.options.EQUITY[region]) return;
+    
+    const data = inspirationState.options.EQUITY[region];
+    
+    data.universes.forEach(u => {
+        const option = document.createElement('option');
+        option.value = u;
+        option.textContent = u;
+        universeSelect.appendChild(option);
+    });
+    
+    data.delays.forEach(d => {
+        const option = document.createElement('option');
+        option.value = d;
+        option.textContent = d;
+        delaySelect.appendChild(option);
+    });
+}
+
+function searchDatasets() {
+    const region = document.getElementById('inspire-region').value;
+    const delay = document.getElementById('inspire-delay').value;
+    const universe = document.getElementById('inspire-universe').value;
+    const search = document.getElementById('inspire-dataset-search').value;
+    
+    if (!region || !delay || !universe) {
+        alert("请先选择区域、延迟和股票池。");
+        return;
+    }
+    
+    const resultsDiv = document.getElementById('inspire-dataset-results');
+    resultsDiv.innerHTML = '正在加载数据集...';
+    
+    fetch('/api/inspiration/datasets', {
+        method: 'POST',
+        headers: getHeaders(),
+        body: JSON.stringify({ region, delay, universe, search })
+    })
+    .then(res => res.json())
+    .then(data => {
+        displayDatasetResults(data);
+    })
+    .catch(err => {
+        resultsDiv.innerHTML = '加载数据集出错: ' + err;
+    });
+}
+
+function displayDatasetResults(datasets) {
+    const resultsDiv = document.getElementById('inspire-dataset-results');
+    resultsDiv.innerHTML = '';
+    
+    if (datasets.length === 0) {
+        resultsDiv.innerHTML = '未找到数据集。';
+        return;
+    }
+    
+    const table = document.createElement('table');
+    table.className = 'dataset-table'; 
+    table.style.width = '100%'; // Ensure full width
+    table.style.borderCollapse = 'collapse';
+    
+    table.innerHTML = `
+        <thead>
+            <tr style="text-align: left; background: #f1f1f1;">
+                <th style="padding: 8px; border-bottom: 1px solid #ddd;">ID</th>
+                <th style="padding: 8px; border-bottom: 1px solid #ddd;">Name</th>
+                <th style="padding: 8px; border-bottom: 1px solid #ddd;">Category</th>
+            </tr>
+        </thead>
+        <tbody></tbody>
+    `;
+    
+    const tbody = table.querySelector('tbody');
+    
+    datasets.forEach(ds => {
+        const tr = document.createElement('tr');
+        tr.dataset.id = ds.id;
+        tr.style.cursor = 'pointer';
+        tr.style.transition = 'background-color 0.2s';
+        
+        // Fix [object Object] issue for category
+        let category = ds.category;
+        if (typeof category === 'object' && category !== null) {
+            // Try to find a meaningful string representation
+            category = category.name || category.id || JSON.stringify(category);
+        }
+        
+        tr.innerHTML = `
+            <td style="padding: 8px; border-bottom: 1px solid #eee; color: #007bff; font-weight: bold;">${ds.id}</td>
+            <td style="padding: 8px; border-bottom: 1px solid #eee;">${ds.name}</td>
+            <td style="padding: 8px; border-bottom: 1px solid #eee;">${category}</td>
+        `;
+        
+        tr.addEventListener('click', function() {
+            selectDataset(ds.id);
+        });
+        
+        tr.addEventListener('mouseenter', function() {
+            if (inspirationState.selectedDataset !== ds.id) {
+                this.style.backgroundColor = '#f8f9fa';
+            }
+        });
+        
+        tr.addEventListener('mouseleave', function() {
+            if (inspirationState.selectedDataset !== ds.id) {
+                this.style.backgroundColor = '';
+            }
+        });
+
+        tbody.appendChild(tr);
+    });
+    
+    resultsDiv.appendChild(table);
+}
+
+function selectDataset(id) {
+    inspirationState.selectedDataset = id;
+    const display = document.getElementById('inspire-selected-dataset');
+    if (display) {
+        display.textContent = "已选数据集: " + id;
+        display.style.display = 'block';
+    }
+    
+    // Highlight the selected row
+    document.querySelectorAll('.dataset-table tr').forEach(tr => tr.style.backgroundColor = '');
+    const row = document.querySelector(`.dataset-table tr[data-id="${id}"]`);
+    if (row) {
+        row.style.backgroundColor = '#e7f1ff';
+    }
+}
+
+// Removed toggleAccordion as we are moving to a horizontal layout
+
+function generateAlphaTemplates() {
+    if (!inspirationState.selectedDataset) {
+        alert("请先选择一个数据集。");
+        return;
+    }
+    
+    const apiKey = document.getElementById('inspire-llm-key').value;
+    const baseUrl = document.getElementById('inspire-llm-url').value;
+    const model = document.getElementById('inspire-llm-model').value;
+    
+    if (!apiKey) {
+        alert("请输入 LLM API Key。");
+        return;
+    }
+    
+    const region = document.getElementById('inspire-region').value;
+    const delay = document.getElementById('inspire-delay').value;
+    const universe = document.getElementById('inspire-universe').value;
+    
+    const outputDiv = document.getElementById('inspire-output');
+    outputDiv.innerHTML = '正在生成模板... 这可能需要几分钟...';
+    
+    fetch('/api/inspiration/generate', {
+        method: 'POST',
+        headers: getHeaders(),
+        body: JSON.stringify({
+            apiKey, baseUrl, model,
+            region, delay, universe,
+            datasetId: inspirationState.selectedDataset
+        })
+    })
+    .then(res => res.json())
+    .then(data => {
+        if (data.error) {
+            outputDiv.innerHTML = '错误: ' + data.error;
+        } else {
+            // Render Markdown
+            // Assuming marked.js or similar is available, or just text for now
+            // If you have a markdown renderer, use it.
+            // For now, simple text or basic HTML replacement
+            outputDiv.innerHTML = formatMarkdown(data.result);
+            inspirationState.lastResult = data.result;
+            const dlBtn = document.getElementById('inspire-download');
+            if (dlBtn) dlBtn.style.display = 'inline-block';
+            
+            const newTaskBtn = document.getElementById('inspire-new-task');
+            if (newTaskBtn) {
+                newTaskBtn.disabled = false;
+                newTaskBtn.style.opacity = '1';
+                newTaskBtn.style.cursor = 'pointer';
+            }
+        }
+    })
+    .catch(err => {
+        outputDiv.innerHTML = '生成模板出错: ' + err;
+    });
+}
+
+function formatMarkdown(text) {
+    if (typeof marked !== 'undefined') {
+        // Split text by code blocks (triple backticks or single backticks)
+        // We want to escape < and > in normal text, but NOT in code blocks
+        const parts = text.split(/(```[\s\S]*?```|`[^`]*`)/g);
+        
+        const escapedParts = parts.map(part => {
+            // If it starts with backtick, it's a code block -> return as is
+            if (part.startsWith('`')) {
+                return part;
+            }
+            // Otherwise, escape < and >
+            return part.replace(/</g, '&lt;').replace(/>/g, '&gt;');
+        });
+        
+        return marked.parse(escapedParts.join(''));
+    }
+    
+    // Fallback simple formatter
+    // Here we DO need to escape because we are building HTML manually
+    const escapedText = text.replace(/</g, '&lt;').replace(/>/g, '&gt;');
+    let html = escapedText
+        .replace(/```([\s\S]*?)```/g, '<pre><code>$1</code></pre>')
+        .replace(/\n/g, '<br>');
+    return html;
+}
+
+function testLLMConnection() {
+    const apiKey = document.getElementById('inspire-llm-key').value;
+    const baseUrl = document.getElementById('inspire-llm-url').value;
+    const model = document.getElementById('inspire-llm-model').value;
+    const testBtn = document.getElementById('inspire-test-llm');
+    const generateBtn = document.getElementById('inspire-generate');
+    
+    if (!apiKey) {
+        alert("请先输入 API Key。");
+        return;
+    }
+    
+    testBtn.textContent = "测试中...";
+    testBtn.disabled = true;
+    
+    fetch('/api/inspiration/test_llm', {
+        method: 'POST',
+        headers: getHeaders(),
+        body: JSON.stringify({ apiKey, baseUrl, model })
+    })
+    .then(res => res.json())
+    .then(data => {
+        testBtn.disabled = false;
+        if (data.success) {
+            testBtn.textContent = "成功";
+            testBtn.className = "btn btn-success";
+            generateBtn.disabled = false;
+            setTimeout(() => { testBtn.textContent = "测试连接"; testBtn.className = "btn btn-secondary"; }, 3000);
+        } else {
+            testBtn.textContent = "失败";
+            testBtn.className = "btn btn-danger";
+            alert("连接失败: " + data.error);
+            generateBtn.disabled = true;
+            setTimeout(() => { testBtn.textContent = "测试连接"; testBtn.className = "btn btn-secondary"; }, 3000);
+        }
+    })
+    .catch(err => {
+        testBtn.disabled = false;
+        testBtn.textContent = "错误";
+        alert("错误: " + err);
+    });
+}
+
+function downloadInspirationResult() {
+    if (!inspirationState.lastResult) return;
+    
+    const blob = new Blob([inspirationState.lastResult], { type: 'text/markdown' });
+    const url = URL.createObjectURL(blob);
+    const a = document.createElement('a');
+    a.href = url;
+    a.download = `alpha_inspiration_${inspirationState.selectedDataset}.md`;
+    document.body.appendChild(a);
+    a.click();
+    document.body.removeChild(a);
+    URL.revokeObjectURL(url);
+}
+
+function resetInspirationTask() {
+    inspirationState.selectedDataset = null;
+    const display = document.getElementById('inspire-selected-dataset');
+    if (display) {
+        display.textContent = "";
+        display.style.display = 'none';
+    }
+    document.getElementById('inspire-output').innerHTML = "";
+    document.getElementById('inspire-dataset-results').innerHTML = "";
+    
+    const dlBtn = document.getElementById('inspire-download');
+    if (dlBtn) dlBtn.style.display = 'none';
+    
+    const newTaskBtn = document.getElementById('inspire-new-task');
+    if (newTaskBtn) {
+        newTaskBtn.disabled = true;
+        newTaskBtn.style.opacity = '0.5';
+        newTaskBtn.style.cursor = 'not-allowed';
+    }
+    // Keep LLM config and Region/Universe selections
+}
+
+function toggleInspireColumn(colId) {
+    const col = document.getElementById(colId);
+    if (col) {
+        col.classList.toggle('collapsed');
+    }
+}