huanghai
/
dsProject
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

'commit'

Browse Source
main
HuangHai 4 weeks ago
parent bf25c6139d
commit 2e26be8e47
1 changed files with 70 additions and 93 deletions
Show Stats Download Patch File Download Diff File

163
dsRag/static/ai.html
Unescape View File

@ -84,16 +84,6 @@
margin-right: 5px; margin-right: 5px;
} }
.loader {
border: 4px solid #f3f3f3;
border-top: 4px solid #3498db;
border-radius: 50%;
width: 30px;
height: 30px;
animation: spin 2s linear infinite;
margin: 20px auto;
display: none;
}
@keyframes spin { @keyframes spin {
0% { 0% {
@ -193,8 +183,7 @@
<div class="input-area"> <div class="input-area">
<label for="questionInput"></label><textarea id="questionInput" placeholder="请输入您的问题..."></textarea> <label for="questionInput"></label><textarea id="questionInput" placeholder="请输入您的问题..."></textarea>
<div class="examples">
<div class="examples">
<div class="example-category"> <div class="example-category">
<h3>知识库内问题示例：</h3> <h3>知识库内问题示例：</h3>
<div class="example-list"> <div class="example-list">
@ -239,37 +228,36 @@
<button id="clearBtn" onclick="clearAll()"><span class="icon">🗑️</span>清空</button> <button id="clearBtn" onclick="clearAll()"><span class="icon">🗑️</span>清空</button>
<button id="saveWordBtn" onclick="saveToWord()"><span class="icon">📄</span>保存为Word</button> <button id="saveWordBtn" onclick="saveToWord()"><span class="icon">📄</span>保存为Word</button>
<div class="loader" id="loader"></div>
</div> </div>
</div> </div>
<script> <script>
// 禁用所有按钮 // 禁用所有按钮
document.addEventListener('DOMContentLoaded', function() { document.addEventListener('DOMContentLoaded', function () {
const buttons = document.querySelectorAll('button'); const buttons = document.querySelectorAll('button');
buttons.forEach(btn => btn.disabled = true); buttons.forEach(btn => btn.disabled = true);
}); });
MathJax = { MathJax = {
tex: { tex: {
inlineMath: [['$', '$'], ['\\(', '\\)']], inlineMath: [['$', '$'], ['\\(', '\\)']],
displayMath: [['$$', '$$'], ['\\[', '\\]']] displayMath: [['$$', '$$'], ['\\[', '\\]']]
}, },
svg: { svg: {
fontCache: 'global' fontCache: 'global'
}, },
startup: { startup: {
pageReady: () => { pageReady: () => {
return MathJax.startup.defaultPageReady().then(() => { return MathJax.startup.defaultPageReady().then(() => {
console.log('MathJax initialized'); console.log('MathJax initialized');
// 启用所有按钮 // 启用所有按钮
const buttons = document.querySelectorAll('button'); const buttons = document.querySelectorAll('button');
buttons.forEach(btn => btn.disabled = false); buttons.forEach(btn => btn.disabled = false);
return MathJax.typesetPromise(); return MathJax.typesetPromise();
}); });
} }
} }
}; };
</script> </script>
<!--国内可以访问的JS,不要使用静态的本地的，因为文件不全的话没有好的显示效果！--> <!--国内可以访问的JS,不要使用静态的本地的，因为文件不全的话没有好的显示效果！-->
<script src="https://cdn.bootcdn.net/ajax/libs/mathjax/3.2.2/es5/tex-mml-chtml.js" id="MathJax-script" async></script> <script src="https://cdn.bootcdn.net/ajax/libs/mathjax/3.2.2/es5/tex-mml-chtml.js" id="MathJax-script" async></script>
@ -282,29 +270,27 @@
function submitQuestion() { function submitQuestion() {
const question = document.getElementById('questionInput').value.trim(); const question = document.getElementById('questionInput').value.trim();
const checkboxes = document.querySelectorAll('input[name="tags"]:checked'); const checkboxes = document.querySelectorAll('input[name="tags"]:checked');
if (!question) { if (!question) {
alert('请输入问题！'); alert('请输入问题！');
return; return;
} }
if (checkboxes.length === 0) { if (checkboxes.length === 0) {
alert('请至少选择一个原稿！'); alert('请至少选择一个原稿！');
return; return;
} }
const selectedDocs = Array.from(checkboxes).map(cb => cb.value); const selectedDocs = Array.from(checkboxes).map(cb => cb.value);
const loader = document.getElementById('loader');
const answerArea = document.getElementById('answerArea'); const answerArea = document.getElementById('answerArea');
answerArea.innerHTML = ''; answerArea.innerHTML = '';
loader.style.display = 'block';
const data = { const data = {
query: question, query: question,
tags: selectedDocs tags: selectedDocs
}; };
fetch('/api/rag', { fetch('/api/rag', {
method: 'POST', method: 'POST',
headers: { headers: {
@ -313,51 +299,47 @@
}, },
body: JSON.stringify(data) body: JSON.stringify(data)
}) })
.then(response => { .then(response => {
const reader = response.body.getReader(); const reader = response.body.getReader();
const decoder = new TextDecoder(); const decoder = new TextDecoder();
let buffer = ''; let buffer = '';
let accumulatedContent = ''; let accumulatedContent = '';
function processChunk() { function processChunk() {
return reader.read().then(({done, value}) => { return reader.read().then(({done, value}) => {
if (done) { if (done) return;
loader.style.display = 'none';
return; buffer += decoder.decode(value, {stream: true});
} const lines = buffer.split('\n');
buffer = lines.pop();
buffer += decoder.decode(value, {stream: true});
const lines = buffer.split('\n'); for (const line of lines) {
buffer = lines.pop(); if (line.includes('data:')) {
const jsonStr = line.replace(/^data:\s*/, '').replace(/^data:\s*/, '').trim();
for (const line of lines) { if (jsonStr) {
if (line.includes('data:')) { try {
const jsonStr = line.replace(/^data:\s*/, '').replace(/^data:\s*/, '').trim(); const data = JSON.parse(jsonStr);
if (jsonStr) { if (data.reply) {
try { accumulatedContent += data.reply;
const data = JSON.parse(jsonStr); answerArea.innerHTML = marked.parse(accumulatedContent);
if (data.reply) { MathJax.typesetPromise();
accumulatedContent += data.reply; }
answerArea.innerHTML = marked.parse(accumulatedContent); } catch (e) {
MathJax.typesetPromise(); console.log('忽略解析错误:', e);
} }
} catch (e) {
console.log('忽略解析错误:', e);
} }
} }
} }
}
return processChunk();
return processChunk(); });
}); }
}
return processChunk();
return processChunk(); })
}) .catch(error => {
.catch(error => { console.error('Error:', error);
console.error('Error:', error); });
loader.style.display = 'none';
});
} }
function clearAll() { function clearAll() {
@ -380,9 +362,6 @@
return; return;
} }
const loader = document.getElementById('loader');
loader.style.display = 'block';
fetch('/api/save-word', { fetch('/api/save-word', {
method: 'POST', method: 'POST',
@ -395,7 +374,6 @@
}) })
.then(response => response.blob()) .then(response => response.blob())
.then(blob => { .then(blob => {
loader.style.display = 'none';
const url = window.URL.createObjectURL(blob); const url = window.URL.createObjectURL(blob);
const a = document.createElement('a'); const a = document.createElement('a');
a.href = url; a.href = url;
@ -406,7 +384,6 @@
document.body.removeChild(a); document.body.removeChild(a);
}) })
.catch(error => { .catch(error => {
loader.style.display = 'none';
console.error('Error:', error); console.error('Error:', error);
alert('保存失败，请稍后再试！'); alert('保存失败，请稍后再试！');
}); });

Write Preview
Loading…
Cancel
Save