From ae93470be33b863c32068ff07f2b5ab15268c2a7 Mon Sep 17 00:00:00 2001
From: HuangHai <10402852@qq.com>
Date: Mon, 7 Jul 2025 11:05:54 +0800
Subject: [PATCH] 'commit'

---
 dsRagAnything/T3_Start.py                     |  11 +-
 .../Math/kv_store_llm_response_cache.json     |  40 ++
 dsRagAnything/static/ai.html                  | 371 ++++++++++++++++++
 3 files changed, 415 insertions(+), 7 deletions(-)
 create mode 100644 dsRagAnything/static/ai.html

diff --git a/dsRagAnything/T3_Start.py b/dsRagAnything/T3_Start.py
index b5fac659..d79dd3ee 100644
--- a/dsRagAnything/T3_Start.py
+++ b/dsRagAnything/T3_Start.py
@@ -41,7 +41,6 @@ async def lifespan(app: FastAPI):
     yield
 
 
-
 async def print_stream(stream):
     async for chunk in stream:
         if chunk:
@@ -53,16 +52,14 @@ app = FastAPI(lifespan=lifespan)
 # 挂载静态文件目录
 app.mount("/static", StaticFiles(directory="Static"), name="static")
 
-# 数学索引文件目录
-# WORKING_PATH = "./Topic/Math"
-
-# 苏轼索引文件目录
-WORKING_PATH = "./Topic/Chinese"
-
 
 @app.post("/api/rag")
 async def rag(request: fastapi.Request):
     data = await request.json()
+    topic = data.get("topic") # Chinese, Math
+    # 拼接路径
+    WORKING_PATH= "./Topic/" + topic
+    # 查询的问题
     query = data.get("query")
     # 关闭参考资料
     query = query + "\n 不要输出参考资料！"
diff --git a/dsRagAnything/Topic/Math/kv_store_llm_response_cache.json b/dsRagAnything/Topic/Math/kv_store_llm_response_cache.json
index bf90e9e6..51e4f506 100644
--- a/dsRagAnything/Topic/Math/kv_store_llm_response_cache.json
+++ b/dsRagAnything/Topic/Math/kv_store_llm_response_cache.json
@@ -2191,6 +2191,46 @@
       "embedding_min": null,
       "embedding_max": null,
       "original_prompt": "小学数学中有哪些模型？"
+    },
+    "9b50a199b83407c62d8532b7b904c17d": {
+      "return": "{\"high_level_keywords\": [\"\\u82cf\\u8f7c\", \"\\u5bb6\\u4eba\", \"\\u5bb6\\u5ead\\u5173\\u7cfb\"], \"low_level_keywords\": [\"\\u7236\\u4eb2\", \"\\u6bcd\\u4eb2\", \"\\u5144\\u5f1f\", \"\\u59bb\\u5b50\", \"\\u5b50\\u5973\"]}",
+      "cache_type": "keywords",
+      "chunk_id": null,
+      "embedding": null,
+      "embedding_shape": null,
+      "embedding_min": null,
+      "embedding_max": null,
+      "original_prompt": "苏轼的家人都有谁？"
+    },
+    "585eef85444b748e7c7fce9281f7152b": {
+      "return": "{\"high_level_keywords\": [\"\\u82cf\\u8f7c\", \"\\u5bb6\\u5ead\\u6210\\u5458\", \"\\u5bb6\\u65cf\\u5173\\u7cfb\"], \"low_level_keywords\": [\"\\u7a0b\\u592b\\u4eba\", \"\\u738b\\u5f17\", \"\\u82cf\\u8f99\", \"\\u82cf\\u8fc8\", \"\\u7709\\u5c71\"]}",
+      "cache_type": "keywords",
+      "chunk_id": null,
+      "embedding": null,
+      "embedding_shape": null,
+      "embedding_min": null,
+      "embedding_max": null,
+      "original_prompt": "苏轼的家人都有谁？\n 不要输出参考资料！"
+    },
+    "5c5ee10a1a6deda17447c905118427e3": {
+      "return": "{\"high_level_keywords\": [\"\\u82cf\\u8f7c\", \"\\u597d\\u53cb\", \"\\u4eba\\u9645\\u5173\\u7cfb\"], \"low_level_keywords\": [\"\\u9ec4\\u5ead\\u575a\", \"\\u79e6\\u89c2\", \"\\u7c73\\u82be\", \"\\u4f5b\\u5370\", \"\\u6587\\u4eba\\u4ea4\\u5f80\"]}",
+      "cache_type": "keywords",
+      "chunk_id": null,
+      "embedding": null,
+      "embedding_shape": null,
+      "embedding_min": null,
+      "embedding_max": null,
+      "original_prompt": "苏轼的好朋友都有谁？\n 不要输出参考资料！"
+    },
+    "aa541ae622441668370c2515c5756622": {
+      "return": "{\"high_level_keywords\": [\"\\u5c0f\\u5b66\\u6570\\u5b66\", \"\\u6570\\u5b66\\u6a21\\u578b\", \"\\u57fa\\u7840\\u6559\\u80b2\"], \"low_level_keywords\": [\"\\u52a0\\u51cf\\u6cd5\", \"\\u4e58\\u9664\\u6cd5\", \"\\u5206\\u6570\", \"\\u5c0f\\u6570\", \"\\u51e0\\u4f55\\u56fe\\u5f62\", \"\\u5e94\\u7528\\u9898\"]}",
+      "cache_type": "keywords",
+      "chunk_id": null,
+      "embedding": null,
+      "embedding_shape": null,
+      "embedding_min": null,
+      "embedding_max": null,
+      "original_prompt": "小学数学中有哪些模型？\n 不要输出参考资料！"
     }
   }
 }
\ No newline at end of file
diff --git a/dsRagAnything/static/ai.html b/dsRagAnything/static/ai.html
new file mode 100644
index 00000000..77add2d4
--- /dev/null
+++ b/dsRagAnything/static/ai.html
@@ -0,0 +1,371 @@
+<!DOCTYPE html>
+<html lang="zh-CN">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>【东师理想】教学大模型</title>
+    <link rel="icon" href="data:,">
+    <style>
+        body {
+            font-family: 'Microsoft YaHei', sans-serif;
+            margin: 0;
+            padding: 0;
+            background-color: #f5f5f5;
+        }
+
+        .container {
+            max-width: 800px;
+            margin: 0 auto;
+            padding: 20px;
+        }
+
+        h1 {
+            color: #333;
+            text-align: center;
+        }
+
+        .data-area {
+            background-color: white;
+            border-radius: 8px;
+            padding: 20px;
+            margin-bottom: 20px;
+            box-shadow: 0 2px 4px rgba(0, 0, 0, 0.1);
+        }
+
+        .input-area {
+            background-color: white;
+            border-radius: 8px;
+            padding: 20px;
+            box-shadow: 0 2px 4px rgba(0, 0, 0, 0.1);
+        }
+
+        textarea {
+            width: 100%;
+            height: 40px;
+            padding: 10px;
+            border: 1px solid #ddd;
+            border-radius: 4px;
+            resize: none;
+            margin-bottom: 10px;
+            margin-right: 20px;
+        }
+
+        button {
+            background-color: #4CAF50;
+            color: white;
+            border: none;
+            padding: 10px 15px;
+            border-radius: 4px;
+            cursor: pointer;
+            margin-right: 10px;
+        }
+
+        button:hover {
+            background-color: #45a049;
+        }
+
+        #clearBtn {
+            background-color: #f44336;
+        }
+
+        #clearBtn:hover {
+            background-color: #d32f2f;
+        }
+
+        #saveWordBtn {
+            background-color: #2196F3;
+        }
+
+        #saveWordBtn:hover {
+            background-color: #0b7dda;
+        }
+
+        .icon {
+            margin-right: 5px;
+        }
+
+        @keyframes spin {
+            0% { transform: rotate(0deg); }
+            100% { transform: rotate(360deg); }
+        }
+
+        .doc-links {
+            display: flex;
+            gap: 20px;
+            margin: 20px 0;
+            align-items: center;
+        }
+
+        .doc-link {
+            padding: 8px 15px;
+            background-color: #e7f3fe;
+            border-radius: 4px;
+            color: #2196F3;
+            text-decoration: none;
+            transition: all 0.3s;
+        }
+
+        .doc-link:hover {
+            background-color: #d0e3fa;
+        }
+
+        .doc-checkboxes {
+            display: flex;
+            gap: 20px;
+            margin-bottom: 10px;
+        }
+
+        .doc-checkboxes label {
+            display: flex;
+            align-items: center;
+            gap: 5px;
+            cursor: pointer;
+        }
+
+        .examples {
+            margin: 20px 0;
+        }
+
+        .example-category {
+            margin-bottom: 15px;
+        }
+
+        .example-category h3 {
+            margin-bottom: 10px;
+            color: #333;
+        }
+
+        .example-list {
+            display: flex;
+            flex-wrap: wrap;
+            gap: 10px;
+        }
+
+        .example-item {
+            padding: 8px 12px;
+            background-color: #e9f5ff;
+            border-radius: 4px;
+            cursor: pointer;
+            transition: all 0.2s;
+        }
+
+        .example-item:hover {
+            background-color: #d0e3fa;
+        }
+
+        .input-area textarea#questionInput {
+            margin-right: 10px !important;
+            width: calc(100% - 10px);
+            box-sizing: border-box;
+        }
+
+        #answerArea {
+            min-height: 240px;
+            height: auto;
+            overflow-y: auto;
+        }
+
+        .loading-animation {
+            display: flex;
+            flex-direction: column;
+            align-items: center;
+            justify-content: center;
+            height: 200px;
+            color: #666;
+        }
+
+        .spinner {
+            width: 40px;
+            height: 40px;
+            border: 4px solid #f3f3f3;
+            border-top: 4px solid #3498db;
+            border-radius: 50%;
+            animation: spin 1s linear infinite;
+            margin-bottom: 10px;
+        }
+    </style>
+</head>
+<body>
+<div class="container">
+    <h1>【东师理想】教学大模型</h1>
+    <div class="data-area" id="answerArea">
+        <div style="color:#666; padding:20px; text-align:center;">
+            <p>请在下方输入您的问题，答案将在此处显示</p>
+            <p>您也可以点击"示例问题"快速体验</p>
+        </div>
+    </div>
+
+    <div class="input-area">
+        <label for="questionInput"></label><textarea id="questionInput" placeholder="请输入您的问题..."></textarea>
+
+        <div class="examples">
+            <div class="example-category">
+                <h3>问题示例：</h3>
+                <div class="example-list">
+                    <div class="example-item" onclick="fillExample('小学数学中有哪些模型？')">小学数学中有哪些模型？</div>
+                    <div class="example-item" onclick="fillExample('帮我写一下 "如何理解点、线、面、体、角"的教学设计')">
+                        帮我写一下 "如何理解点、线、面、体、角"的教学设计
+                    </div>
+                    <div class="example-item" onclick="fillExample('苏轼的好朋友都有谁？')">苏轼的好朋友都有谁？</div>
+                    <div class="example-item" onclick="fillExample('苏轼的家人都有谁？')">苏轼的家人都有谁</div>
+                    <div class="example-item" onclick="fillExample('微积分的基本定理是什么？')">微积分的基本定理是什么？</div>
+                    <div class="example-item" onclick="fillExample('线性代数在AI中的应用')">线性代数在AI中的应用</div>
+                </div>
+            </div>
+        </div>
+
+        <div class="doc-links">
+            <h3>知识库范围</h3>
+            <div class="doc-checkboxes">
+                <label>
+                    <input type="radio" name="topic" value="Math" checked>
+                    小学数学
+                </label>
+                <label>
+                    <input type="radio" name="topic" value="Chinese">
+                    苏轼知识
+                </label>
+            </div>
+        </div>
+
+        <button id="submitBtn" onclick="submitQuestion()"><span class="icon">💡</span>提问</button>
+        <button id="clearBtn" onclick="clearAll()"><span class="icon">🗑️</span>清空</button>
+    </div>
+</div>
+
+<script>
+    // 在页面加载时显示加载动画
+    document.addEventListener('DOMContentLoaded', function () {
+        const answerArea = document.getElementById('answerArea');
+        answerArea.innerHTML = '<div class="loading-animation"><div class="spinner"></div><div>正在加载资源...</div></div>';
+
+        // 禁用所有按钮
+        const buttons = document.querySelectorAll('button');
+        buttons.forEach(btn => btn.disabled = true);
+    });
+
+    MathJax = {
+        loader: {load: ['[tex]/mhchem']},
+        tex: {
+            inlineMath: [['$', '$'], ['\\(', '\\)']],
+            displayMath: [['$$', '$$'], ['\\[', '\\]']],
+            packages: {'[+]': ['mhchem']}
+        },
+        options: {
+            enableMenu: false,
+            processHtmlClass: 'tex2jax_process',
+            ignoreHtmlClass: 'tex2jax_ignore'
+        },
+        svg: {
+            fontCache: 'global'
+        },
+        startup: {
+            pageReady: () => {
+                return MathJax.startup.defaultPageReady().then(() => {
+                    console.log('MathJax initialized');
+                    // 移除加载动画并启用所有按钮
+                    const answerArea = document.getElementById('answerArea');
+                    answerArea.innerHTML = '<div style="color:#666; padding:20px; text-align:center;"><p>请在下方输入您的问题，答案将在此处显示</p><p>您也可以点击"示例问题"快速体验</p></div>';
+
+                    const buttons = document.querySelectorAll('button');
+                    buttons.forEach(btn => btn.disabled = false);
+                    return MathJax.typesetPromise();
+                });
+            }
+        }
+    };
+
+    // marked配置
+    const markedOptions = {
+        mangle: false,
+        headerIds: false
+    };
+
+    function fillExample(question) {
+        document.getElementById('questionInput').value = question;
+    }
+
+    function submitQuestion() {
+        const question = document.getElementById('questionInput').value.trim();
+        const answerArea = document.getElementById('answerArea');
+        const topic = document.querySelector('input[name="topic"]:checked').value;
+    
+        if (!question) {
+            alert('请输入问题！');
+            return;
+        }
+    
+        // 添加加载动画
+        answerArea.innerHTML = '<div class="loading-animation"><div class="spinner"></div><div>思考中...</div></div>';
+    
+        fetch('/api/rag', {
+            method: 'POST',
+            headers: {
+                'Content-Type': 'application/json',
+                'Accept': 'text/event-stream'
+            },
+            body: JSON.stringify({
+                query: question,
+                topic: topic
+            })
+        })
+        .then(response => {
+            const reader = response.body.getReader();
+            const decoder = new TextDecoder();
+            let buffer = '';
+            let accumulatedContent = '';
+    
+            function processChunk() {
+                return reader.read().then(({done, value}) => {
+                    if (done) return;
+    
+                    buffer += decoder.decode(value, {stream: true});
+                    const lines = buffer.split('\n');
+                    buffer = lines.pop();
+    
+                    for (const line of lines) {
+                        if (line.includes('data:')) {
+                            const jsonStr = line.replace(/^data:\s*/, '').replace(/^data:\s*/, '').trim();
+                            if (jsonStr) {
+                                try {
+                                    const data = JSON.parse(jsonStr);
+                                    if (data.reply) {
+                                        accumulatedContent += data.reply;
+                                        answerArea.innerHTML = marked.parse(accumulatedContent, markedOptions);
+                                        MathJax.typesetPromise();
+                                    }
+                                } catch (e) {
+                                    console.log('忽略解析错误:', e);
+                                }
+                            }
+                        }
+                    }
+    
+                    return processChunk().then(() => {
+                        // 在流处理完成后保存完整的markdown内容
+                        localStorage.setItem('lastMarkdownContent', accumulatedContent);
+                    });
+                });
+            }
+    
+            return processChunk();
+        })
+        .catch(error => {
+            // 移除加载动画
+            answerArea.innerHTML = '';
+    
+            console.error('Error:', error);
+            // 出错时也移除加载动画
+            answerArea.innerHTML = '<div style="color:red">请求出错，请重试</div>';
+        });
+    }
+
+    function clearAll() {
+        document.getElementById('questionInput').value = '';
+        document.getElementById('answerArea').innerHTML = '';
+    }
+</script>
+<script src="/static/mathjax/es5/tex-mml-chtml.js" id="MathJax-script" async></script>
+<script src="/static/js/marked.min.js"></script>
+</body>
+</html>
\ No newline at end of file