Upload 25 files

agents/agent_354996b8_1742743575.json

@@ -77,7 +77,7 @@
       "created_at": 1742743611,
       "token": "db1779829c2b4ce5920be2181574655c",
       "expires_at": 0,
-      "usage_count": 0
+      "usage_count": 5
     },
     {
       "id": "dist_eaa6a1",
@@ -95,7 +95,7 @@
     }
   ],
   "stats": {
-    "usage_count": 2,
-    "last_used": 1742956769
+    "usage_count": 7,
+    "last_used": 1743320740
   }
 }

agents/agent_bff4886f_1742724690.json

@@ -19,7 +19,7 @@
       "created_at": 1742732156,
       "token": "4781dc4214e041b38b1935eb4bb2e592",
       "expires_at": 0,
-      "usage_count": 4
+      "usage_count": 5
     },
     {
       "id": "dist_123907",
@@ -44,7 +44,7 @@
     }
   ],
   "stats": {
-    "usage_count": 10,
-    "last_used": 1742956846
+    "usage_count": 11,
+    "last_used": 1743320208
   }
 }

app.py

@@ -1,517 +1,939 @@
-from flask import Flask, request, jsonify, send_from_directory, render_template, redirect, url_for
-from flask_cors import CORS
-import os
-import time
-import traceback
-import json
-import re
-import sys
-import io
-import threading
-import queue
-import contextlib
-import signal
-import psutil
-from dotenv import load_dotenv
-
-# 导入模块路由
-from modules.knowledge_base.routes import knowledge_bp
-from modules.code_executor.routes import code_executor_bp
-from modules.visualization.routes import visualization_bp
-from modules.agent_builder.routes import agent_builder_bp
-
-# 加载环境变量
-load_dotenv()
-
-app = Flask(__name__)
-CORS(app)
-
-# 注册蓝图
-app.register_blueprint(knowledge_bp, url_prefix='/api/knowledge')
-app.register_blueprint(code_executor_bp, url_prefix='/api/code')
-app.register_blueprint(visualization_bp, url_prefix='/api/visualization')
-app.register_blueprint(agent_builder_bp, url_prefix='/api/agent')
-
-# 确保目录存在
-os.makedirs('static', exist_ok=True)
-os.makedirs('uploads', exist_ok=True)
-os.makedirs('agents', exist_ok=True)
-
-# 用于代码执行的上下文
-execution_contexts = {}
-
-def get_memory_usage():
-    """获取当前进程的内存使用情况"""
-    process = psutil.Process(os.getpid())
-    return f"{process.memory_info().rss / 1024 / 1024:.1f} MB"
-
-class CustomStdin:
-    def __init__(self, input_queue):
-        self.input_queue = input_queue
-        self.buffer = ""
-    
-    def readline(self):
-        if not self.buffer:
-            self.buffer = self.input_queue.get() + "\n"
-        
-        result = self.buffer
-        self.buffer = ""
-        return result
-
-class InteractiveExecution:
-    """管理Python代码的交互式执行"""
-    def __init__(self, code):
-        self.code = code
-        self.context_id = str(time.time())
-        self.is_complete = False
-        self.is_waiting_for_input = False
-        self.stdout_buffer = io.StringIO()
-        self.last_read_position = 0
-        self.input_queue = queue.Queue()
-        self.error = None
-        self.thread = None
-        self.should_terminate = False
-    
-    def run(self):
-        """在单独的线程中启动执行"""
-        self.thread = threading.Thread(target=self._execute)
-        self.thread.daemon = True
-        self.thread.start()
-        
-        # 给执行一点时间开始
-        time.sleep(0.1)
-        return self.context_id
-    
-    def _execute(self):
-        """执行代码，处理标准输入输出"""
-        try:
-            # 保存原始的stdin/stdout
-            orig_stdin = sys.stdin
-            orig_stdout = sys.stdout
-            
-            # 创建自定义stdin
-            custom_stdin = CustomStdin(self.input_queue)
-            
-            # 重定向stdin和stdout
-            sys.stdin = custom_stdin
-            sys.stdout = self.stdout_buffer
-            
-            try:
-                # 检查终止的函数
-                self._last_check_time = 0
-                
-                def check_termination():
-                    if self.should_terminate:
-                        raise KeyboardInterrupt("Execution terminated by user")
-                
-                # 设置一个模拟__main__模块的命名空间
-                shared_namespace = {
-                    "__builtins__": __builtins__,
-                    "_check_termination": check_termination,
-                    "time": time, 
-                    "__name__": "__main__"
-                }
-                
-                # 在这个命名空间中执行用户代码
-                try:
-                    exec(self.code, shared_namespace)
-                except KeyboardInterrupt:
-                    print("\nExecution terminated by user")
-                
-            except Exception as e:
-                self.error = {
-                    "error": str(e),
-                    "traceback": traceback.format_exc()
-                }
-            
-            finally:
-                # 恢复原始stdin/stdout
-                sys.stdin = orig_stdin
-                sys.stdout = orig_stdout
-                
-                # 标记执行完成
-                self.is_complete = True
-                
-        except Exception as e:
-            self.error = {
-                "error": str(e),
-                "traceback": traceback.format_exc()
-            }
-            self.is_complete = True
-    
-    def terminate(self):
-        """终止执行"""
-        self.should_terminate = True
-        
-        # 如果在等待输入，放入一些内容以解除阻塞
-        if self.is_waiting_for_input:
-            self.input_queue.put("\n")
-        
-        # 给执行一点时间终止
-        time.sleep(0.2)
-        
-        # 标记为完成
-        self.is_complete = True
-        
-        return True
-    
-    def provide_input(self, user_input):
-        """为运行的代码提供输入"""
-        self.input_queue.put(user_input)
-        self.is_waiting_for_input = False
-        return True
-    
-    def get_output(self):
-        """获取stdout缓冲区的当前内容"""
-        output = self.stdout_buffer.getvalue()
-        return output
-        
-    def get_new_output(self):
-        """只获取自上次读取以来的新输出"""
-        current_value = self.stdout_buffer.getvalue()
-        if self.last_read_position < len(current_value):
-            new_output = current_value[self.last_read_position:]
-            self.last_read_position = len(current_value)
-            return new_output
-        return ""
-
-@app.route('/')
-def index():
-    """主界面"""
-    return render_template('index.html')
-@app.route('/code_execution.html')
-def index2():
-    """主界面"""
-    return render_template('code_execution.html')
-@app.route('/api/progress/<task_id>', methods=['GET'])
-def get_progress(task_id):
-    """获取文档处理进度"""
-    try:
-        # 从知识库模块访问处理任务
-        from modules.knowledge_base.routes import processing_tasks
-        
-        progress_data = processing_tasks.get(task_id, {
-            'progress': 0,
-            'status': '未找到任务',
-            'error': True
-        })
-        
-        return jsonify({"success": True, "data": progress_data})
-    except Exception as e:
-        traceback.print_exc()
-        return jsonify({"success": False, "message": str(e)}), 500
-
-@app.route('/student/<agent_id>')
-def student_view(agent_id):
-    """学生访问Agent界面"""
-    token = request.args.get('token', '')
-    
-    # 验证Agent存在
-    agent_path = os.path.join('agents', f"{agent_id}.json")
-    if not os.path.exists(agent_path):
-        return render_template('error.html', 
-                              message="找不到指定的Agent",
-                              error_code=404)
-    
-    # 加载Agent配置
-    with open(agent_path, 'r', encoding='utf-8') as f:
-        try:
-            agent_config = json.load(f)
-        except:
-            return render_template('error.html', 
-                                  message="Agent配置无效",
-                                  error_code=500)
-    
-    # 验证访问令牌
-    if token:
-        valid_token = False
-        if "distributions" in agent_config:
-            for dist in agent_config["distributions"]:
-                if dist.get("token") == token:
-                    valid_token = True
-                    break
-        
-        if not valid_token:
-            return render_template('error.html', 
-                                  message="访问令牌无效",
-                                  error_code=403)
-    
-    # 渲染学生页面
-    return render_template('student.html', 
-                          agent_id=agent_id,
-                          agent_name=agent_config.get('name', 'AI学习助手'),
-                          agent_description=agent_config.get('description', ''),
-                          token=token)
-
-@app.route('/code_execution.html')
-def code_execution_page():
-    """代码执行页面"""
-    return send_from_directory(os.path.dirname(os.path.abspath(__file__)), 'code_execution.html')
-
-@app.route('/api/student/chat/<agent_id>', methods=['POST'])
-def student_chat(agent_id):
-    """学生与Agent聊天的API"""
-    try:
-        data = request.json
-        message = data.get('message', '')
-        token = data.get('token', '')
-        
-        if not message:
-            return jsonify({"success": False, "message": "消息不能为空"}), 400
-        
-        # 验证Agent和令牌
-        agent_path = os.path.join('agents', f"{agent_id}.json")
-        if not os.path.exists(agent_path):
-            return jsonify({"success": False, "message": "Agent不存在"}), 404
-        
-        with open(agent_path, 'r', encoding='utf-8') as f:
-            agent_config = json.load(f)
-        
-        # 验证令牌（如果提供）
-        if token and "distributions" in agent_config:
-            valid_token = False
-            for dist in agent_config["distributions"]:
-                if dist.get("token") == token:
-                    valid_token = True
-                    
-                    # 更新使用计数
-                    dist["usage_count"] = dist.get("usage_count", 0) + 1
-                    break
-            
-            if not valid_token:
-                return jsonify({"success": False, "message": "访问令牌无效"}), 403
-                
-        # 更新Agent使用统计
-        if "stats" not in agent_config:
-            agent_config["stats"] = {}
-        
-        agent_config["stats"]["usage_count"] = agent_config["stats"].get("usage_count", 0) + 1
-        agent_config["stats"]["last_used"] = int(time.time())
-        
-        # 保存更新后的Agent配置
-        with open(agent_path, 'w', encoding='utf-8') as f:
-            json.dump(agent_config, f, ensure_ascii=False, indent=2)
-        
-        # 获取Agent关联的知识库和插件
-        knowledge_bases = agent_config.get('knowledge_bases', [])
-        plugins = agent_config.get('plugins', [])
-        
-        # 获取学科和指导者信息
-        subject = agent_config.get('subject', agent_config.get('name', '通用学科'))
-        instructor = agent_config.get('instructor', '教师')
-        
-        # 创建Generator实例，传入学科和指导者信息
-        from modules.knowledge_base.generator import Generator
-        generator = Generator(subject=subject, instructor=instructor)
-        
-        # 检测需要使用的插件
-        suggested_plugins = []
-            
-        # 检测是否需要代码执行插件
-        if 'code' in plugins and ('代码' in message or 'python' in message.lower() or '编程' in message or 'code' in message.lower() or 'program' in message.lower()):
-            suggested_plugins.append('code')
-                
-        # 检测是否需要3D可视化插件
-        if 'visualization' in plugins and ('3d' in message.lower() or '可视化' in message or '图形' in message):
-            suggested_plugins.append('visualization')
-                
-        # 检测是否需要思维导图插件
-        if 'mindmap' in plugins and ('思维导图' in message or 'mindmap' in message.lower()):
-            suggested_plugins.append('mindmap')
-        
-        # 检查是否有配置知识库
-        if not knowledge_bases:
-            # 没有知识库，直接使用模型进行回答
-            print(f"\n=== 处理查询: {message} (无知识库) ===")
-            
-            # 使用空的文档列表调用生成器进行回答
-            final_response = ""
-            for chunk in generator.generate_stream(message, []):
-                if isinstance(chunk, dict):
-                    continue  # 跳过处理数据
-                final_response += chunk
-            
-            # 返回生成的回答
-            return jsonify({
-                "success": True,
-                "message": final_response,
-                "tools": suggested_plugins
-            })
-        
-        # 有知识库配置，执行知识库查询流程
-        try:
-            # 导入RAG系统组件
-            from modules.knowledge_base.retriever import Retriever
-            from modules.knowledge_base.reranker import Reranker
-            
-            retriever = Retriever()
-            reranker = Reranker()
-            
-            # 构建工具定义 - 将所有知识库作为工具
-            tools = []
-            
-            # 创建工具名称到索引的映射
-            tool_to_index = {}
-            
-            for i, index in enumerate(knowledge_bases):
-                display_name = index[4:] if index.startswith('rag_') else index
-                
-                # 判断是否是视频知识库
-                is_video = "视频" in display_name or "video" in display_name.lower()
-                
-                # 根据内容类型生成适当的工具名称
-                if is_video:
-                    tool_name = f"video_knowledge_base_{i+1}"
-                    description = f"在'{display_name}'视频知识库中搜索，返回带时间戳的视频链接。适用于需要视频讲解的问题。"
-                else:
-                    tool_name = f"knowledge_base_{i+1}"
-                    description = f"在'{display_name}'知识库中搜索专业知识、概念和原理。适用于需要文本说明的问题。"
-                
-                # 添加工具名到索引的映射
-                tool_to_index[tool_name] = index
-                
-                tools.append({
-                    "type": "function",
-                    "function": {
-                        "name": tool_name,
-                        "description": description,
-                        "parameters": {
-                            "type": "object",
-                            "properties": {
-                                "keywords": {
-                                    "type": "array",
-                                    "items": {"type": "string"},
-                                    "description": "搜索的关键词列表"
-                                }
-                            },
-                            "required": ["keywords"],
-                            "additionalProperties": False
-                        },
-                        "strict": True
-                    }
-                })
-            
-            # 第一阶段：工具选择决策
-            print(f"\n=== 处理查询: {message} ===")
-            tool_calls = generator.extract_keywords_with_tools(message, tools)
-            
-            # 如果不需要调用工具，直接回答
-            if not tool_calls:
-                print("未检测到需要使用知识库，直接回答")
-                final_response = ""
-                for chunk in generator.generate_stream(message, []):
-                    if isinstance(chunk, dict):
-                        continue  # 跳过处理数据
-                    final_response += chunk
-                
-                return jsonify({
-                    "success": True,
-                    "message": final_response,
-                    "tools": suggested_plugins
-                })
-            
-            # 收集来自工具执行的所有文档
-            all_docs = []
-            
-            # 执行每个工具调用
-            for tool_call in tool_calls:
-                try:
-                    tool_name = tool_call["function"]["name"]
-                    actual_index = tool_to_index.get(tool_name)
-                    
-                    if not actual_index:
-                        print(f"找不到工具名称 '{tool_name}' 对应的索引")
-                        continue
-                    
-                    print(f"\n执行工具 '{tool_name}' -> 使用索引 '{actual_index}'")
-                    
-                    arguments = json.loads(tool_call["function"]["arguments"])
-                    keywords = " ".join(arguments.get("keywords", []))
-                    
-                    if not keywords:
-                        print("没有提供关键词，跳过检索")
-                        continue
-                    
-                    print(f"检索关键词: {keywords}")
-                    
-                    # 执行检索
-                    retrieved_docs, _ = retriever.retrieve(keywords, specific_index=actual_index)
-                    print(f"检索到 {len(retrieved_docs)} 个文档")
-                    
-                    # 重排序文档
-                    reranked_docs = reranker.rerank(message, retrieved_docs, actual_index)
-                    print(f"重排序完成，排序后有 {len(reranked_docs)} 个文档")
-                    
-                    # 添加结果
-                    all_docs.extend(reranked_docs)
-                    
-                except Exception as e:
-                    print(f"执行工具 '{tool_call.get('function', {}).get('name', '未知')}' 调用时出错: {str(e)}")
-                    import traceback
-                    traceback.print_exc()
-            
-            # 如果没有检索到任何文档，直接回答
-            if not all_docs:
-                print("未检索到任何相关文档，直接回答")
-                final_response = ""
-                for chunk in generator.generate_stream(message, []):
-                    if isinstance(chunk, dict):
-                        continue  # 跳过处理数据
-                    final_response += chunk
-                
-                return jsonify({
-                    "success": True,
-                    "message": final_response,
-                    "tools": suggested_plugins
-                })
-            
-            # 按相关性排序
-            all_docs.sort(key=lambda x: x.get('rerank_score', 0), reverse=True)
-            print(f"\n最终收集到 {len(all_docs)} 个文档用于生成回答")
-            
-            # 提取参考信息
-            references = []
-            for i, doc in enumerate(all_docs[:3], 1):  # 只展示前3个参考来源
-                file_name = doc['metadata'].get('file_name', '未知文件')
-                content = doc['content']
-                
-                # 提取大约前100字符作为摘要
-                summary = content[:100] + ('...' if len(content) > 100 else '')
-                
-                references.append({
-                    'index': i,
-                    'file_name': file_name,
-                    'content': content,
-                    'summary': summary
-                })
-            
-            # 第二阶段：生成最终答案
-            final_response = ""
-            for chunk in generator.generate_stream(message, all_docs):
-                if isinstance(chunk, dict):
-                    continue  # 跳过处理数据
-                final_response += chunk
-            
-            # 构建回复
-            return jsonify({
-                "success": True,
-                "message": final_response,
-                "tools": suggested_plugins,
-                "references": references
-            })
-            
-        except Exception as e:
-            import traceback
-            traceback.print_exc()
-            return jsonify({
-                "success": False, 
-                "message": f"处理查询时出错: {str(e)}"
-            }), 500
-            
-    except Exception as e:
-        import traceback
-        traceback.print_exc()
-        return jsonify({"success": False, "message": str(e)}), 500
-
-if __name__ == '__main__':
+from flask import Flask, request, jsonify, send_from_directory, render_template, redirect, url_for, session
+from flask_cors import CORS
+import os
+import time
+import traceback
+import json
+import re
+import sys
+import io
+import threading
+import queue
+import contextlib
+import signal
+import psutil
+from dotenv import load_dotenv
+
+# 导入模块路由
+from modules.knowledge_base.routes import knowledge_bp
+from modules.code_executor.routes import code_executor_bp
+from modules.visualization.routes import visualization_bp
+from modules.agent_builder.routes import agent_builder_bp
+
+# 加载环境变量
+load_dotenv()
+
+app = Flask(__name__)
+CORS(app)
+
+# 设置session密钥
+app.secret_key = os.getenv("SECRET_KEY", "your_secret_key_here")
+
+# 注册蓝图
+app.register_blueprint(knowledge_bp, url_prefix='/api/knowledge')
+app.register_blueprint(code_executor_bp, url_prefix='/api/code')
+app.register_blueprint(visualization_bp, url_prefix='/api/visualization')
+app.register_blueprint(agent_builder_bp, url_prefix='/api/agent')
+
+# 确保目录存在
+os.makedirs('static', exist_ok=True)
+os.makedirs('uploads', exist_ok=True)
+os.makedirs('agents', exist_ok=True)
+
+# 用于代码执行的上下文
+execution_contexts = {}
+
+# 硬编码用户（仅用于演示）
+users = {
+    "teachers": [
+        {"username": "teacher", "password": "123456", "name": "李志刚"},
+        {"username": "admin", "password": "admin123", "name": "管理员"}
+    ],
+    "students": [
+        {"username": "student1", "password": "123456", "name": "张三"},
+        {"username": "student2", "password": "123456", "name": "李四"}
+    ]
+}
+
+# 学生活动记录存储（实际应用中应使用数据库）
+student_activities = {}
+
+def get_memory_usage():
+    """获取当前进程的内存使用情况"""
+    process = psutil.Process(os.getpid())
+    return f"{process.memory_info().rss / 1024 / 1024:.1f} MB"
+
+class CustomStdin:
+    def __init__(self, input_queue):
+        self.input_queue = input_queue
+        self.buffer = ""
+    
+    def readline(self):
+        if not self.buffer:
+            self.buffer = self.input_queue.get() + "\n"
+        
+        result = self.buffer
+        self.buffer = ""
+        return result
+
+class InteractiveExecution:
+    """管理Python代码的交互式执行"""
+    def __init__(self, code):
+        self.code = code
+        self.context_id = str(time.time())
+        self.is_complete = False
+        self.is_waiting_for_input = False
+        self.stdout_buffer = io.StringIO()
+        self.last_read_position = 0
+        self.input_queue = queue.Queue()
+        self.error = None
+        self.thread = None
+        self.should_terminate = False
+    
+    def run(self):
+        """在单独的线程中启动执行"""
+        self.thread = threading.Thread(target=self._execute)
+        self.thread.daemon = True
+        self.thread.start()
+        
+        # 给执行一点时间开始
+        time.sleep(0.1)
+        return self.context_id
+    
+    def _execute(self):
+        """执行代码，处理标准输入输出"""
+        try:
+            # 保存原始的stdin/stdout
+            orig_stdin = sys.stdin
+            orig_stdout = sys.stdout
+            
+            # 创建自定义stdin
+            custom_stdin = CustomStdin(self.input_queue)
+            
+            # 重定向stdin和stdout
+            sys.stdin = custom_stdin
+            sys.stdout = self.stdout_buffer
+            
+            try:
+                # 检查终止的函数
+                self._last_check_time = 0
+                
+                def check_termination():
+                    if self.should_terminate:
+                        raise KeyboardInterrupt("Execution terminated by user")
+                
+                # 设置一个模拟__main__模块的命名空间
+                shared_namespace = {
+                    "__builtins__": __builtins__,
+                    "_check_termination": check_termination,
+                    "time": time, 
+                    "__name__": "__main__"
+                }
+                
+                # 在这个命名空间中执行用户代码
+                try:
+                    exec(self.code, shared_namespace)
+                except KeyboardInterrupt:
+                    print("\nExecution terminated by user")
+                
+            except Exception as e:
+                self.error = {
+                    "error": str(e),
+                    "traceback": traceback.format_exc()
+                }
+            
+            finally:
+                # 恢复原始stdin/stdout
+                sys.stdin = orig_stdin
+                sys.stdout = orig_stdout
+                
+                # 标记执行完成
+                self.is_complete = True
+                
+        except Exception as e:
+            self.error = {
+                "error": str(e),
+                "traceback": traceback.format_exc()
+            }
+            self.is_complete = True
+    
+    def terminate(self):
+        """终止执行"""
+        self.should_terminate = True
+        
+        # 如果在等待输入，放入一些内容以解除阻塞
+        if self.is_waiting_for_input:
+            self.input_queue.put("\n")
+        
+        # 给执行一点时间终止
+        time.sleep(0.2)
+        
+        # 标记为完成
+        self.is_complete = True
+        
+        return True
+    
+    def provide_input(self, user_input):
+        """为运行的代码提供输入"""
+        self.input_queue.put(user_input)
+        self.is_waiting_for_input = False
+        return True
+    
+    def get_output(self):
+        """获取stdout缓冲区的当前内容"""
+        output = self.stdout_buffer.getvalue()
+        return output
+        
+    def get_new_output(self):
+        """只获取自上次读取以来的新输出"""
+        current_value = self.stdout_buffer.getvalue()
+        if self.last_read_position < len(current_value):
+            new_output = current_value[self.last_read_position:]
+            self.last_read_position = len(current_value)
+            return new_output
+        return ""
+
+# 记录活动函数（可在各个操作点调用）
+def record_student_activity(username, activity_type, title, agent_id=None, agent_name=None):
+    """记录学生活动"""
+    if username not in student_activities:
+        student_activities[username] = []
+    
+    # 创建活动记录
+    activity = {
+        "type": activity_type,  # 'chat', 'code', 'viz', 'mindmap'
+        "title": title,
+        "timestamp": int(time.time()),
+        "agent_id": agent_id,
+        "agent_name": agent_name
+    }
+    
+    # 添加到用户活动列表（最多保存20条记录）
+    student_activities[username].insert(0, activity)
+    if len(student_activities[username]) > 20:
+        student_activities[username] = student_activities[username][:20]
+    
+    return activity
+
+# 登录相关路由
+@app.route('/login.html')
+def login_page():
+    """登录页面"""
+    return render_template('login.html')
+
+@app.route('/api/auth/login', methods=['POST'])
+def login():
+    """处理登录请求"""
+    data = request.json
+    username = data.get('username')
+    password = data.get('password')
+    user_type = data.get('type', 'teacher')  # 默认为教师
+    
+    if user_type == 'teacher':
+        user_list = users['teachers']
+    else:
+        user_list = users['students']
+    
+    for user in user_list:
+        if user['username'] == username and user['password'] == password:
+            # 设置session
+            session['logged_in'] = True
+            session['username'] = username
+            session['user_type'] = user_type
+            session['user_name'] = user['name']
+            
+            return jsonify({
+                'success': True,
+                'user': {
+                    'name': user['name'],
+                    'type': user_type
+                }
+            })
+    
+    return jsonify({
+        'success': False,
+        'message': '用户名或密码错误'
+    }), 401
+
+@app.route('/api/auth/logout', methods=['POST'])
+def logout():
+    """处理登出请求"""
+    session.clear()
+    return jsonify({
+        'success': True
+    })
+
+@app.route('/api/auth/check', methods=['GET'])
+def check_auth():
+    """检查用户是否已登录"""
+    if session.get('logged_in'):
+        return jsonify({
+            'success': True,
+            'user': {
+                'name': session.get('user_name'),
+                'type': session.get('user_type')
+            }
+        })
+    
+    return jsonify({
+        'success': False
+    }), 401
+
+# 登录验证装饰器
+def login_required(f):
+    def decorated_function(*args, **kwargs):
+        if not session.get('logged_in'):
+            return redirect(url_for('login_page'))
+        return f(*args, **kwargs)
+    decorated_function.__name__ = f.__name__
+    return decorated_function
+
+# 教师角色验证装饰器
+def teacher_required(f):
+    def decorated_function(*args, **kwargs):
+        if not session.get('logged_in') or session.get('user_type') != 'teacher':
+            return jsonify({
+                'success': False,
+                'message': '需要教师权限'
+            }), 403
+        return f(*args, **kwargs)
+    decorated_function.__name__ = f.__name__
+    return decorated_function
+
+# 学生角色验证装饰器
+def student_required(f):
+    def decorated_function(*args, **kwargs):
+        if not session.get('logged_in') or session.get('user_type') != 'student':
+            return jsonify({
+                'success': False,
+                'message': '需要学生权限'
+            }), 403
+        return f(*args, **kwargs)
+    decorated_function.__name__ = f.__name__
+    return decorated_function
+
+# 首页路由
+@app.route('/')
+def root():
+    """重定向到登录页面或主界面"""
+    if session.get('logged_in'):
+        if session.get('user_type') == 'teacher':
+            return redirect('/index.html')
+        else:
+            return redirect('/student_portal.html')
+    return redirect('/login.html')
+
+@app.route('/index.html')
+@login_required
+def index():
+    """教师端主界面"""
+    if session.get('user_type') != 'teacher':
+        return redirect('/student_portal.html')
+    return render_template('index.html')
+
+@app.route('/student_portal.html')
+@login_required
+def student_portal():
+    """学生端门户"""
+    if session.get('user_type') != 'student':
+        return redirect('/index.html')
+    return render_template('student_portal.html')
+
+@app.route('/code_execution.html')
+def code_execution_page():
+    """代码执行页面"""
+    return send_from_directory(os.path.dirname(os.path.abspath(__file__)), 'templates/code_execution.html')
+
+@app.route('/verify_token.html')
+def verify_token_page():
+    """令牌验证页面"""
+    return render_template('token_verification.html')
+
+@app.route('/api/progress/<task_id>', methods=['GET'])
+def get_progress(task_id):
+    """获取文档处理进度"""
+    try:
+        # 从知识库模块访问处理任务
+        from modules.knowledge_base.routes import processing_tasks
+        
+        progress_data = processing_tasks.get(task_id, {
+            'progress': 0,
+            'status': '未找到任务',
+            'error': True
+        })
+        
+        return jsonify({"success": True, "data": progress_data})
+    except Exception as e:
+        traceback.print_exc()
+        return jsonify({"success": False, "message": str(e)}), 500
+
+@app.route('/student/<agent_id>')
+def student_view(agent_id):
+    """学生访问Agent界面"""
+    token = request.args.get('token', '')
+    
+    # 验证Agent存在
+    agent_path = os.path.join('agents', f"{agent_id}.json")
+    if not os.path.exists(agent_path):
+        return render_template('error.html', 
+                              message="找不到指定的Agent",
+                              error_code=404)
+    
+    # 加载Agent配置
+    with open(agent_path, 'r', encoding='utf-8') as f:
+        try:
+            agent_config = json.load(f)
+        except:
+            return render_template('error.html', 
+                                  message="Agent配置无效",
+                                  error_code=500)
+    
+    # 验证访问令牌
+    if token:
+        valid_token = False
+        if "distributions" in agent_config:
+            for dist in agent_config["distributions"]:
+                if dist.get("token") == token:
+                    valid_token = True
+                    break
+        
+        if not valid_token:
+            return render_template('token_verification.html', 
+                                  message="访问令牌无效",
+                                  error_code=403)
+    
+    # 更新使用统计
+    if "distributions" in agent_config:
+        for dist in agent_config["distributions"]:
+            if dist.get("token") == token:
+                # 更新分发使用次数
+                dist["usage_count"] = dist.get("usage_count", 0) + 1
+                
+                # 更新Agent使用统计
+                if "stats" not in agent_config:
+                    agent_config["stats"] = {}
+                
+                agent_config["stats"]["usage_count"] = agent_config["stats"].get("usage_count", 0) + 1
+                agent_config["stats"]["last_used"] = int(time.time())
+                
+                # 保存更新后的Agent配置
+                with open(agent_path, 'w', encoding='utf-8') as f:
+                    json.dump(agent_config, f, ensure_ascii=False, indent=2)
+                
+                break
+    
+    # 渲染学生页面
+    return render_template('student.html', 
+                          agent_id=agent_id,
+                          agent_name=agent_config.get('name', 'AI学习助手'),
+                          agent_description=agent_config.get('description', ''),
+                          token=token)
+
+@app.route('/api/student/chat/<agent_id>', methods=['POST'])
+def student_chat(agent_id):
+    """学生与Agent聊天的API"""
+    try:
+        data = request.json
+        message = data.get('message', '')
+        token = data.get('token', '')
+        
+        if not message:
+            return jsonify({"success": False, "message": "消息不能为空"}), 400
+        
+        # 验证Agent和令牌
+        agent_path = os.path.join('agents', f"{agent_id}.json")
+        if not os.path.exists(agent_path):
+            return jsonify({"success": False, "message": "Agent不存在"}), 404
+        
+        with open(agent_path, 'r', encoding='utf-8') as f:
+            agent_config = json.load(f)
+        
+        # 验证令牌（如果提供）
+        if token and "distributions" in agent_config:
+            valid_token = False
+            for dist in agent_config["distributions"]:
+                if dist.get("token") == token:
+                    valid_token = True
+                    
+                    # 更新使用计数
+                    dist["usage_count"] = dist.get("usage_count", 0) + 1
+                    break
+            
+            if not valid_token:
+                return jsonify({"success": False, "message": "访问令牌无效"}), 403
+                
+        # 更新Agent使用统计
+        if "stats" not in agent_config:
+            agent_config["stats"] = {}
+        
+        agent_config["stats"]["usage_count"] = agent_config["stats"].get("usage_count", 0) + 1
+        agent_config["stats"]["last_used"] = int(time.time())
+        
+        # 保存更新后的Agent配置
+        with open(agent_path, 'w', encoding='utf-8') as f:
+            json.dump(agent_config, f, ensure_ascii=False, indent=2)
+        
+        # 获取Agent关联的知识库和插件
+        knowledge_bases = agent_config.get('knowledge_bases', [])
+        plugins = agent_config.get('plugins', [])
+        
+        # 获取学科和指导者信息
+        subject = agent_config.get('subject', agent_config.get('name', '通用学科'))
+        instructor = agent_config.get('instructor', '教师')
+        
+        # 创建Generator实例，传入学科和指导者信息
+        from modules.knowledge_base.generator import Generator
+        generator = Generator(subject=subject, instructor=instructor)
+        
+        # 检测需要使用的插件
+        suggested_plugins = []
+            
+        # 检测是否需要代码执行插件
+        if 'code' in plugins and ('代码' in message or 'python' in message.lower() or '编程' in message or 'code' in message.lower() or 'program' in message.lower()):
+            suggested_plugins.append('code')
+                
+        # 检测是否需要3D可视化插件
+        if 'visualization' in plugins and ('3d' in message.lower() or '可视化' in message or '图形' in message):
+            suggested_plugins.append('visualization')
+                
+        # 检测是否需要思维导图插件
+        if 'mindmap' in plugins and ('思维导图' in message or 'mindmap' in message.lower()):
+            suggested_plugins.append('mindmap')
+        
+        # 记录活动（添加此部分代码）
+        if session.get('logged_in'):
+            username = session.get('username')
+            # 记录对话活动
+            record_student_activity(
+                username=username,
+                activity_type='chat',
+                title=f'与 {agent_config.get("name", "AI助手")} 进行了对话',
+                agent_id=agent_id,
+                agent_name=agent_config.get('name')
+            )
+            
+            # 如果使用了插件，记录相应的插件活动
+            if 'code' in suggested_plugins:
+                record_student_activity(
+                    username=username,
+                    activity_type='code',
+                    title='执行了Python代码',
+                    agent_id=agent_id,
+                    agent_name=agent_config.get('name')
+                )
+            
+            if 'visualization' in suggested_plugins:
+                record_student_activity(
+                    username=username,
+                    activity_type='viz',
+                    title='查看了3D可视化图形',
+                    agent_id=agent_id,
+                    agent_name=agent_config.get('name')
+                )
+                
+            if 'mindmap' in suggested_plugins:
+                record_student_activity(
+                    username=username,
+                    activity_type='mindmap',
+                    title='生成了思维导图',
+                    agent_id=agent_id,
+                    agent_name=agent_config.get('name')
+                )
+        
+        # 检查是否有配置知识库
+        if not knowledge_bases:
+            # 没有知识库，直接使用模型进行回答
+            print(f"\n=== 处理查询: {message} (无知识库) ===")
+            
+            # 使用空的文档列表调用生成器进行回答
+            final_response = ""
+            for chunk in generator.generate_stream(message, []):
+                if isinstance(chunk, dict):
+                    continue  # 跳过处理数据
+                final_response += chunk
+            
+            # 返回生成的回答
+            return jsonify({
+                "success": True,
+                "message": final_response,
+                "tools": suggested_plugins
+            })
+        
+        # 有知识库配置，执行知识库查询流程
+        try:
+            # 导入RAG系统组件
+            from modules.knowledge_base.retriever import Retriever
+            from modules.knowledge_base.reranker import Reranker
+            
+            retriever = Retriever()
+            reranker = Reranker()
+            
+            # 构建工具定义 - 将所有知识库作为工具
+            tools = []
+            
+            # 创建工具名称到索引的映射
+            tool_to_index = {}
+            
+            for i, index in enumerate(knowledge_bases):
+                display_name = index[4:] if index.startswith('rag_') else index
+                
+                # 判断是否是视频知识库
+                is_video = "视频" in display_name or "video" in display_name.lower()
+                
+                # 根据内容类型生成适当的工具名称
+                if is_video:
+                    tool_name = f"video_knowledge_base_{i+1}"
+                    description = f"在'{display_name}'视频知识库中搜索，返回带时间戳的视频链接。适用于需要视频讲解的问题。"
+                else:
+                    tool_name = f"knowledge_base_{i+1}"
+                    description = f"在'{display_name}'知识库中搜索专业知识、概念和原理。适用于需要文本说明的问题。"
+                
+                # 添加工具名到索引的映射
+                tool_to_index[tool_name] = index
+                
+                tools.append({
+                    "type": "function",
+                    "function": {
+                        "name": tool_name,
+                        "description": description,
+                        "parameters": {
+                            "type": "object",
+                            "properties": {
+                                "keywords": {
+                                    "type": "array",
+                                    "items": {"type": "string"},
+                                    "description": "搜索的关键词列表"
+                                }
+                            },
+                            "required": ["keywords"],
+                            "additionalProperties": False
+                        },
+                        "strict": True
+                    }
+                })
+            
+            # 第一阶段：工具选择决策
+            print(f"\n=== 处理查询: {message} ===")
+            tool_calls = generator.extract_keywords_with_tools(message, tools)
+            
+            # 如果不需要调用工具，直接回答
+            if not tool_calls:
+                print("未检测到需要使用知识库，直接回答")
+                final_response = ""
+                for chunk in generator.generate_stream(message, []):
+                    if isinstance(chunk, dict):
+                        continue  # 跳过处理数据
+                    final_response += chunk
+                
+                return jsonify({
+                    "success": True,
+                    "message": final_response,
+                    "tools": suggested_plugins
+                })
+            
+            # 收集来自工具执行的所有文档
+            all_docs = []
+            
+            # 执行每个工具调用
+            for tool_call in tool_calls:
+                try:
+                    tool_name = tool_call["function"]["name"]
+                    actual_index = tool_to_index.get(tool_name)
+                    
+                    if not actual_index:
+                        print(f"找不到工具名称 '{tool_name}' 对应的索引")
+                        continue
+                    
+                    print(f"\n执行工具 '{tool_name}' -> 使用索引 '{actual_index}'")
+                    
+                    arguments = json.loads(tool_call["function"]["arguments"])
+                    keywords = " ".join(arguments.get("keywords", []))
+                    
+                    if not keywords:
+                        print("没有提供关键词，跳过检索")
+                        continue
+                    
+                    print(f"检索关键词: {keywords}")
+                    
+                    # 执行检索
+                    retrieved_docs, _ = retriever.retrieve(keywords, specific_index=actual_index)
+                    print(f"检索到 {len(retrieved_docs)} 个文档")
+                    
+                    # 重排序文档
+                    reranked_docs = reranker.rerank(message, retrieved_docs, actual_index)
+                    print(f"重排序完成，排序后有 {len(reranked_docs)} 个文档")
+                    
+                    # 添加结果
+                    all_docs.extend(reranked_docs)
+                    
+                except Exception as e:
+                    print(f"执行工具 '{tool_call.get('function', {}).get('name', '未知')}' 调用时出错: {str(e)}")
+                    import traceback
+                    traceback.print_exc()
+            
+            # 如果没有检索到任何文档，直接回答
+            if not all_docs:
+                print("未检索到任何相关文档，直接回答")
+                final_response = ""
+                for chunk in generator.generate_stream(message, []):
+                    if isinstance(chunk, dict):
+                        continue  # 跳过处理数据
+                    final_response += chunk
+                
+                return jsonify({
+                    "success": True,
+                    "message": final_response,
+                    "tools": suggested_plugins
+                })
+            
+            # 按相关性排序
+            all_docs.sort(key=lambda x: x.get('rerank_score', 0), reverse=True)
+            print(f"\n最终收集到 {len(all_docs)} 个文档用于生成回答")
+            
+            # 提取参考信息
+            references = []
+            for i, doc in enumerate(all_docs[:3], 1):  # 只展示前3个参考来源
+                file_name = doc['metadata'].get('file_name', '未知文件')
+                content = doc['content']
+                
+                # 提取大约前100字符作为摘要
+                summary = content[:100] + ('...' if len(content) > 100 else '')
+                
+                references.append({
+                    'index': i,
+                    'file_name': file_name,
+                    'content': content,
+                    'summary': summary
+                })
+            
+            # 第二阶段：生成最终答案
+            final_response = ""
+            for chunk in generator.generate_stream(message, all_docs):
+                if isinstance(chunk, dict):
+                    continue  # 跳过处理数据
+                final_response += chunk
+            
+            # 构建回复
+            return jsonify({
+                "success": True,
+                "message": final_response,
+                "tools": suggested_plugins,
+                "references": references
+            })
+            
+        except Exception as e:
+            import traceback
+            traceback.print_exc()
+            return jsonify({
+                "success": False, 
+                "message": f"处理查询时出错: {str(e)}"
+            }), 500
+            
+    except Exception as e:
+        import traceback
+        traceback.print_exc()
+        return jsonify({"success": False, "message": str(e)}), 500
+
+# API端点：获取学生活动记录
+@app.route('/api/student/activities', methods=['GET'])
+@student_required
+def get_student_activities():
+    """获取学生活动记录"""
+    try:
+        username = session.get('username')
+        
+        # 获取该学生的活动记录
+        activities = student_activities.get(username, [])
+        
+        # 格式化输出
+        formatted_activities = []
+        for activity in activities:
+            # 格式化时间显示
+            timestamp = activity['timestamp']
+            current_time = int(time.time())
+            
+            if current_time - timestamp < 86400:  # 24小时内
+                if current_time - timestamp < 3600:  # 1小时内
+                    time_text = f"{(current_time - timestamp) // 60}分钟前"
+                else:
+                    time_text = f"今天 {time.strftime('%H:%M', time.localtime(timestamp))}"
+            elif current_time - timestamp < 172800:  # 48小时内
+                time_text = f"昨天 {time.strftime('%H:%M', time.localtime(timestamp))}"
+            else:
+                time_text = time.strftime('%m月%d日 %H:%M', time.localtime(timestamp))
+            
+            formatted_activities.append({
+                "type": activity['type'],
+                "title": activity['title'],
+                "time": time_text,
+                "agent_id": activity.get('agent_id'),
+                "agent_name": activity.get('agent_name')
+            })
+        
+        return jsonify({
+            "success": True,
+            "activities": formatted_activities
+        })
+    
+    except Exception as e:
+        import traceback
+        traceback.print_exc()
+        return jsonify({
+            "success": False,
+            "message": str(e)
+        }), 500
+
+# API端点：验证访问令牌
+@app.route('/api/verify_token', methods=['POST'])
+def verify_token():
+    """验证访问令牌有效性"""
+    try:
+        data = request.json
+        token = data.get('token', '')
+        agent_id = data.get('agent_id', '')
+        
+        if not token:
+            return jsonify({
+                "success": False,
+                "message": "未提供访问令牌"
+            }), 400
+        
+        # 如果提供了agent_id，验证特定Agent的令牌
+        if agent_id:
+            agent_path = os.path.join('agents', f"{agent_id}.json")
+            if not os.path.exists(agent_path):
+                return jsonify({
+                    "success": False,
+                    "message": "Agent不存在"
+                }), 404
+            
+            with open(agent_path, 'r', encoding='utf-8') as f:
+                agent_config = json.load(f)
+            
+            # 验证令牌
+            if "distributions" in agent_config:
+                for dist in agent_config["distributions"]:
+                    if dist.get("token") == token:
+                        # 检查是否过期
+                        if dist.get("expires_at", 0) > 0 and dist.get("expires_at", 0) < time.time():
+                            return jsonify({
+                                "success": False,
+                                "message": "访问令牌已过期"
+                            })
+                        
+                        return jsonify({
+                            "success": True,
+                            "agent": {
+                                "id": agent_id,
+                                "name": agent_config.get('name', 'AI学习助手'),
+                                "description": agent_config.get('description', ''),
+                                "subject": agent_config.get('subject', ''),
+                                "instructor": agent_config.get('instructor', '教师')
+                            }
+                        })
+            
+            return jsonify({
+                "success": False,
+                "message": "访问令牌无效"
+            })
+        
+        # 如果没有提供agent_id，搜索所有Agent
+        valid_agent = None
+        
+        for filename in os.listdir('agents'):
+            if filename.endswith('.json'):
+                agent_path = os.path.join('agents', filename)
+                with open(agent_path, 'r', encoding='utf-8') as f:
+                    agent_config = json.load(f)
+                
+                # 验证令牌
+                if "distributions" in agent_config:
+                    for dist in agent_config["distributions"]:
+                        if dist.get("token") == token:
+                            # 检查是否过期
+                            if dist.get("expires_at", 0) > 0 and dist.get("expires_at", 0) < time.time():
+                                continue
+                            
+                            valid_agent = {
+                                "id": agent_config.get('id'),
+                                "name": agent_config.get('name', 'AI学习助手'),
+                                "description": agent_config.get('description', ''),
+                                "subject": agent_config.get('subject', ''),
+                                "instructor": agent_config.get('instructor', '教师')
+                            }
+                            break
+                
+                if valid_agent:
+                    break
+        
+        if valid_agent:
+            return jsonify({
+                "success": True,
+                "agent": valid_agent
+            })
+        
+        return jsonify({
+            "success": False,
+            "message": "未找到匹配的访问令牌"
+        })
+    
+    except Exception as e:
+        import traceback
+        traceback.print_exc()
+        return jsonify({
+            "success": False,
+            "message": f"验证访问令牌时出错: {str(e)}"
+        }), 500
+
+# API端点：获取学生的Agent列表
+@app.route('/api/student/agents', methods=['GET'])
+@student_required
+def get_student_agents():
+    """获取学生可访问的Agent列表"""
+    try:
+        # 实际应用中应根据学生ID过滤
+        # 这里简化为获取所有Agent
+        agents = []
+        
+        for filename in os.listdir('agents'):
+            if filename.endswith('.json'):
+                agent_path = os.path.join('agents', filename)
+                with open(agent_path, 'r', encoding='utf-8') as f:
+                    agent_config = json.load(f)
+                
+                # 简化信息
+                agent_info = {
+                    "id": agent_config.get('id'),
+                    "name": agent_config.get('name', 'AI学习助手'),
+                    "description": agent_config.get('description', ''),
+                    "subject": agent_config.get('subject', ''),
+                    "instructor": agent_config.get('instructor', '教师'),
+                    "plugins": agent_config.get('plugins', []),
+                    "last_used": agent_config.get('stats', {}).get('last_used')
+                }
+                
+                # 添加TOKEN（实际应用中应严格控制令牌访问）
+                if "distributions" in agent_config and agent_config["distributions"]:
+                    # 仅添加第一个分发的令牌
+                    agent_info["token"] = agent_config["distributions"][0].get("token")
+                
+                agents.append(agent_info)
+        
+        # 按最后使用时间排序
+        agents.sort(key=lambda x: x.get('last_used', 0) or 0, reverse=True)
+        
+        return jsonify({
+            "success": True,
+            "agents": agents
+        })
+    
+    except Exception as e:
+        import traceback
+        traceback.print_exc()
+        return jsonify({
+            "success": False,
+            "message": str(e)
+        }), 500
+
+if __name__ == '__main__':
     app.run(debug=True, host='0.0.0.0', port=7860)

modules/agent_builder/routes.py

@@ -1,504 +1,504 @@
-# modules/agent_builder/routes.py
-from flask import Blueprint, request, jsonify
-import os
-import json
-import time
-import uuid
-import requests
-from config import STREAM_API_KEY, STREAM_BASE_URL, DEFAULT_MODEL
-
-agent_builder_bp = Blueprint('agent_builder', __name__)
-
-# 确保Agent存储目录存在
-AGENTS_DIR = 'agents'
-os.makedirs(AGENTS_DIR, exist_ok=True)
-
-@agent_builder_bp.route('/create', methods=['POST'])
-def create_agent():
-    """创建新的Agent"""
-    try:
-        data = request.json
-        name = data.get('name')
-        description = data.get('description', '')
-        subject = data.get('subject', name)  # 默认使用名称作为学科
-        instructor = data.get('instructor', '教师')  # 默认使用教师作为指导者
-        plugins = data.get('plugins', [])
-        knowledge_bases = data.get('knowledge_bases', [])
-        workflow = data.get('workflow')
-        
-        if not name:
-            return jsonify({
-                "success": False,
-                "message": "Agent名称不能为空"
-            }), 400
-        
-        # 创建Agent ID
-        agent_id = f"agent_{uuid.uuid4().hex[:8]}_{int(time.time())}"
-        
-        # 构建Agent配置
-        agent_config = {
-            "id": agent_id,
-            "name": name,
-            "description": description,
-            "subject": subject,  # 添加学科
-            "instructor": instructor,  # 添加指导者
-            "created_at": int(time.time()),
-            "plugins": plugins,
-            "knowledge_bases": knowledge_bases,
-            "workflow": workflow or {
-                "nodes": [],
-                "edges": []
-            },
-            "distributions": [],
-            "stats": {
-                "usage_count": 0,
-                "last_used": None
-            }
-        }
-        
-        # 保存Agent配置
-        with open(os.path.join(AGENTS_DIR, f"{agent_id}.json"), 'w', encoding='utf-8') as f:
-            json.dump(agent_config, f, ensure_ascii=False, indent=2)
-        
-        return jsonify({
-            "success": True,
-            "agent_id": agent_id,
-            "message": f"Agent '{name}' 创建成功"
-        })
-        
-    except Exception as e:
-        import traceback
-        traceback.print_exc()
-        return jsonify({
-            "success": False,
-            "message": str(e)
-        }), 500
-
-@agent_builder_bp.route('/list', methods=['GET'])
-def list_agents():
-    """获取所有Agent列表"""
-    try:
-        agents = []
-        
-        for filename in os.listdir(AGENTS_DIR):
-            if filename.endswith('.json'):
-                with open(os.path.join(AGENTS_DIR, filename), 'r', encoding='utf-8') as f:
-                    agent_config = json.load(f)
-                    
-                    # 简化版本，只返回关键信息
-                    agents.append({
-                        "id": agent_config.get("id"),
-                        "name": agent_config.get("name"),
-                        "description": agent_config.get("description"),
-                        "subject": agent_config.get("subject", agent_config.get("name")),
-                        "instructor": agent_config.get("instructor", "教师"),
-                        "created_at": agent_config.get("created_at"),
-                        "plugins": agent_config.get("plugins", []),
-                        "knowledge_bases": agent_config.get("knowledge_bases", []),
-                        "usage_count": agent_config.get("stats", {}).get("usage_count", 0),
-                        "distribution_count": len(agent_config.get("distributions", []))
-                    })
-        
-        # 按创建时间排序
-        agents.sort(key=lambda x: x.get("created_at", 0), reverse=True)
-        
-        return jsonify({
-            "success": True,
-            "agents": agents
-        })
-        
-    except Exception as e:
-        import traceback
-        traceback.print_exc()
-        return jsonify({
-            "success": False,
-            "message": str(e)
-        }), 500
-
-@agent_builder_bp.route('/<agent_id>', methods=['GET'])
-def get_agent(agent_id):
-    """获取特定Agent的配置"""
-    try:
-        agent_path = os.path.join(AGENTS_DIR, f"{agent_id}.json")
-        
-        if not os.path.exists(agent_path):
-            return jsonify({
-                "success": False,
-                "message": "Agent不存在"
-            }), 404
-        
-        with open(agent_path, 'r', encoding='utf-8') as f:
-            agent_config = json.load(f)
-        
-        return jsonify({
-            "success": True,
-            "agent": agent_config
-        })
-        
-    except Exception as e:
-        import traceback
-        traceback.print_exc()
-        return jsonify({
-            "success": False,
-            "message": str(e)
-        }), 500
-
-@agent_builder_bp.route('/<agent_id>', methods=['PUT'])
-def update_agent(agent_id):
-    """更新Agent配置"""
-    try:
-        data = request.json
-        agent_path = os.path.join(AGENTS_DIR, f"{agent_id}.json")
-        
-        if not os.path.exists(agent_path):
-            return jsonify({
-                "success": False,
-                "message": "Agent不存在"
-            }), 404
-        
-        # 读取现有配置
-        with open(agent_path, 'r', encoding='utf-8') as f:
-            agent_config = json.load(f)
-        
-        # 更新允许的字段
-        if 'name' in data:
-            agent_config['name'] = data['name']
-        
-        if 'description' in data:
-            agent_config['description'] = data['description']
-        
-        if 'plugins' in data:
-            agent_config['plugins'] = data['plugins']
-        
-        if 'knowledge_bases' in data:
-            agent_config['knowledge_bases'] = data['knowledge_bases']
-        
-        if 'workflow' in data:
-            agent_config['workflow'] = data['workflow']
-        
-        # 添加更新时间
-        agent_config['updated_at'] = int(time.time())
-        
-        # 保存更新后的配置
-        with open(agent_path, 'w', encoding='utf-8') as f:
-            json.dump(agent_config, f, ensure_ascii=False, indent=2)
-        
-        return jsonify({
-            "success": True,
-            "message": "Agent更新成功"
-        })
-        
-    except Exception as e:
-        import traceback
-        traceback.print_exc()
-        return jsonify({
-            "success": False,
-            "message": str(e)
-        }), 500
-
-@agent_builder_bp.route('/<agent_id>', methods=['DELETE'])
-def delete_agent(agent_id):
-    """删除Agent"""
-    try:
-        agent_path = os.path.join(AGENTS_DIR, f"{agent_id}.json")
-        
-        if not os.path.exists(agent_path):
-            return jsonify({
-                "success": False,
-                "message": "Agent不存在"
-            }), 404
-        
-        # 删除Agent配置文件
-        os.remove(agent_path)
-        
-        return jsonify({
-            "success": True,
-            "message": "Agent删除成功"
-        })
-        
-    except Exception as e:
-        import traceback
-        traceback.print_exc()
-        return jsonify({
-            "success": False,
-            "message": str(e)
-        }), 500
-
-@agent_builder_bp.route('/<agent_id>/distribute', methods=['POST'])
-def distribute_agent(agent_id):
-    """为Agent创建分发链接"""
-    try:
-        data = request.json
-        expires_in = data.get('expires_in', 0)  # 0表示永不过期
-        
-        agent_path = os.path.join(AGENTS_DIR, f"{agent_id}.json")
-        
-        if not os.path.exists(agent_path):
-            return jsonify({
-                "success": False,
-                "message": "Agent不存在"
-            }), 404
-        
-        # 读取Agent配置
-        with open(agent_path, 'r', encoding='utf-8') as f:
-            agent_config = json.load(f)
-        
-        # 创建访问令牌
-        token = uuid.uuid4().hex
-        
-        # 计算过期时间
-        expiry = int(time.time() + expires_in) if expires_in > 0 else 0
-        
-        # 创建分发记录
-        distribution = {
-            "id": f"dist_{uuid.uuid4().hex[:6]}",
-            "created_at": int(time.time()),
-            "token": token,
-            "expires_at": expiry,
-            "usage_count": 0
-        }
-        
-        # 更新Agent配置
-        if "distributions" not in agent_config:
-            agent_config["distributions"] = []
-            
-        agent_config["distributions"].append(distribution)
-        
-        # 保存更新后的配置
-        with open(agent_path, 'w', encoding='utf-8') as f:
-            json.dump(agent_config, f, ensure_ascii=False, indent=2)
-        
-        # 构建访问链接
-        access_link = f"/student/{agent_id}?token={token}"
-        
-        return jsonify({
-            "success": True,
-            "distribution": {
-                "id": distribution["id"],
-                "link": access_link,
-                "token": token,
-                "expires_at": expiry
-            },
-            "message": "分发链接创建成功"
-        })
-        
-    except Exception as e:
-        import traceback
-        traceback.print_exc()
-        return jsonify({
-            "success": False,
-            "message": str(e)
-        }), 500
-
-@agent_builder_bp.route('/ai-assist', methods=['POST'])
-def ai_assisted_workflow():
-    """使用AI辅助Agent工作流编排"""
-    try:
-        data = request.json
-        description = data.get('description', '')
-        subject = data.get('subject', '通用学科')
-        knowledge_bases = data.get('knowledge_bases', [])
-        plugins = data.get('plugins', [])
-        
-        if not description:
-            return jsonify({
-                "success": False,
-                "message": "请提供Agent描述"
-            }), 400
-            
-        # 获取所有可用的知识库
-        available_knowledge_bases = []
-        try:
-            # ���取知识库列表的API调用
-            kb_response = requests.get("https://samlax12-agent.hf.space/api/knowledge")
-            if kb_response.status_code == 200:
-                kb_data = kb_response.json()
-                if kb_data.get("success"):
-                    for kb in kb_data.get("data", []):
-                        available_knowledge_bases.append(kb["id"])
-        except:
-            # 如果API调用失败，使用默认值
-            pass
-            
-        # 可用的插件
-        available_plugins = ["code", "visualization", "mindmap"]
-        
-        # 构建提示
-        system_prompt = """你是一个专业的AI工作流设计师。你需要根据用户描述，设计一个适合教育场景的Agent工作流。
-        你不仅要设计工作流结构，还要推荐合适的知识库和插件。请确保工作流逻辑合理，能够满足用户的需求。
-        
-        工作流应包含以下类型的节点：
-        1. 意图识别：识别用户输入的意图
-        2. 知识库查询：从指定知识库中检索信息
-        3. 插件调用：调用特定插件（如代码执行、3D可视化或思维导图）
-        4. 回复生成：生成最终回复
-        
-        用户当前选择的知识库：{knowledge_bases}
-        系统中可用的知识库有：{available_knowledge_bases}
-        
-        用户当前选择的插件：{plugins}
-        系统中可用的插件有：{available_plugins}
-        
-        这个Agent的主题领域是：{subject}
-        
-        重要：根据Agent描述，推荐最合适的知识库和插件。在推荐时，只能使用系统中真实可用的知识库和插件。
-        
-        请返回三部分内容：
-        1. 推荐的知识库列表（只能从可用知识库中选择）
-        2. 推荐的插件列表（只能从可用插件中选择）
-        3. 完整的工作流JSON结构
-        
-        JSON格式示例：
-        {{
-          "recommended_knowledge_bases": ["rag_knowledge1", "rag_knowledge2"],
-          "recommended_plugins": ["code", "visualization"],
-          "workflow": {{
-            "nodes": [
-              {{ "id": "node1", "type": "intent_recognition", "data": {{ "name": "意图识别" }} }},
-              {{ "id": "node2", "type": "knowledge_query", "data": {{ "name": "知识库查询", "knowledge_base_id": "rag_knowledge1" }} }},
-              {{ "id": "node3", "type": "plugin_call", "data": {{ "name": "调用代码执行插件", "plugin_id": "code" }} }},
-              {{ "id": "node4", "type": "generate_response", "data": {{ "name": "生成回复" }} }}
-            ],
-            "edges": [
-              {{ "id": "edge1", "source": "node1", "target": "node2", "condition": "需要知识" }},
-              {{ "id": "edge2", "source": "node1", "target": "node3", "condition": "需要代码执行" }},
-              {{ "id": "edge3", "source": "node2", "target": "node4" }},
-              {{ "id": "edge4", "source": "node3", "target": "node4" }}
-            ]
-          }}
-        }}
-        """
-        
-        system_prompt = system_prompt.format(
-            knowledge_bases=", ".join(knowledge_bases) if knowledge_bases else "无",
-            available_knowledge_bases=", ".join(available_knowledge_bases) if available_knowledge_bases else "无可用知识库",
-            plugins=", ".join(plugins) if plugins else "无",
-            available_plugins=", ".join(available_plugins),
-            subject=subject
-        )
-        
-        # 使用流式API
-        try:
-            headers = {
-                "Authorization": f"Bearer {STREAM_API_KEY}",
-                "Content-Type": "application/json"
-            }
-            
-            response = requests.post(
-                f"{STREAM_BASE_URL}/chat/completions",
-                headers=headers,
-                json={
-                    "model": DEFAULT_MODEL,
-                    "messages": [
-                        {"role": "system", "content": system_prompt},
-                        {"role": "user", "content": f"请为以下描述的教育Agent设计工作流并推荐知识库和插件：\n\n{description}"}
-                    ]
-                }
-            )
-            
-            if response.status_code != 200:
-                return jsonify({
-                    "success": False,
-                    "message": f"Error code: {response.status_code} - {response.text}",
-                    "workflow": create_default_workflow(),
-                    "recommended_knowledge_bases": [],
-                    "recommended_plugins": []
-                }), 200  # 返回200但包含错误信息和备用工作流
-            
-            result = response.json()
-            content = result['choices'][0]['message']['content']
-            
-        except Exception as api_error:
-            return jsonify({
-                "success": False,
-                "message": f"无法连接到AI模型服务: {str(api_error)}",
-                "workflow": create_default_workflow(),
-                "recommended_knowledge_bases": [],
-                "recommended_plugins": []
-            }), 200
-        
-        # 查找JSON部分
-        import re
-        json_match = re.search(r'```json\n([\s\S]*?)\n```', content)
-        
-        if json_match:
-            workflow_json = json_match.group(1)
-        else:
-            # 尝试直接解析整个内容
-            workflow_json = content
-        
-        # 解析JSON
-        try:
-            result_data = json.loads(workflow_json)
-        except:
-            # 如果解析失败，使用正则表达式清理
-            workflow_json = re.sub(r'```json\n|\n```', '', content)
-            try:
-                result_data = json.loads(workflow_json)
-            except:
-                # 如果仍然解析失败，提取标准的JSON结构 { ... }
-                import re
-                json_patterns = re.findall(r'\{[\s\S]*?\}', content)
-                if json_patterns:
-                    try:
-                        # 尝试解析最长的JSON结构
-                        longest_json = max(json_patterns, key=len)
-                        result_data = json.loads(longest_json)
-                    except:
-                        # 仍然失败，返回默认值
-                        return jsonify({
-                            "success": True,
-                            "message": "使用默认工作流（AI生成的JSON无效）",
-                            "workflow": create_default_workflow(),
-                            "recommended_knowledge_bases": [],
-                            "recommended_plugins": []
-                        })
-                else:
-                    # 没有找到有效的JSON结构
-                    return jsonify({
-                        "success": True,
-                        "message": "使用默认工作流（未找到JSON结构）",
-                        "workflow": create_default_workflow(),
-                        "recommended_knowledge_bases": [],
-                        "recommended_plugins": []
-                    })
-        
-        # 提取推荐的知识库和插件
-        recommended_knowledge_bases = result_data.get("recommended_knowledge_bases", [])
-        recommended_plugins = result_data.get("recommended_plugins", [])
-        workflow = result_data.get("workflow", create_default_workflow())
-        
-        # 验证推荐的知识库都存在
-        valid_knowledge_bases = []
-        for kb in recommended_knowledge_bases:
-            if kb in available_knowledge_bases:
-                valid_knowledge_bases.append(kb)
-        
-        # 验证推荐的插件都存在
-        valid_plugins = []
-        for plugin in recommended_plugins:
-            if plugin in available_plugins:
-                valid_plugins.append(plugin)
-        
-        return jsonify({
-            "success": True,
-            "workflow": workflow,
-            "recommended_knowledge_bases": valid_knowledge_bases,
-            "recommended_plugins": valid_plugins,
-            "message": "已成功创建工作流"
-        })
-        
-    except Exception as e:
-        import traceback
-        traceback.print_exc()
-        return jsonify({
-            "success": False, 
-            "message": str(e)
-        }), 500
-
-def create_default_workflow():
-    """创建一个默认的空工作流"""
-    return {
-        "nodes": [],
-        "edges": []
+# modules/agent_builder/routes.py
+from flask import Blueprint, request, jsonify
+import os
+import json
+import time
+import uuid
+import requests
+from config import STREAM_API_KEY, STREAM_BASE_URL, DEFAULT_MODEL
+
+agent_builder_bp = Blueprint('agent_builder', __name__)
+
+# 确保Agent存储目录存在
+AGENTS_DIR = 'agents'
+os.makedirs(AGENTS_DIR, exist_ok=True)
+
+@agent_builder_bp.route('/create', methods=['POST'])
+def create_agent():
+    """创建新的Agent"""
+    try:
+        data = request.json
+        name = data.get('name')
+        description = data.get('description', '')
+        subject = data.get('subject', name)  # 默认使用名称作为学科
+        instructor = data.get('instructor', '教师')  # 默认使用教师作为指导者
+        plugins = data.get('plugins', [])
+        knowledge_bases = data.get('knowledge_bases', [])
+        workflow = data.get('workflow')
+        
+        if not name:
+            return jsonify({
+                "success": False,
+                "message": "Agent名称不能为空"
+            }), 400
+        
+        # 创建Agent ID
+        agent_id = f"agent_{uuid.uuid4().hex[:8]}_{int(time.time())}"
+        
+        # 构建Agent配置
+        agent_config = {
+            "id": agent_id,
+            "name": name,
+            "description": description,
+            "subject": subject,  # 添加学科
+            "instructor": instructor,  # 添加指导者
+            "created_at": int(time.time()),
+            "plugins": plugins,
+            "knowledge_bases": knowledge_bases,
+            "workflow": workflow or {
+                "nodes": [],
+                "edges": []
+            },
+            "distributions": [],
+            "stats": {
+                "usage_count": 0,
+                "last_used": None
+            }
+        }
+        
+        # 保存Agent配置
+        with open(os.path.join(AGENTS_DIR, f"{agent_id}.json"), 'w', encoding='utf-8') as f:
+            json.dump(agent_config, f, ensure_ascii=False, indent=2)
+        
+        return jsonify({
+            "success": True,
+            "agent_id": agent_id,
+            "message": f"Agent '{name}' 创建成功"
+        })
+        
+    except Exception as e:
+        import traceback
+        traceback.print_exc()
+        return jsonify({
+            "success": False,
+            "message": str(e)
+        }), 500
+
+@agent_builder_bp.route('/list', methods=['GET'])
+def list_agents():
+    """获取所有Agent列表"""
+    try:
+        agents = []
+        
+        for filename in os.listdir(AGENTS_DIR):
+            if filename.endswith('.json'):
+                with open(os.path.join(AGENTS_DIR, filename), 'r', encoding='utf-8') as f:
+                    agent_config = json.load(f)
+                    
+                    # 简化版本，只返回关键信息
+                    agents.append({
+                        "id": agent_config.get("id"),
+                        "name": agent_config.get("name"),
+                        "description": agent_config.get("description"),
+                        "subject": agent_config.get("subject", agent_config.get("name")),
+                        "instructor": agent_config.get("instructor", "教师"),
+                        "created_at": agent_config.get("created_at"),
+                        "plugins": agent_config.get("plugins", []),
+                        "knowledge_bases": agent_config.get("knowledge_bases", []),
+                        "usage_count": agent_config.get("stats", {}).get("usage_count", 0),
+                        "distribution_count": len(agent_config.get("distributions", []))
+                    })
+        
+        # 按创建时间排序
+        agents.sort(key=lambda x: x.get("created_at", 0), reverse=True)
+        
+        return jsonify({
+            "success": True,
+            "agents": agents
+        })
+        
+    except Exception as e:
+        import traceback
+        traceback.print_exc()
+        return jsonify({
+            "success": False,
+            "message": str(e)
+        }), 500
+
+@agent_builder_bp.route('/<agent_id>', methods=['GET'])
+def get_agent(agent_id):
+    """获取特定Agent的配置"""
+    try:
+        agent_path = os.path.join(AGENTS_DIR, f"{agent_id}.json")
+        
+        if not os.path.exists(agent_path):
+            return jsonify({
+                "success": False,
+                "message": "Agent不存在"
+            }), 404
+        
+        with open(agent_path, 'r', encoding='utf-8') as f:
+            agent_config = json.load(f)
+        
+        return jsonify({
+            "success": True,
+            "agent": agent_config
+        })
+        
+    except Exception as e:
+        import traceback
+        traceback.print_exc()
+        return jsonify({
+            "success": False,
+            "message": str(e)
+        }), 500
+
+@agent_builder_bp.route('/<agent_id>', methods=['PUT'])
+def update_agent(agent_id):
+    """更新Agent配置"""
+    try:
+        data = request.json
+        agent_path = os.path.join(AGENTS_DIR, f"{agent_id}.json")
+        
+        if not os.path.exists(agent_path):
+            return jsonify({
+                "success": False,
+                "message": "Agent不存在"
+            }), 404
+        
+        # 读取现有配置
+        with open(agent_path, 'r', encoding='utf-8') as f:
+            agent_config = json.load(f)
+        
+        # 更新允许的字段
+        if 'name' in data:
+            agent_config['name'] = data['name']
+        
+        if 'description' in data:
+            agent_config['description'] = data['description']
+        
+        if 'plugins' in data:
+            agent_config['plugins'] = data['plugins']
+        
+        if 'knowledge_bases' in data:
+            agent_config['knowledge_bases'] = data['knowledge_bases']
+        
+        if 'workflow' in data:
+            agent_config['workflow'] = data['workflow']
+        
+        # 添加更新时间
+        agent_config['updated_at'] = int(time.time())
+        
+        # 保存更新后的配置
+        with open(agent_path, 'w', encoding='utf-8') as f:
+            json.dump(agent_config, f, ensure_ascii=False, indent=2)
+        
+        return jsonify({
+            "success": True,
+            "message": "Agent更新成功"
+        })
+        
+    except Exception as e:
+        import traceback
+        traceback.print_exc()
+        return jsonify({
+            "success": False,
+            "message": str(e)
+        }), 500
+
+@agent_builder_bp.route('/<agent_id>', methods=['DELETE'])
+def delete_agent(agent_id):
+    """删除Agent"""
+    try:
+        agent_path = os.path.join(AGENTS_DIR, f"{agent_id}.json")
+        
+        if not os.path.exists(agent_path):
+            return jsonify({
+                "success": False,
+                "message": "Agent不存在"
+            }), 404
+        
+        # 删除Agent配置文件
+        os.remove(agent_path)
+        
+        return jsonify({
+            "success": True,
+            "message": "Agent删除成功"
+        })
+        
+    except Exception as e:
+        import traceback
+        traceback.print_exc()
+        return jsonify({
+            "success": False,
+            "message": str(e)
+        }), 500
+
+@agent_builder_bp.route('/<agent_id>/distribute', methods=['POST'])
+def distribute_agent(agent_id):
+    """为Agent创建分发链接"""
+    try:
+        data = request.json
+        expires_in = data.get('expires_in', 0)  # 0表示永不过期
+        
+        agent_path = os.path.join(AGENTS_DIR, f"{agent_id}.json")
+        
+        if not os.path.exists(agent_path):
+            return jsonify({
+                "success": False,
+                "message": "Agent不存在"
+            }), 404
+        
+        # 读取Agent配置
+        with open(agent_path, 'r', encoding='utf-8') as f:
+            agent_config = json.load(f)
+        
+        # 创建访问令牌
+        token = uuid.uuid4().hex
+        
+        # 计算过期时间
+        expiry = int(time.time() + expires_in) if expires_in > 0 else 0
+        
+        # 创建分发记录
+        distribution = {
+            "id": f"dist_{uuid.uuid4().hex[:6]}",
+            "created_at": int(time.time()),
+            "token": token,
+            "expires_at": expiry,
+            "usage_count": 0
+        }
+        
+        # 更新Agent配置
+        if "distributions" not in agent_config:
+            agent_config["distributions"] = []
+            
+        agent_config["distributions"].append(distribution)
+        
+        # 保存更新后的配置
+        with open(agent_path, 'w', encoding='utf-8') as f:
+            json.dump(agent_config, f, ensure_ascii=False, indent=2)
+        
+        # 构建访问链接
+        access_link = f"/student/{agent_id}?token={token}"
+        
+        return jsonify({
+            "success": True,
+            "distribution": {
+                "id": distribution["id"],
+                "link": access_link,
+                "token": token,
+                "expires_at": expiry
+            },
+            "message": "分发链接创建成功"
+        })
+        
+    except Exception as e:
+        import traceback
+        traceback.print_exc()
+        return jsonify({
+            "success": False,
+            "message": str(e)
+        }), 500
+
+@agent_builder_bp.route('/ai-assist', methods=['POST'])
+def ai_assisted_workflow():
+    """使用AI辅助Agent工作流编排"""
+    try:
+        data = request.json
+        description = data.get('description', '')
+        subject = data.get('subject', '通用学科')
+        knowledge_bases = data.get('knowledge_bases', [])
+        plugins = data.get('plugins', [])
+        
+        if not description:
+            return jsonify({
+                "success": False,
+                "message": "请提供Agent描述"
+            }), 400
+            
+        # 获取所有可用的知识库
+        available_knowledge_bases = []
+        try:
+            # 获取知识库列表的API调用
+            kb_response = requests.get("https://samlax12-agent.hf.space/api/knowledge")
+            if kb_response.status_code == 200:
+                kb_data = kb_response.json()
+                if kb_data.get("success"):
+                    for kb in kb_data.get("data", []):
+                        available_knowledge_bases.append(kb["id"])
+        except:
+            # 如果API调用失败，使用默认值
+            pass
+            
+        # 可用的插件
+        available_plugins = ["code", "visualization", "mindmap"]
+        
+        # 构建提示
+        system_prompt = """你是一个专业的AI工作流设计师。你需要根据用户描述，设计一个适合教育场景的Agent工作流。
+        你不仅要设计工作流结构，还要推荐合适的知识库和插件。请确保工作流逻辑合理，能够满足用户的需求。
+        
+        工作流应包含以下类型的节点：
+        1. 意图识别：识别用户输入的意图
+        2. 知识库查询：从指定知识库中检索信息
+        3. 插件调用：调用特定插件（如代码执行、3D可视化或思维导图）
+        4. 回复生成：生成最终回复
+        
+        用户当前选择的知识库：{knowledge_bases}
+        系统中可用的知识库有：{available_knowledge_bases}
+        
+        用户当前选择的插件：{plugins}
+        系统中可用的插件有：{available_plugins}
+        
+        这个Agent的主题领域是：{subject}
+        
+        重要：根据Agent描述，推荐最合适的知识库和插件。在推荐时，只能使用系统中真实可用的知识库和插件。
+        
+        请返回三部分内容：
+        1. 推荐的知识库列表（只能从可用知识库中选择）
+        2. 推荐的插件列表（只能从可用插件中选择）
+        3. 完整的工作流JSON结构
+        
+        JSON格式示例：
+        {{
+          "recommended_knowledge_bases": ["rag_knowledge1", "rag_knowledge2"],
+          "recommended_plugins": ["code", "visualization"],
+          "workflow": {{
+            "nodes": [
+              {{ "id": "node1", "type": "intent_recognition", "data": {{ "name": "意图识别" }} }},
+              {{ "id": "node2", "type": "knowledge_query", "data": {{ "name": "知识库查询", "knowledge_base_id": "rag_knowledge1" }} }},
+              {{ "id": "node3", "type": "plugin_call", "data": {{ "name": "调用代码执行插件", "plugin_id": "code" }} }},
+              {{ "id": "node4", "type": "generate_response", "data": {{ "name": "生成回复" }} }}
+            ],
+            "edges": [
+              {{ "id": "edge1", "source": "node1", "target": "node2", "condition": "需要知识" }},
+              {{ "id": "edge2", "source": "node1", "target": "node3", "condition": "需要代码执行" }},
+              {{ "id": "edge3", "source": "node2", "target": "node4" }},
+              {{ "id": "edge4", "source": "node3", "target": "node4" }}
+            ]
+          }}
+        }}
+        """
+        
+        system_prompt = system_prompt.format(
+            knowledge_bases=", ".join(knowledge_bases) if knowledge_bases else "无",
+            available_knowledge_bases=", ".join(available_knowledge_bases) if available_knowledge_bases else "无可用知识库",
+            plugins=", ".join(plugins) if plugins else "无",
+            available_plugins=", ".join(available_plugins),
+            subject=subject
+        )
+        
+        # 使用流式API
+        try:
+            headers = {
+                "Authorization": f"Bearer {STREAM_API_KEY}",
+                "Content-Type": "application/json"
+            }
+            
+            response = requests.post(
+                f"{STREAM_BASE_URL}/chat/completions",
+                headers=headers,
+                json={
+                    "model": DEFAULT_MODEL,
+                    "messages": [
+                        {"role": "system", "content": system_prompt},
+                        {"role": "user", "content": f"请为以下描述的教育Agent设计工作流并推荐知识库和插件：\n\n{description}"}
+                    ]
+                }
+            )
+            
+            if response.status_code != 200:
+                return jsonify({
+                    "success": False,
+                    "message": f"Error code: {response.status_code} - {response.text}",
+                    "workflow": create_default_workflow(),
+                    "recommended_knowledge_bases": [],
+                    "recommended_plugins": []
+                }), 200  # 返回200但包含错误信息和备用工作流
+            
+            result = response.json()
+            content = result['choices'][0]['message']['content']
+            
+        except Exception as api_error:
+            return jsonify({
+                "success": False,
+                "message": f"无法连接到AI模型服务: {str(api_error)}",
+                "workflow": create_default_workflow(),
+                "recommended_knowledge_bases": [],
+                "recommended_plugins": []
+            }), 200
+        
+        # 查找JSON部分
+        import re
+        json_match = re.search(r'```json\n([\s\S]*?)\n```', content)
+        
+        if json_match:
+            workflow_json = json_match.group(1)
+        else:
+            # 尝试直接解析整个内容
+            workflow_json = content
+        
+        # 解析JSON
+        try:
+            result_data = json.loads(workflow_json)
+        except:
+            # 如果解析失败，使用正则表达式清理
+            workflow_json = re.sub(r'```json\n|\n```', '', content)
+            try:
+                result_data = json.loads(workflow_json)
+            except:
+                # 如果仍然解析失败，提取标准的JSON结构 { ... }
+                import re
+                json_patterns = re.findall(r'\{[\s\S]*?\}', content)
+                if json_patterns:
+                    try:
+                        # 尝试解析最长的JSON结构
+                        longest_json = max(json_patterns, key=len)
+                        result_data = json.loads(longest_json)
+                    except:
+                        # 仍然失败，返回默认值
+                        return jsonify({
+                            "success": True,
+                            "message": "使用默认工作流（AI生成的JSON无效）",
+                            "workflow": create_default_workflow(),
+                            "recommended_knowledge_bases": [],
+                            "recommended_plugins": []
+                        })
+                else:
+                    # 没有找到有效的JSON结构
+                    return jsonify({
+                        "success": True,
+                        "message": "使用默认工作流（未找到JSON结构）",
+                        "workflow": create_default_workflow(),
+                        "recommended_knowledge_bases": [],
+                        "recommended_plugins": []
+                    })
+        
+        # 提取推荐的知识库和插件
+        recommended_knowledge_bases = result_data.get("recommended_knowledge_bases", [])
+        recommended_plugins = result_data.get("recommended_plugins", [])
+        workflow = result_data.get("workflow", create_default_workflow())
+        
+        # 验证推荐的知识库都存在
+        valid_knowledge_bases = []
+        for kb in recommended_knowledge_bases:
+            if kb in available_knowledge_bases:
+                valid_knowledge_bases.append(kb)
+        
+        # 验证推荐的插件都存在
+        valid_plugins = []
+        for plugin in recommended_plugins:
+            if plugin in available_plugins:
+                valid_plugins.append(plugin)
+        
+        return jsonify({
+            "success": True,
+            "workflow": workflow,
+            "recommended_knowledge_bases": valid_knowledge_bases,
+            "recommended_plugins": valid_plugins,
+            "message": "已成功创建工作流"
+        })
+        
+    except Exception as e:
+        import traceback
+        traceback.print_exc()
+        return jsonify({
+            "success": False, 
+            "message": str(e)
+        }), 500
+
+def create_default_workflow():
+    """创建一个默认的空工作流"""
+    return {
+        "nodes": [],
+        "edges": []
     }

modules/knowledge_base/generator.py

@@ -1,396 +1,396 @@
-# modules/knowledge_base/generator.py
-from typing import List, Dict, Generator, Union, Optional, Any
-import requests
-import os
-import json
-import time
-import re
-from dotenv import load_dotenv
-
-load_dotenv()
-
-class Generator:
-    def __init__(self, subject="", instructor=""):
-        # Set defaults if not provided
-        self.subject = subject or "通用学科"
-        self.instructor = instructor or "教师"
-        
-        # 基础API配置
-        self.api_key = os.getenv("API_KEY")
-        self.api_base = os.getenv("BASE_URL")
-        
-        # 流式文本问答配置
-        self.stream_api_key = os.getenv("STREAM_API_KEY")
-        self.stream_api_base = os.getenv("STREAM_BASE_URL")
-        self.stream_model = os.getenv("STREAM_MODEL")
-        
-        # 通用提示词模板 - 填充主题和讲师信息
-        self.system_prompt = self.get_system_prompt_template().format(
-            subject=self.subject,
-            instructor=self.instructor
-        )
-
-    def get_system_prompt_template(self):
-        """返回可定制的系统提示词模板"""
-        return """<system>
-你是一位{subject}课程的智能助教，由{instructor}指导开发。你的目标是帮助学生理解和掌握{subject}课程的关键概念、原理和方法。
-
-<knowledge_base>
-你拥有《{subject}》课程的专业知识库，包含教材内容、课件、习题解析等材料。当回答问题时，你应该优先使用知识库中检索到的相关内容，而不是依赖你的通用知识。
-</knowledge_base>
-
-<role_definition>
-作为{subject}助教，你应该:
-1. 用专业且易于理解的方式解释复杂概念
-2. 提供准确的技术信息和计算示例
-3. 在适当时使用比喻或类比帮助理解
-4. 引导学生思考而不是直接给出所有答案
-5. ⁠提供进一步学习的建议和资源
-</role_definition>
-
-<answering_guidelines>
-当回答问题时，请遵循以下原则：
-1. 先从知识库中检索与问题最相关的内容
-2. 将检索结果整合成连贯、清晰的回答
-3. 保持学术严谨性，确保概念解释和计算过程准确无误
-4. 使用专业术语的同时，确保解释足够通俗易懂
-5. 回答问题时注明知识来源，例如"根据教材第X章..."
-6. 当遇到计算题时，展示完整的计算步骤和思路
-7. 当知识库中没有直接相关内容时，明确告知学生并提供基于可靠原理的解答
-8. 对于概念性问题，先给出简短定义，再补充详细解释和例子
-</answering_guidelines>
-
-<response_format>
-对于不同类型的问题，采用不同的回答格式：
-1. 概念解释类问题：
-   - 先给出简明定义
-   - 提供详细解释
-   - 举例说明
-   - 补充相关知识点连接
-2. 计算类问题：
-   - 明确列出已知条件和所求内容
-   - 说明解题思路和所用公式
-   - 展示详细计算步骤
-   - 给出最终答案并解释其含义
-3. 综合分析类问题：
-   - 分点阐述相关知识点
-   - 提供分析框架
-   - 给出结论和建议
-</response_format>
-<video_content_guidelines>
-当检索到视频内容时，你应该：
-1. 明确告知用户你找到了相关视频资源
-2. 提供视频链接并确保包含时间戳
-3. 简要描述视频内容和主要学习点
-4. 建议用户观看视频以获得可视化理解
-5. 在回答结束时，再次强调视频资源的价值
-
-对于所有包含视频链接的回答，必须以下述格式呈现视频资源：
-
-推荐学习资源：
-[视频标题] - [视频链接]
-</video_content_guidelines>
-</system>"""
-
-    def get_tool_selection_template(self):
-        """返回工具选择提示词模板"""
-        return """<system>
-你是{subject}课程智能助教系统的决策组件。你的唯一任务是判断用户问题类型并决定是否调用知识库工具，以及提取精准的搜索关键词。
-
-<decision_guidelines>
-对于所有涉及{subject}专业知识的问题，必须调用至少一个知识库工具。系统依赖这些知识库提供准确信息，而不是依赖模型的通用知识。
-
-判断标准：
-1. 所有涉及课程概念、原理、计算方法的问题 → 必须调用相关知识库
-2. 所有需要专业解释或例子的问题 → 必须调用相关知识库
-3. 所有学习指导或复习相关的问题 → 必须调用相关知识库
-4. 仅对于纯粹的问候语或与课程无关的闲聊 → 不调用任何工具
-
-当决定调用知识库时，提取的关键词必须满足以下条件：
-1. 准确反映问题的核心主题
-2. 包含{subject}相关的专业术语或技术名词
-3. 去除无关紧要的修饰词
-4. 每个关键词尽量简洁，优先使用专业术语
-5. 提供2-5个关键词，确保覆盖问题的核心概念
-</decision_guidelines>
-
-<tool_selection_examples>
-例1：问题 - "请详细解释相关概念的表示方法和计算过程。"
-判断：需要专业知识解释
-工具选择：教材知识库（包含基础概念和详细解释）
-关键词：["表示方法", "计算过程"]
-
-例2：问题 - "这个概念的工作原理是什么？能给我一个直观的例子吗？"
-判断：需要专业知识解释，且需要直观演示
-工具选择：教材知识库（基础概念）和视频知识库（直观演示）
-关键词：["工作原理", "例子"]
-
-例3：问题 - "你好，今天天气怎么样？"
-判断：与课程无关的闲聊
-工具选择：不调用任何工具
-关键词：[]
-</tool_selection_examples>
-
-<video_knowledge_criteria>
-以下情况应优先选择视频知识库工具：
-1. 用户明确要求视频讲解或视频资料
-2. 问题涉及复杂的步骤或流程，可能需要可视化展示
-3. 问题关于动态过程的理解
-4. 问题涉及图形或结构的理解
-5. 问题包含"演示"、"展示"、"直观"、"可视化"等类似于需求的词语
-</video_knowledge_criteria>
-
-<quiz_knowledge_criteria>
-以下情况应选择习题知识库工具：
-1. 用户明确询问习题、例题或解题方法
-2. 问题是关于如何解决特定类型的问题
-3. 用户寻求考试或作业的帮助
-4. 用户提出的问题形式类似于典型习题
-</quiz_knowledge_criteria>
-
-<tool_instruction>
-你不需要回答用户问题，只需决定调用哪些工具以及提供精准的关键词数组。
-</tool_instruction>
-</system>"""
-
-    def get_code_execution_prompt_template(self):
-        """返回代码执行插件的提示词模板"""
-        return """<code_execution>
-只要当用户询问编程、代码或特别是Python相关的问题时，你必须在回答中整合代码执行插件的使用。
-
-使用代码执行插件的指南：
-1. 创建Python代码示例时，请使用正确的Markdown语法，用```python和```作为代码块的分隔符。
-2. 确保你的代码示例完整、可运行，并附有适当的注释。
-3. 在代码前后提供解释，帮助用户理解代码的功能和原理。
-4. 当代码与用户问题相关时，明确告知用户可以使用代码执行环境运行这段代码。
-5. 对于教学场景，考虑创建循序渐进的代码示例，让用户可以逐步学习和理解。
-
-示例回答格式：
-"这是一个[描述]的Python程序：
-
-```python
-# 你的完整、可运行的代码
-print('Hello, world!')
-```
-
-你可以通过点击'运行'按钮在代码执行环境中运行这段代码。
-如果你想修改代码，只需在编辑器中编辑并再次运行即可。"
-</code_execution>"""
-
-    def get_visualization_prompt_template(self):
-        """返回可视化插件的提示词模板"""
-        return """<visualization>
-当用户询问有关数学图形、函数、几何或需要3D可视化的内容时，你应该在回答中提供一个完整的Python函数来生成3D图形。
-
-使用3D可视化插件的指南：
-1. 你必须提供一个名为create_3d_plot的Python函数，该函数不接受任何参数。
-2. 这个函数应该导入必要的库（主要是numpy as np）。
-3. 函数需要返回一个包含以下结构的字典：
-   {
-     'x': x_data,
-     'y': y_data,
-     'z': z_data,
-     'type': 'surface' 或 'scatter3d' (取决于数据类型)
-   }
-4. 确保你的代码可以直接运行，无需额外修改。
-
-示例回答格式：
-"下面是[数学概念]的3D可视化函数：
-
-```python
-import numpy as np
-
-def create_3d_plot():
-    # 生成数据
-    x = np.linspace(-5, 5, 100)
-    y = np.linspace(-5, 5, 100)
-    X, Y = np.meshgrid(x, y)
-    Z = np.sin(np.sqrt(X**2 + Y**2))
-    
-    return {
-        'x': X.tolist(),
-        'y': Y.tolist(),
-        'z': Z.tolist(),
-        'type': 'surface'
-    }
-```
-
-这个函数创建了[概念描述]的3D图形，你可以观察[关键特征]。"
-</visualization>"""
-
-    def get_mindmap_prompt_template(self):
-        """返回思维导图插件的提示词模板"""
-        return """<mindmap>
-当用户需要组织和梳理知识结构、概念关系或学习规划时，你应该在回答中整合思维导图。
-
-使用思维导图的指南：
-1. 提供一个完整的思维导图结构，使用PlantUML格式。
-2. 使用@startmindmap和@endmindmap标记包裹内容。
-3. 使用星号(*)表示层级：*为中央主题，**为主要主题，***为子主题，****为叶子节点。
-4. 确保思维导图结构清晰、逻辑合理，能够帮助用户理解知识体系。
-
-示例格式：
-@startmindmap
-* 中心主题
-** 主要分支1
-*** 子主题1.1
-**** 叶子节点1.1.1
-*** 子主题1.2
-** 主要分支2
-*** 子主题2.1
-@endmindmap
-
-确保思维导图涵盖主题的关键概念和它们之间的关系，帮助用户建立完整的知识体系。
-</mindmap>"""
-
-    def extract_keywords_with_tools(self, question: str, tools: List[Dict]) -> List[Dict]:
-        """使用工具化架构分析问题，决定使用哪些知识库以及提取关键词"""
-        system_prompt = self.get_tool_selection_template().format(subject=self.subject)
-
-        headers = {
-            "Authorization": f"Bearer {self.stream_api_key}",
-            "Content-Type": "application/json"
-        }
-        
-        response = requests.post(
-            f"{self.stream_api_base}/chat/completions",
-            headers=headers,
-            json={
-                "model": self.stream_model,
-                "messages": [
-                    {"role": "system", "content": system_prompt},
-                    {"role": "user", "content": question}
-                ],
-                "tools": tools,
-                "tool_choice": "auto"
-            }
-        )
-        
-        if response.status_code != 200:
-            raise Exception(f"工具调用出错: {response.text}")
-            
-        response_data = response.json()
-        message = response_data["choices"][0]["message"]
-        
-        # 如果模型决定调用工具
-        if "tool_calls" in message and message["tool_calls"]:
-            return message["tool_calls"]
-        else:
-            # 模型没有调用工具，返回空列表
-            return []
-        
-    def generate_stream(self, query: str, context_docs: List[Dict], process_data: Optional[Dict] = None) -> Generator[Union[str, Dict], None, None]:
-        """流式生成回答 - 用于所有类型的问答"""
-        start_time = time.time()
-        
-        # 构建带有引用标记的上下文
-        context_with_refs = []
-        
-        # 检查是否有图片URL
-        has_images = any(doc['metadata'].get('img_url', '') for doc in context_docs)
-        
-        # 检查查询是否与特定插件相关
-        is_code_related = any(kw in query.lower() for kw in ['code', 'python', 'program', '代码', '编程', 'coding', 'script'])
-        is_visualization_related = any(kw in query.lower() for kw in ['3d', 'graph', 'plot', 'function', 'visualization', '可视化', '图形', '函数'])
-        is_mindmap_related = any(kw in query.lower() for kw in ['mindmap', 'mind map', 'concept map', '思维导图', '概念图', '知识图'])
-        
-        # 处理每个文档
-        for i, doc in enumerate(context_docs, 1):
-            # 直接使用文件名作为来源
-            file_name = doc['metadata'].get('file_name', '未知文件')
-            img_url = doc['metadata'].get('img_url', '')
-            
-            # 构建文档内容，如果有图片URL则包含在内
-            content = doc['content']
-            if img_url:
-                content += f"\n[图片地址: {img_url}]"
-            
-            context_with_refs.append(f"[{i}] {content}\n来源：{file_name}")
-        
-        context = "\n\n".join(context_with_refs)
-        
-        # 增强系统提示词，根据查询内容添加插件特定提示
-        enhanced_system_prompt = self.system_prompt
-        
-        # 如果包含图片，添加图片处理相关指令
-        if has_images:
-            enhanced_system_prompt += """
-此外，如果参考内容中包含图片地址，请在回答中适当引用这些图片信息，并在回答的最后列出所有参考的图片来源。
-"""
-        
-        # 添加插件特定提示
-        if is_code_related:
-            enhanced_system_prompt += "\n\n" + self.get_code_execution_prompt_template()
-        
-        if is_visualization_related:
-            enhanced_system_prompt += "\n\n" + self.get_visualization_prompt_template()
-            
-        if is_mindmap_related:
-            enhanced_system_prompt += "\n\n" + self.get_mindmap_prompt_template()
-        
-        # 流式API
-        headers = {
-            "Authorization": f"Bearer {self.stream_api_key}",
-            "Content-Type": "application/json"
-        }
-        
-        try:
-            response = requests.post(
-                f"{self.stream_api_base}/chat/completions",
-                headers=headers,
-                json={
-                    "model": self.stream_model,
-                    "messages": [
-                        {"role": "system", "content": enhanced_system_prompt},
-                        {"role": "user", "content": f"""
-参考内容：
-{context}
-
-问题：{query}
-
-请按照要求回答问题，包括引用标注和来源列表。
-
-如果在参考内容中找到视频资源，请使用以下格式标记视频链接：
-<video_link>视频链接</video_link>
-
-在回答结束时，请使用以下固定格式列出所有获取的参考内容作为参考来源：
-
-===参考来源开始===
-[1] 摘要内容，"文件名"
-[2] 摘要内容，"文件名"
-===参考来源结束===
-
-如果没有参考内容，则无需包含上述部分。
-"""}
-                    ],
-                    "stream": True
-                }
-            )
-            
-            if response.status_code != 200:
-                yield f"生成回答时出错: {response.text}"
-                return
-                
-            # 返回流式响应
-            for line in response.iter_lines():
-                if not line:
-                    continue
-                    
-                line_text = line.decode('utf-8')
-                if line_text.startswith('data: ') and line_text != 'data: [DONE]':
-                    try:
-                        json_str = line_text[6:]  # 移除 "data: " 前缀
-                        data = json.loads(json_str)
-                        content = data.get('choices', [{}])[0].get('delta', {}).get('content', '')
-                        if content:
-                            yield content
-                    except Exception as e:
-                        yield f"解析响应出错: {str(e)}"
-            
-            # 如果需要返回处理数据
-            if process_data:
-                process_data["generation"]["time"] = round(time.time() - start_time, 3)
-                yield {"process_data": process_data}
-                
-        except Exception as e:
-            yield f"连接错误: {str(e)}"
+# modules/knowledge_base/generator.py
+from typing import List, Dict, Generator, Union, Optional, Any
+import requests
+import os
+import json
+import time
+import re
+from dotenv import load_dotenv
+
+load_dotenv()
+
+class Generator:
+    def __init__(self, subject="", instructor=""):
+        # Set defaults if not provided
+        self.subject = subject or "通用学科"
+        self.instructor = instructor or "教师"
+        
+        # 基础API配置
+        self.api_key = os.getenv("API_KEY")
+        self.api_base = os.getenv("BASE_URL")
+        
+        # 流式文本问答配置
+        self.stream_api_key = os.getenv("STREAM_API_KEY")
+        self.stream_api_base = os.getenv("STREAM_BASE_URL")
+        self.stream_model = os.getenv("STREAM_MODEL")
+        
+        # 通用提示词模板 - 填充主题和讲师信息
+        self.system_prompt = self.get_system_prompt_template().format(
+            subject=self.subject,
+            instructor=self.instructor
+        )
+
+    def get_system_prompt_template(self):
+        """返回可定制的系统提示词模板"""
+        return """<system>
+你是一位{subject}课程的智能助教，由{instructor}指导开发。你的目标是帮助学生理解和掌握{subject}课程的关键概念、原理和方法。
+
+<knowledge_base>
+你拥有《{subject}》课程的专业知识库，包含教材内容、课件、习题解析等材料。当回答问题时，你应该优先使用知识库中检索到的相关内容，而不是依赖你的通用知识。
+</knowledge_base>
+
+<role_definition>
+作为{subject}助教，你应该:
+1. 用专业且易于理解的方式解释复杂概念
+2. 提供准确的技术信息和计算示例
+3. 在适当时使用比喻或类比帮助理解
+4. 引导学生思考而不是直接给出所有答案
+5. ⁠提供进一步学习的建议和资源
+</role_definition>
+
+<answering_guidelines>
+当回答问题时，请遵循以下原则：
+1. 先从知识库中检索与问题最相关的内容
+2. 将检索结果整合成连贯、清晰的回答
+3. 保持学术严谨性，确保概念解释和计算过程准确无误
+4. 使用专业术语的同时，确保解释足够通俗易懂
+5. 回答问题时注明知识来源，例如"根据教材第X章..."
+6. 当遇到计算题时，展示完整的计算步骤和思路
+7. 当知识库中没有直接相关内容时，明确告知学生并提供基于可靠原理的解答
+8. 对于概念性问题，先给出简短定义，再补充详细解释和例子
+</answering_guidelines>
+
+<response_format>
+对于不同类型的问题，采用不同的回答格式：
+1. 概念解释类问题：
+   - 先给出简明定义
+   - 提供详细解释
+   - 举例说明
+   - 补充相关知识点连接
+2. 计算类问题：
+   - 明确列出已知条件和所求内容
+   - 说明解题思路和所用公式
+   - 展示详细计算步骤
+   - 给出最终答案并解释其含义
+3. 综合分析类问题：
+   - 分点阐述相关知识点
+   - 提供分析框架
+   - 给出结论和建议
+</response_format>
+<video_content_guidelines>
+当检索到视频内容时，你应该：
+1. 明确告知用户你找到了相关视频资源
+2. 提供视频链接并确保包含时间戳
+3. 简要描述视频内容和主要学习点
+4. 建议用户观看视频以获得可视化理解
+5. 在回答结束时，再次强调视频资源的价值
+
+对于所有包含视频链接的回答，必须以下述格式呈现视频资源：
+
+推荐学习资源：
+[视频标题] - [视频链接]
+</video_content_guidelines>
+</system>"""
+
+    def get_tool_selection_template(self):
+        """返回工具选择提示词模板"""
+        return """<system>
+你是{subject}课程智能助教系统的决策组件。你的唯一任务是判断用户问题类型并决定是否调用知识库工具，以及提取精准的搜索关键词。
+
+<decision_guidelines>
+对于所有涉及{subject}专业知识的问题，必须调用至少一个知识库工具。系统依赖这些知识库提供准确信息，而不是依赖模型的通用知识。
+
+判断标准：
+1. 所有涉及课程概念、原理、计算方法的问题 → 必须调用相关知识库
+2. 所有需要专业解释或例子的问题 → 必须调用相关知识库
+3. 所有学习指导或复习相关的问题 → 必须调用相关知识库
+4. 仅对于纯粹的问候语或与课程无关的闲聊 → 不调用任何工具
+
+当决定调用知识库时，提取的关键词必须满足以下条件：
+1. 准确反映问题的核心主题
+2. 包含{subject}相关的专业术语或技术名词
+3. 去除无关紧要的修饰词
+4. 每个关键词尽量简洁，优先使用专业术语
+5. 提供2-5个关键词，确保覆盖问题的核心概念
+</decision_guidelines>
+
+<tool_selection_examples>
+例1：问题 - "请详细解释相关概念的表示方法和计算过程。"
+判断：需要专业知识解释
+工具选择：教材知识库（包含基础概念和详细解释）
+关键词：["表示方法", "计算过程"]
+
+例2：问题 - "这个概念的工作原理是什么？能给我一个直观的例子吗？"
+判断：需要专业知识解释，且需要直观演示
+工具选择：教材知识库（基础概念）和视频知识库（直观演示）
+关键词：["工作原理", "例子"]
+
+例3：问题 - "你好，今天天气怎么样？"
+判断：与课程无关的闲聊
+工具选择：不调用任何工具
+关键词：[]
+</tool_selection_examples>
+
+<video_knowledge_criteria>
+以下情况应优先选择视频知识库工具：
+1. 用户明确要求视频讲解或视频资料
+2. 问题涉及复杂的步骤或流程，可能需要可视化展示
+3. 问题关于动态过程的理解
+4. 问题涉及图形或结构的理解
+5. 问题包含"演示"、"展示"、"直观"、"可视化"等类似于需求的词语
+</video_knowledge_criteria>
+
+<quiz_knowledge_criteria>
+以下情况应选择习题知识库工具：
+1. 用户明确询问习题、例题或解题方法
+2. 问题是关于如何解决特定类型的问题
+3. 用户寻求考试或作业的帮助
+4. 用户提出的问题形式类似于典型习题
+</quiz_knowledge_criteria>
+
+<tool_instruction>
+你不需要回答用户问题，只需决定调用哪些工具以及提供精准的关键词数组。
+</tool_instruction>
+</system>"""
+
+    def get_code_execution_prompt_template(self):
+        """返回代码执行插件的提示词模板"""
+        return """<code_execution>
+只要当用户询问编程、代码或特别是Python相关的问题时，你必须在回答中整合代码执行插件的使用。
+
+使用代码执行插件的指南：
+1. 创建Python代码示例时，请使用正确的Markdown语法，用```python和```作为代码块的分隔符。
+2. 确保你的代码示例完整、可运行，并附有适当的注释。
+3. 在代码前后提供解释，帮助用户理解代码的功能和原理。
+4. 当代码与用户问题相关时，明确告知用户可以使用代码执行环境运行这段代码。
+5. 对于教学场景，考虑创建循序渐进的代码示例，让用户可以逐步学习和理解。
+
+示例回答格式：
+"这是一个[描述]的Python程序：
+
+```python
+# 你的完整、可运行的代码
+print('Hello, world!')
+```
+
+你可以通过点击'运行'按钮在代码执行环境中运行这段代码。
+如果你想修改代码，只需在编辑器中编辑并再次运行即可。"
+</code_execution>"""
+
+    def get_visualization_prompt_template(self):
+        """返回可视化插件的提示词模板"""
+        return """<visualization>
+当用户询问有关数学图形、函数、几何或需要3D可视化的内容时，你应该在回答中提供一个完整的Python函数来生成3D图形。
+
+使用3D可视化插件的指南：
+1. 你必须提供一个名为create_3d_plot的Python函数，该函数不接受任何参数。
+2. 这个函数应该导入必要的库（主要是numpy as np）。
+3. 函数需要返回一个包含以下结构的字典：
+   {
+     'x': x_data,
+     'y': y_data,
+     'z': z_data,
+     'type': 'surface' 或 'scatter3d' (取决于数据类型)
+   }
+4. 确保你的代码可以直接运行，无需额外修改。
+
+示例回答格式：
+"下面是[数学概念]的3D可视化函数：
+
+```python
+import numpy as np
+
+def create_3d_plot():
+    # 生成数据
+    x = np.linspace(-5, 5, 100)
+    y = np.linspace(-5, 5, 100)
+    X, Y = np.meshgrid(x, y)
+    Z = np.sin(np.sqrt(X**2 + Y**2))
+    
+    return {
+        'x': X.tolist(),
+        'y': Y.tolist(),
+        'z': Z.tolist(),
+        'type': 'surface'
+    }
+```
+
+这个函数创建了[概念描述]的3D图形，你可以观察[关键特征]。"
+</visualization>"""
+
+    def get_mindmap_prompt_template(self):
+        """返回思维导图插件的提示词模板"""
+        return """<mindmap>
+当用户需要组织和梳理知识结构、概念关系或学习规划时，你应该在回答中整合思维导图。
+
+使用思维导图的指南：
+1. 提供一个完整的思维导图结构，使用PlantUML格式。
+2. 使用@startmindmap和@endmindmap标记包裹内容。
+3. 使用星号(*)表示层级：*为中央主题，**为主要主题，***为子主题，****为叶子节点。
+4. 确保思维导图结构清晰、逻辑合理，能够帮助用户理解知识体系。
+
+示例格式：
+@startmindmap
+* 中心主题
+** 主要分支1
+*** 子主题1.1
+**** 叶子节点1.1.1
+*** 子主题1.2
+** 主要分支2
+*** 子主题2.1
+@endmindmap
+
+确保思维导图涵盖主题的关键概念和它们之间的关系，帮助用户建立完整的知识体系。
+</mindmap>"""
+
+    def extract_keywords_with_tools(self, question: str, tools: List[Dict]) -> List[Dict]:
+        """使用工具化架构分析问题，决定使用哪些知识库以及提取关键词"""
+        system_prompt = self.get_tool_selection_template().format(subject=self.subject)
+
+        headers = {
+            "Authorization": f"Bearer {self.stream_api_key}",
+            "Content-Type": "application/json"
+        }
+        
+        response = requests.post(
+            f"{self.stream_api_base}/chat/completions",
+            headers=headers,
+            json={
+                "model": self.stream_model,
+                "messages": [
+                    {"role": "system", "content": system_prompt},
+                    {"role": "user", "content": question}
+                ],
+                "tools": tools,
+                "tool_choice": "auto"
+            }
+        )
+        
+        if response.status_code != 200:
+            raise Exception(f"工具调用出错: {response.text}")
+            
+        response_data = response.json()
+        message = response_data["choices"][0]["message"]
+        
+        # 如果模型决定调用工具
+        if "tool_calls" in message and message["tool_calls"]:
+            return message["tool_calls"]
+        else:
+            # 模型没有调用工具，返回空列表
+            return []
+        
+    def generate_stream(self, query: str, context_docs: List[Dict], process_data: Optional[Dict] = None) -> Generator[Union[str, Dict], None, None]:
+        """流式生成回答 - 用于所有类型的问答"""
+        start_time = time.time()
+        
+        # 构建带有引用标记的上下文
+        context_with_refs = []
+        
+        # 检查是否有图片URL
+        has_images = any(doc['metadata'].get('img_url', '') for doc in context_docs)
+        
+        # 检查查询是否与特定插件相关
+        is_code_related = any(kw in query.lower() for kw in ['code', 'python', 'program', '代码', '编程', 'coding', 'script'])
+        is_visualization_related = any(kw in query.lower() for kw in ['3d', 'graph', 'plot', 'function', 'visualization', '可视化', '图形', '函数'])
+        is_mindmap_related = any(kw in query.lower() for kw in ['mindmap', 'mind map', 'concept map', '思维导图', '概念图', '知识图'])
+        
+        # 处理每个文档
+        for i, doc in enumerate(context_docs, 1):
+            # 直接使用文件名作为来源
+            file_name = doc['metadata'].get('file_name', '未知文件')
+            img_url = doc['metadata'].get('img_url', '')
+            
+            # 构建文档内容，如果有图片URL则包含在内
+            content = doc['content']
+            if img_url:
+                content += f"\n[图片地址: {img_url}]"
+            
+            context_with_refs.append(f"[{i}] {content}\n来源：{file_name}")
+        
+        context = "\n\n".join(context_with_refs)
+        
+        # 增强系统提示词，根据查询内容添加插件特定提示
+        enhanced_system_prompt = self.system_prompt
+        
+        # 如果包含图片，添加图片处理相关指令
+        if has_images:
+            enhanced_system_prompt += """
+此外，如果参考内容中包含图片地址，请在回答中适当引用这些图片信息，并在回答的最后列出所有参考的图片来源。
+"""
+        
+        # 添加插件特定提示
+        if is_code_related:
+            enhanced_system_prompt += "\n\n" + self.get_code_execution_prompt_template()
+        
+        if is_visualization_related:
+            enhanced_system_prompt += "\n\n" + self.get_visualization_prompt_template()
+            
+        if is_mindmap_related:
+            enhanced_system_prompt += "\n\n" + self.get_mindmap_prompt_template()
+        
+        # 流式API
+        headers = {
+            "Authorization": f"Bearer {self.stream_api_key}",
+            "Content-Type": "application/json"
+        }
+        
+        try:
+            response = requests.post(
+                f"{self.stream_api_base}/chat/completions",
+                headers=headers,
+                json={
+                    "model": self.stream_model,
+                    "messages": [
+                        {"role": "system", "content": enhanced_system_prompt},
+                        {"role": "user", "content": f"""
+参考内容：
+{context}
+
+问题：{query}
+
+请按照要求回答问题，包括引用标注和来源列表。
+
+如果在参考内容中找到视频资源，请使用以下格式标记视频链接：
+<video_link>视频链接</video_link>
+
+在回答结束时，请使用以下固定格式列出所有获取的参考内容作为参考来源：
+
+===参考来源开始===
+[1] 摘要内容，"文件名"
+[2] 摘要内容，"文件名"
+===参考来源结束===
+
+如果没有参考内容，则无需包含上述部分。
+"""}
+                    ],
+                    "stream": True
+                }
+            )
+            
+            if response.status_code != 200:
+                yield f"生成回答时出错: {response.text}"
+                return
+                
+            # 返回流式响应
+            for line in response.iter_lines():
+                if not line:
+                    continue
+                    
+                line_text = line.decode('utf-8')
+                if line_text.startswith('data: ') and line_text != 'data: [DONE]':
+                    try:
+                        json_str = line_text[6:]  # 移除 "data: " 前缀
+                        data = json.loads(json_str)
+                        content = data.get('choices', [{}])[0].get('delta', {}).get('content', '')
+                        if content:
+                            yield content
+                    except Exception as e:
+                        yield f"解析响应出错: {str(e)}"
+            
+            # 如果需要返回处理数据
+            if process_data:
+                process_data["generation"]["time"] = round(time.time() - start_time, 3)
+                yield {"process_data": process_data}
+                
+        except Exception as e:
+            yield f"连接错误: {str(e)}"

modules/knowledge_base/processor.py

@@ -1,201 +1,201 @@
-from typing import List, Dict, Callable, Optional
-from langchain.text_splitter import RecursiveCharacterTextSplitter
-from langchain_community.document_loaders import (
-    DirectoryLoader,
-    UnstructuredMarkdownLoader,
-    PyPDFLoader,
-    TextLoader
-)
-import os
-import requests
-import base64
-from PIL import Image
-import io
-
-class DocumentLoader:
-    """通用文档加载器"""
-    def __init__(self, file_path: str):
-        self.file_path = file_path
-        self.extension = os.path.splitext(file_path)[1].lower()
-        self.api_key = os.getenv("API_KEY")
-        self.api_base = os.getenv("BASE_URL")
-        
-    def process_image(self, image_path: str) -> str:
-        """使用 SiliconFlow VLM 模型处理图片"""
-        try:
-            # 读取图片并转换为base64
-            with open(image_path, 'rb') as image_file:
-                image_data = image_file.read()
-                base64_image = base64.b64encode(image_data).decode('utf-8')
-            
-            # 调用 SiliconFlow API
-            headers = {
-                "Authorization": f"Bearer {self.api_key}",
-                "Content-Type": "application/json"
-            }
-            
-            response = requests.post(
-                f"{self.api_base}/chat/completions",
-                headers=headers,
-                json={
-                    "model": "Qwen/Qwen2.5-VL-72B-Instruct",
-                    "messages": [
-                        {
-                            "role": "user",
-                            "content": [
-                                {
-                                    "type": "image_url",
-                                    "image_url": {
-                                        "url": f"data:image/jpeg;base64,{base64_image}",
-                                        "detail": "high"
-                                    }
-                                },
-                                {
-                                    "type": "text",
-                                    "text": "请详细描述这张图片的内容，包括主要对象、场景、活动、颜色、布局等关键信息。"
-                                }
-                            ]
-                        }
-                    ],
-                    "temperature": 0.7,
-                    "max_tokens": 500
-                }
-            )
-            
-            if response.status_code != 200:
-                raise Exception(f"图片处理API调用失败: {response.text}")
-                
-            description = response.json()["choices"][0]["message"]["content"]
-            return description
-            
-        except Exception as e:
-            print(f"处理图片时出错: {str(e)}")
-            return "图片处理失败"
-    
-    def load(self):
-        try:
-            if self.extension == '.md':
-                loader = UnstructuredMarkdownLoader(self.file_path, encoding='utf-8')
-                return loader.load()
-            elif self.extension == '.pdf':
-                loader = PyPDFLoader(self.file_path)
-                return loader.load()
-            elif self.extension == '.txt':
-                loader = TextLoader(self.file_path, encoding='utf-8')
-                return loader.load()
-            elif self.extension in ['.png', '.jpg', '.jpeg', '.gif', '.bmp']:
-                # 处理图片
-                description = self.process_image(self.file_path)
-                # 创建一个包含图片描述的文档
-                from langchain.schema import Document
-                doc = Document(
-                    page_content=description,
-                    metadata={
-                        'source': self.file_path,
-                        'img_url': os.path.abspath(self.file_path)  # 存储图片的绝对路径
-                    }
-                )
-                return [doc]
-            else:
-                raise ValueError(f"不支持的文件格式: {self.extension}")
-                
-        except UnicodeDecodeError:
-            # 如果 utf-8 失败，尝试 gbk
-            if self.extension in ['.md', '.txt']:
-                loader = TextLoader(self.file_path, encoding='gbk')
-                return loader.load()
-            raise
-
-class DocumentProcessor:
-    def __init__(self):
-        self.text_splitter = RecursiveCharacterTextSplitter(
-            chunk_size=1000,
-            chunk_overlap=200,
-            length_function=len,
-        )
-        
-    def get_index_name(self, path: str) -> str:
-        """根据文件路径生成索引名称"""
-        if os.path.isdir(path):
-            # 如果是目录，使用目录名
-            return f"rag_{os.path.basename(path).lower()}"
-        else:
-            # 如果是文件，使用文件名（不含扩展名）
-            return f"rag_{os.path.splitext(os.path.basename(path))[0].lower()}"
-        
-    def process(self, path: str, progress_callback: Optional[Callable] = None) -> List[Dict]:
-        """
-        加载并处理文档，支持目录或单个文件
-        参数：
-            path: 文档路径
-            progress_callback: 进度回调函数，用于报告处理进度
-        返回：处理后的文档列表
-        """
-        if os.path.isdir(path):
-            documents = []
-            total_files = sum([len(files) for _, _, files in os.walk(path)])
-            processed_files = 0
-            processed_size = 0
-            
-            for root, _, files in os.walk(path):
-                for file in files:
-                    file_path = os.path.join(root, file)
-                    try:
-                        # 更新处理进度
-                        if progress_callback:
-                            file_size = os.path.getsize(file_path)
-                            processed_size += file_size
-                            processed_files += 1
-                            progress_callback(processed_size, f"处理文件 {processed_files}/{total_files}: {file}")
-                            
-                        loader = DocumentLoader(file_path)
-                        docs = loader.load()
-                        # 添加文件名到metadata
-                        for doc in docs:
-                            doc.metadata['file_name'] = os.path.basename(file_path)
-                        documents.extend(docs)
-                    except Exception as e:
-                        print(f"警告：加载文件 {file_path} 时出错: {str(e)}")
-                        continue
-        else:
-            try:
-                if progress_callback:
-                    file_size = os.path.getsize(path)
-                    progress_callback(file_size * 0.3, f"加载文件: {os.path.basename(path)}")
-                    
-                loader = DocumentLoader(path)
-                documents = loader.load()
-                
-                # 更新进度
-                if progress_callback:
-                    progress_callback(file_size * 0.6, f"处理文件内容...")
-                    
-                # 添加文件名到metadata
-                file_name = os.path.basename(path)
-                for doc in documents:
-                    doc.metadata['file_name'] = file_name
-            except Exception as e:
-                print(f"加载文件时出错: {str(e)}")
-                raise
-        
-        # 分块
-        chunks = self.text_splitter.split_documents(documents)
-        
-        # 更新进度
-        if progress_callback:
-            if os.path.isdir(path):
-                progress_callback(processed_size, f"文档分块完成，共{len(chunks)}个文档片段")
-            else:
-                file_size = os.path.getsize(path)
-                progress_callback(file_size * 0.9, f"文档分块完成，共{len(chunks)}个文档片段")
-        
-        # 处理成统一格式
-        processed_docs = []
-        for i, chunk in enumerate(chunks):
-            processed_docs.append({
-                'id': f'doc_{i}',
-                'content': chunk.page_content,
-                'metadata': chunk.metadata
-            })
-            
+from typing import List, Dict, Callable, Optional
+from langchain.text_splitter import RecursiveCharacterTextSplitter
+from langchain_community.document_loaders import (
+    DirectoryLoader,
+    UnstructuredMarkdownLoader,
+    PyPDFLoader,
+    TextLoader
+)
+import os
+import requests
+import base64
+from PIL import Image
+import io
+
+class DocumentLoader:
+    """通用文档加载器"""
+    def __init__(self, file_path: str):
+        self.file_path = file_path
+        self.extension = os.path.splitext(file_path)[1].lower()
+        self.api_key = os.getenv("API_KEY")
+        self.api_base = os.getenv("BASE_URL")
+        
+    def process_image(self, image_path: str) -> str:
+        """使用 SiliconFlow VLM 模型处理图片"""
+        try:
+            # 读取图片并转换为base64
+            with open(image_path, 'rb') as image_file:
+                image_data = image_file.read()
+                base64_image = base64.b64encode(image_data).decode('utf-8')
+            
+            # 调用 SiliconFlow API
+            headers = {
+                "Authorization": f"Bearer {self.api_key}",
+                "Content-Type": "application/json"
+            }
+            
+            response = requests.post(
+                f"{self.api_base}/chat/completions",
+                headers=headers,
+                json={
+                    "model": "Qwen/Qwen2.5-VL-72B-Instruct",
+                    "messages": [
+                        {
+                            "role": "user",
+                            "content": [
+                                {
+                                    "type": "image_url",
+                                    "image_url": {
+                                        "url": f"data:image/jpeg;base64,{base64_image}",
+                                        "detail": "high"
+                                    }
+                                },
+                                {
+                                    "type": "text",
+                                    "text": "请详细描述这张图片的内容，包括主要对象、场景、活动、颜色、布局等关键信息。"
+                                }
+                            ]
+                        }
+                    ],
+                    "temperature": 0.7,
+                    "max_tokens": 500
+                }
+            )
+            
+            if response.status_code != 200:
+                raise Exception(f"图片处理API调用失败: {response.text}")
+                
+            description = response.json()["choices"][0]["message"]["content"]
+            return description
+            
+        except Exception as e:
+            print(f"处理图片时出错: {str(e)}")
+            return "图片处理失败"
+    
+    def load(self):
+        try:
+            if self.extension == '.md':
+                loader = UnstructuredMarkdownLoader(self.file_path, encoding='utf-8')
+                return loader.load()
+            elif self.extension == '.pdf':
+                loader = PyPDFLoader(self.file_path)
+                return loader.load()
+            elif self.extension == '.txt':
+                loader = TextLoader(self.file_path, encoding='utf-8')
+                return loader.load()
+            elif self.extension in ['.png', '.jpg', '.jpeg', '.gif', '.bmp']:
+                # 处理图片
+                description = self.process_image(self.file_path)
+                # 创建一个包含图片描述的文档
+                from langchain.schema import Document
+                doc = Document(
+                    page_content=description,
+                    metadata={
+                        'source': self.file_path,
+                        'img_url': os.path.abspath(self.file_path)  # 存储图片的绝对路径
+                    }
+                )
+                return [doc]
+            else:
+                raise ValueError(f"不支持的文件格式: {self.extension}")
+                
+        except UnicodeDecodeError:
+            # 如果 utf-8 失败，尝试 gbk
+            if self.extension in ['.md', '.txt']:
+                loader = TextLoader(self.file_path, encoding='gbk')
+                return loader.load()
+            raise
+
+class DocumentProcessor:
+    def __init__(self):
+        self.text_splitter = RecursiveCharacterTextSplitter(
+            chunk_size=1000,
+            chunk_overlap=200,
+            length_function=len,
+        )
+        
+    def get_index_name(self, path: str) -> str:
+        """根据文件路径生成索引名称"""
+        if os.path.isdir(path):
+            # 如果是目录，使用目录名
+            return f"rag_{os.path.basename(path).lower()}"
+        else:
+            # 如果是文件，使用文件名（不含扩展名）
+            return f"rag_{os.path.splitext(os.path.basename(path))[0].lower()}"
+        
+    def process(self, path: str, progress_callback: Optional[Callable] = None) -> List[Dict]:
+        """
+        加载并处理文档，支持目录或单个文件
+        参数：
+            path: 文档路径
+            progress_callback: 进度回调函数，用于报告处理进度
+        返回：处理后的文档列表
+        """
+        if os.path.isdir(path):
+            documents = []
+            total_files = sum([len(files) for _, _, files in os.walk(path)])
+            processed_files = 0
+            processed_size = 0
+            
+            for root, _, files in os.walk(path):
+                for file in files:
+                    file_path = os.path.join(root, file)
+                    try:
+                        # 更新处理进度
+                        if progress_callback:
+                            file_size = os.path.getsize(file_path)
+                            processed_size += file_size
+                            processed_files += 1
+                            progress_callback(processed_size, f"处理文件 {processed_files}/{total_files}: {file}")
+                            
+                        loader = DocumentLoader(file_path)
+                        docs = loader.load()
+                        # 添加文件名到metadata
+                        for doc in docs:
+                            doc.metadata['file_name'] = os.path.basename(file_path)
+                        documents.extend(docs)
+                    except Exception as e:
+                        print(f"警告：加载文件 {file_path} 时出错: {str(e)}")
+                        continue
+        else:
+            try:
+                if progress_callback:
+                    file_size = os.path.getsize(path)
+                    progress_callback(file_size * 0.3, f"加载文件: {os.path.basename(path)}")
+                    
+                loader = DocumentLoader(path)
+                documents = loader.load()
+                
+                # 更新进度
+                if progress_callback:
+                    progress_callback(file_size * 0.6, f"处理文件内容...")
+                    
+                # 添加文件名到metadata
+                file_name = os.path.basename(path)
+                for doc in documents:
+                    doc.metadata['file_name'] = file_name
+            except Exception as e:
+                print(f"加载文件时出错: {str(e)}")
+                raise
+        
+        # 分块
+        chunks = self.text_splitter.split_documents(documents)
+        
+        # 更新进度
+        if progress_callback:
+            if os.path.isdir(path):
+                progress_callback(processed_size, f"文档分块完成，共{len(chunks)}个文档片段")
+            else:
+                file_size = os.path.getsize(path)
+                progress_callback(file_size * 0.9, f"文档分块完成，共{len(chunks)}个文档片段")
+        
+        # 处理成统一格式
+        processed_docs = []
+        for i, chunk in enumerate(chunks):
+            processed_docs.append({
+                'id': f'doc_{i}',
+                'content': chunk.page_content,
+                'metadata': chunk.metadata
+            })
+            
         return processed_docs

modules/knowledge_base/reranker.py

@@ -1,49 +1,49 @@
-from typing import List, Dict
-import requests
-import time
-from dotenv import load_dotenv
-import os
-
-load_dotenv()
-
-class Reranker:
-    def __init__(self):
-        self.api_key = os.getenv("API_KEY")
-        self.api_base = os.getenv("BASE_URL")
-        
-    def rerank(self, query: str, documents: List[Dict], index_name: str, top_k: int = 5) -> List[Dict]:
-        """使用SiliconFlow的rerank API重排序文档"""
-        headers = {
-            "Authorization": f"Bearer {self.api_key}",
-            "Content-Type": "application/json"
-        }
-        
-        # 准备文档列表
-        docs = [doc['content'] for doc in documents]
-        
-        response = requests.post(
-            f"{self.api_base}/rerank",
-            headers=headers,
-            json={
-                "model": "BAAI/bge-reranker-v2-m3",
-                "query": query,
-                "documents": docs,
-                "top_n": top_k
-            }
-        )
-        
-        if response.status_code != 200:
-            raise Exception(f"Error in reranking: {response.text}")
-            
-        # 处理结果
-        results = response.json()["results"]
-        reranked_docs = []
-        
-        for result in results:
-            doc_index = result["index"]
-            original_doc = documents[doc_index].copy()
-            original_doc['rerank_score'] = result["relevance_score"]
-            original_doc['index_name'] = index_name  
-            reranked_docs.append(original_doc)
-            
+from typing import List, Dict
+import requests
+import time
+from dotenv import load_dotenv
+import os
+
+load_dotenv()
+
+class Reranker:
+    def __init__(self):
+        self.api_key = os.getenv("API_KEY")
+        self.api_base = os.getenv("BASE_URL")
+        
+    def rerank(self, query: str, documents: List[Dict], index_name: str, top_k: int = 5) -> List[Dict]:
+        """使用SiliconFlow的rerank API重排序文档"""
+        headers = {
+            "Authorization": f"Bearer {self.api_key}",
+            "Content-Type": "application/json"
+        }
+        
+        # 准备文档列表
+        docs = [doc['content'] for doc in documents]
+        
+        response = requests.post(
+            f"{self.api_base}/rerank",
+            headers=headers,
+            json={
+                "model": "BAAI/bge-reranker-v2-m3",
+                "query": query,
+                "documents": docs,
+                "top_n": top_k
+            }
+        )
+        
+        if response.status_code != 200:
+            raise Exception(f"Error in reranking: {response.text}")
+            
+        # 处理结果
+        results = response.json()["results"]
+        reranked_docs = []
+        
+        for result in results:
+            doc_index = result["index"]
+            original_doc = documents[doc_index].copy()
+            original_doc['rerank_score'] = result["relevance_score"]
+            original_doc['index_name'] = index_name  
+            reranked_docs.append(original_doc)
+            
         return reranked_docs

modules/knowledge_base/retriever.py

@@ -1,109 +1,109 @@
-from typing import List, Dict, Tuple
-import requests
-from elasticsearch import Elasticsearch
-import os
-import time
-from dotenv import load_dotenv
-
-load_dotenv()
-
-class Retriever:
-    def __init__(self):
-        # 使用与 vector_store.py 相同的 ES 配置
-        self.es = Elasticsearch(
-            "https://samlax12-elastic.hf.space",  # 注意是 https
-            basic_auth=("elastic", os.getenv("PASSWORD")),  # 使用相同的密码
-            verify_certs=False  # 开发环境可以禁用证书验证
-        )
-        self.api_key = os.getenv("API_KEY")
-        self.api_base = os.getenv("BASE_URL")
-        
-    def get_embedding(self, text: str) -> List[float]:
-        """调用SiliconFlow的embedding API获取向量"""
-        headers = {
-            "Authorization": f"Bearer {self.api_key}",
-            "Content-Type": "application/json"
-        }
-        
-        response = requests.post(
-            f"{self.api_base}/embeddings",
-            headers=headers,
-            json={
-                "model": "BAAI/bge-m3",
-                "input": text
-            }
-        )
-        
-        if response.status_code == 200:
-            return response.json()["data"][0]["embedding"]
-        else:
-            raise Exception(f"Error getting embedding: {response.text}")
-    
-    def get_all_indices(self) -> List[str]:
-        """获取所有 RAG 相关的索引"""
-        indices = self.es.indices.get_alias().keys()
-        return [idx for idx in indices if idx.startswith('rag_')]
-        
-    def retrieve(self, query: str, top_k: int = 10, specific_index: str = None) -> Tuple[List[Dict], str]:
-        """混合检索：结合 BM25 和向量检索，支持指定特定索引"""
-        # 获取检索索引
-        if specific_index:
-            indices = [specific_index] if self.es.indices.exists(index=specific_index) else []
-        else:
-            indices = self.get_all_indices()
-            
-        if not indices:
-            raise Exception("没有找到可用的文档索引！")
-            
-        # 计算查询向量
-        query_vector = self.get_embedding(query)
-        
-        # 在所有索引中搜索
-        all_results = []
-        for index in indices:
-            # 构建混合查询
-            script_query = {
-                "script_score": {
-                    "query": {
-                        "match": {
-                            "content": query  # BM25
-                        }
-                    },
-                    "script": {
-                        "source": "cosineSimilarity(params.query_vector, 'vector') + 1.0",
-                        "params": {"query_vector": query_vector}
-                    }
-                }
-            }
-            
-            # 执行检索
-            response = self.es.search(
-                index=index,
-                body={
-                    "query": script_query,
-                    "size": top_k
-                }
-            )
-            
-            # 处理结果
-            for hit in response['hits']['hits']:
-                result = {
-                    'id': hit['_id'],
-                    'content': hit['_source']['content'],
-                    'score': hit['_score'],
-                    'metadata': hit['_source']['metadata'],
-                    'index': index
-                }
-                all_results.append(result)
-        
-        # 按分数排序并选择最相关的文档
-        all_results.sort(key=lambda x: x['score'], reverse=True)
-        top_results = all_results[:top_k]
-        
-        # 如果有结果，返回最相关文档所在的索引
-        if top_results:
-            most_relevant_index = top_results[0]['index']
-        else:
-            most_relevant_index = indices[0] if indices else ""
-            
+from typing import List, Dict, Tuple
+import requests
+from elasticsearch import Elasticsearch
+import os
+import time
+from dotenv import load_dotenv
+
+load_dotenv()
+
+class Retriever:
+    def __init__(self):
+        # 使用与 vector_store.py 相同的 ES 配置
+        self.es = Elasticsearch(
+            "https://samlax12-elastic.hf.space",  # 注意是 https
+            basic_auth=("elastic", os.getenv("PASSWORD")),  # 使用相同的密码
+            verify_certs=False  # 开发环境可以禁用证书验证
+        )
+        self.api_key = os.getenv("API_KEY")
+        self.api_base = os.getenv("BASE_URL")
+        
+    def get_embedding(self, text: str) -> List[float]:
+        """调用SiliconFlow的embedding API获取向量"""
+        headers = {
+            "Authorization": f"Bearer {self.api_key}",
+            "Content-Type": "application/json"
+        }
+        
+        response = requests.post(
+            f"{self.api_base}/embeddings",
+            headers=headers,
+            json={
+                "model": "BAAI/bge-m3",
+                "input": text
+            }
+        )
+        
+        if response.status_code == 200:
+            return response.json()["data"][0]["embedding"]
+        else:
+            raise Exception(f"Error getting embedding: {response.text}")
+    
+    def get_all_indices(self) -> List[str]:
+        """获取所有 RAG 相关的索引"""
+        indices = self.es.indices.get_alias().keys()
+        return [idx for idx in indices if idx.startswith('rag_')]
+        
+    def retrieve(self, query: str, top_k: int = 10, specific_index: str = None) -> Tuple[List[Dict], str]:
+        """混合检索：结合 BM25 和向量检索，支持指定特定索引"""
+        # 获取检索索引
+        if specific_index:
+            indices = [specific_index] if self.es.indices.exists(index=specific_index) else []
+        else:
+            indices = self.get_all_indices()
+            
+        if not indices:
+            raise Exception("没有找到可用的文档索引！")
+            
+        # 计算查询向量
+        query_vector = self.get_embedding(query)
+        
+        # 在所有索引中搜索
+        all_results = []
+        for index in indices:
+            # 构建混合查询
+            script_query = {
+                "script_score": {
+                    "query": {
+                        "match": {
+                            "content": query  # BM25
+                        }
+                    },
+                    "script": {
+                        "source": "cosineSimilarity(params.query_vector, 'vector') + 1.0",
+                        "params": {"query_vector": query_vector}
+                    }
+                }
+            }
+            
+            # 执行检索
+            response = self.es.search(
+                index=index,
+                body={
+                    "query": script_query,
+                    "size": top_k
+                }
+            )
+            
+            # 处理结果
+            for hit in response['hits']['hits']:
+                result = {
+                    'id': hit['_id'],
+                    'content': hit['_source']['content'],
+                    'score': hit['_score'],
+                    'metadata': hit['_source']['metadata'],
+                    'index': index
+                }
+                all_results.append(result)
+        
+        # 按分数排序并选择最相关的文档
+        all_results.sort(key=lambda x: x['score'], reverse=True)
+        top_results = all_results[:top_k]
+        
+        # 如果有结果，返回最相关文档所在的索引
+        if top_results:
+            most_relevant_index = top_results[0]['index']
+        else:
+            most_relevant_index = indices[0] if indices else ""
+            
         return top_results, most_relevant_index

modules/knowledge_base/vector_store.py

@@ -1,187 +1,187 @@
-from typing import List, Dict
-import requests
-import numpy as np
-from elasticsearch import Elasticsearch
-import urllib3
-from dotenv import load_dotenv
-import os
-
-load_dotenv()
-
-urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
-
-class VectorStore:
-    def __init__(self):
-        # ES 8.x 的连接配置
-        self.es = Elasticsearch(
-            "https://samlax12-elastic.hf.space",
-            basic_auth=("elastic", os.getenv("PASSWORD")),
-            verify_certs=False,
-            request_timeout=30,
-            # 忽略系统索引警告
-            headers={"accept": "application/vnd.elasticsearch+json; compatible-with=8"},
-        )
-        self.api_key = os.getenv("API_KEY")
-        self.api_base = os.getenv("BASE_URL")
-        
-    def get_embedding(self, text: str) -> List[float]:
-        """调用SiliconFlow的embedding API获取向量"""
-        headers = {
-            "Authorization": f"Bearer {self.api_key}",
-            "Content-Type": "application/json"
-        }
-        
-        response = requests.post(
-            f"{self.api_base}/embeddings",
-            headers=headers,
-            json={
-                "model": "BAAI/bge-m3",
-                "input": text
-            }
-        )
-        
-        if response.status_code == 200:
-            return response.json()["data"][0]["embedding"]
-        else:
-            raise Exception(f"Error getting embedding: {response.text}")
-    
-    def store(self, documents: List[Dict], index_name: str) -> None:
-        """将文档存储到 Elasticsearch"""
-        # 创建索引（如果不存在）
-        if not self.es.indices.exists(index=index_name):
-            self.create_index(index_name)
-        
-        # 获取当前索引中的文档数量
-        try:
-            response = self.es.count(index=index_name)
-            last_id = response['count'] - 1  # 文档数量减1作为最后的ID
-            if last_id < 0:
-                last_id = -1
-        except Exception as e:
-            print(f"获取文档数量时出错，假设为-1: {str(e)}")
-            last_id = -1
-        
-        # 批量索引文档
-        bulk_data = []
-        for i, doc in enumerate(documents, start=last_id + 1):
-            # 获取文档向量
-            vector = self.get_embedding(doc['content'])
-            
-            # 准备索引数据
-            bulk_data.append({
-                "index": {
-                    "_index": index_name,
-                    "_id": f"doc_{i}"
-                }
-            })
-            
-            # 构建文档数据，包含新的img_url字段
-            doc_data = {
-                "content": doc['content'],
-                "vector": vector,
-                "metadata": {
-                    "file_name": doc['metadata'].get('file_name', '未知文件'),
-                    "source": doc['metadata'].get('source', ''),
-                    "page": doc['metadata'].get('page', ''),
-                    "img_url": doc['metadata'].get('img_url', '')  # 添加img_url字段
-                }
-            }
-            bulk_data.append(doc_data)
-            
-        # 批量写入
-        if bulk_data:
-            response = self.es.bulk(operations=bulk_data, refresh=True)
-            if response.get('errors'):
-                print("批量写入时出现错误：", response)
-    
-    def get_files_in_index(self, index_name: str) -> List[str]:
-        """获取索引中的所有文件名"""
-        try:
-            response = self.es.search(
-                index=index_name,
-                body={
-                    "size": 0,
-                    "aggs": {
-                        "unique_files": {
-                            "terms": {
-                                "field": "metadata.file_name",
-                                "size": 1000
-                            }
-                        }
-                    }
-                }
-            )
-            
-            files = [bucket['key'] for bucket in response['aggregations']['unique_files']['buckets']]
-            return sorted(files)
-        except Exception as e:
-            print(f"获取文件列表时出错: {str(e)}")
-            return []
-
-    def create_index(self, index_name: str):
-        """创建 Elasticsearch 索引"""
-        settings = {
-            "mappings": {
-                "properties": {
-                    "content": {"type": "text"},
-                    "vector": {
-                        "type": "dense_vector",
-                        "dims": 1024
-                    },
-                    "metadata": {
-                        "properties": {
-                            "file_name": {
-                                "type": "keyword",
-                                "ignore_above": 256
-                            },
-                            "source": {
-                                "type": "keyword"
-                            },
-                            "page": {
-                                "type": "keyword"
-                            },
-                            "img_url": {  # 新增图片URL字段
-                                "type": "keyword",
-                                "ignore_above": 2048
-                            }
-                        }
-                    }
-                }
-            }
-        }
-        
-        # 如果索引已存在，先删除
-        if self.es.indices.exists(index=index_name):
-            self.es.indices.delete(index=index_name)
-        
-        self.es.indices.create(index=index_name, body=settings)
-        
-    def delete_index(self, index_id: str) -> bool:
-        """删除一个索引"""
-        try:
-            if self.es.indices.exists(index=index_id):
-                self.es.indices.delete(index=index_id)
-                return True
-            return False
-        except Exception as e:
-            print(f"删除索引时出错: {str(e)}")
-            return False
-            
-    def delete_document(self, index_id: str, file_name: str) -> bool:
-        """根据文件名删除文档"""
-        try:
-            response = self.es.delete_by_query(
-                index=index_id,
-                body={
-                    "query": {
-                        "term": {
-                            "metadata.file_name": file_name
-                        }
-                    }
-                },
-                refresh=True
-            )
-            return True
-        except Exception as e:
-            print(f"删除文档时出错: {str(e)}")
+from typing import List, Dict
+import requests
+import numpy as np
+from elasticsearch import Elasticsearch
+import urllib3
+from dotenv import load_dotenv
+import os
+
+load_dotenv()
+
+urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
+
+class VectorStore:
+    def __init__(self):
+        # ES 8.x 的连接配置
+        self.es = Elasticsearch(
+            "https://samlax12-elastic.hf.space",
+            basic_auth=("elastic", os.getenv("PASSWORD")),
+            verify_certs=False,
+            request_timeout=30,
+            # 忽略系统索引警告
+            headers={"accept": "application/vnd.elasticsearch+json; compatible-with=8"},
+        )
+        self.api_key = os.getenv("API_KEY")
+        self.api_base = os.getenv("BASE_URL")
+        
+    def get_embedding(self, text: str) -> List[float]:
+        """调用SiliconFlow的embedding API获取向量"""
+        headers = {
+            "Authorization": f"Bearer {self.api_key}",
+            "Content-Type": "application/json"
+        }
+        
+        response = requests.post(
+            f"{self.api_base}/embeddings",
+            headers=headers,
+            json={
+                "model": "BAAI/bge-m3",
+                "input": text
+            }
+        )
+        
+        if response.status_code == 200:
+            return response.json()["data"][0]["embedding"]
+        else:
+            raise Exception(f"Error getting embedding: {response.text}")
+    
+    def store(self, documents: List[Dict], index_name: str) -> None:
+        """将文档存储到 Elasticsearch"""
+        # 创建索引（如果不存在）
+        if not self.es.indices.exists(index=index_name):
+            self.create_index(index_name)
+        
+        # 获取当前索引中的文档数量
+        try:
+            response = self.es.count(index=index_name)
+            last_id = response['count'] - 1  # 文档数量减1作为最后的ID
+            if last_id < 0:
+                last_id = -1
+        except Exception as e:
+            print(f"获取文档数量时出错，假设为-1: {str(e)}")
+            last_id = -1
+        
+        # 批量索引文档
+        bulk_data = []
+        for i, doc in enumerate(documents, start=last_id + 1):
+            # 获取文档向量
+            vector = self.get_embedding(doc['content'])
+            
+            # 准备索引数据
+            bulk_data.append({
+                "index": {
+                    "_index": index_name,
+                    "_id": f"doc_{i}"
+                }
+            })
+            
+            # 构建文档数据，包含新的img_url字段
+            doc_data = {
+                "content": doc['content'],
+                "vector": vector,
+                "metadata": {
+                    "file_name": doc['metadata'].get('file_name', '未知文件'),
+                    "source": doc['metadata'].get('source', ''),
+                    "page": doc['metadata'].get('page', ''),
+                    "img_url": doc['metadata'].get('img_url', '')  # 添加img_url字段
+                }
+            }
+            bulk_data.append(doc_data)
+            
+        # 批量写入
+        if bulk_data:
+            response = self.es.bulk(operations=bulk_data, refresh=True)
+            if response.get('errors'):
+                print("批量写入时出现错误：", response)
+    
+    def get_files_in_index(self, index_name: str) -> List[str]:
+        """获取索引中的所有文件名"""
+        try:
+            response = self.es.search(
+                index=index_name,
+                body={
+                    "size": 0,
+                    "aggs": {
+                        "unique_files": {
+                            "terms": {
+                                "field": "metadata.file_name",
+                                "size": 1000
+                            }
+                        }
+                    }
+                }
+            )
+            
+            files = [bucket['key'] for bucket in response['aggregations']['unique_files']['buckets']]
+            return sorted(files)
+        except Exception as e:
+            print(f"获取文件列表时出错: {str(e)}")
+            return []
+
+    def create_index(self, index_name: str):
+        """创建 Elasticsearch 索引"""
+        settings = {
+            "mappings": {
+                "properties": {
+                    "content": {"type": "text"},
+                    "vector": {
+                        "type": "dense_vector",
+                        "dims": 1024
+                    },
+                    "metadata": {
+                        "properties": {
+                            "file_name": {
+                                "type": "keyword",
+                                "ignore_above": 256
+                            },
+                            "source": {
+                                "type": "keyword"
+                            },
+                            "page": {
+                                "type": "keyword"
+                            },
+                            "img_url": {  # 新增图片URL字段
+                                "type": "keyword",
+                                "ignore_above": 2048
+                            }
+                        }
+                    }
+                }
+            }
+        }
+        
+        # 如果索引已存在，先删除
+        if self.es.indices.exists(index=index_name):
+            self.es.indices.delete(index=index_name)
+        
+        self.es.indices.create(index=index_name, body=settings)
+        
+    def delete_index(self, index_id: str) -> bool:
+        """删除一个索引"""
+        try:
+            if self.es.indices.exists(index=index_id):
+                self.es.indices.delete(index=index_id)
+                return True
+            return False
+        except Exception as e:
+            print(f"删除索引时出错: {str(e)}")
+            return False
+            
+    def delete_document(self, index_id: str, file_name: str) -> bool:
+        """根据文件名删除文档"""
+        try:
+            response = self.es.delete_by_query(
+                index=index_id,
+                body={
+                    "query": {
+                        "term": {
+                            "metadata.file_name": file_name
+                        }
+                    }
+                },
+                refresh=True
+            )
+            return True
+        except Exception as e:
+            print(f"删除文档时出错: {str(e)}")
             return False

requirements.txt

@@ -1,14 +1,14 @@
-flask
-flask_cors
-python-dotenv
-requests
-openai
-elasticsearch
-psutil
-urllib3
-matplotlib
-plotly
-numpy
-pillow
-langchain
+flask
+flask_cors
+python-dotenv
+requests
+openai
+elasticsearch
+psutil
+urllib3
+matplotlib
+plotly
+numpy
+pillow
+langchain
 langchain_community

templates/code_execution.html

@@ -1,718 +1,718 @@
-<!DOCTYPE html>
-<html lang="zh-CN">
-<head>
-    <meta charset="UTF-8">
-    <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <title>AI代码助手 - Python执行环境</title>
-    <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/[email protected]/dist/css/bootstrap.min.css">
-    <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/[email protected]/font/bootstrap-icons.css">
-    <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/highlight.js/11.8.0/styles/vs2015.min.css">
-    <script src="https://cdnjs.cloudflare.com/ajax/libs/highlight.js/11.8.0/highlight.min.js"></script>
-    <script src="https://cdnjs.cloudflare.com/ajax/libs/highlight.js/11.8.0/languages/python.min.js"></script>
-    <style>
-        /* Base styles */
-        :root {
-            --primary-color: #4361ee;
-            --secondary-color: #3f37c9;
-            --accent-color: #4cc9f0;
-            --success-color: #4caf50;
-            --warning-color: #ff9800;
-            --danger-color: #f44336;
-            --light-color: #f8f9fa;
-            --dark-color: #212529;
-            --border-color: #dee2e6;
-            --border-radius: 0.375rem;
-        }
-        
-        body {
-            font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
-            margin: 0;
-            padding: 0;
-            height: 100vh;
-            background-color: #f5f7fa;
-            color: #333;
-            display: flex;
-            flex-direction: column;
-        }
-        
-        /* Layout structure */
-        .workspace {
-            display: grid;
-            grid-template-columns: 1fr 1fr;
-            gap: 16px;
-            flex: 1;
-            padding: 16px;
-        }
-        
-        @media (max-width: 992px) {
-            .workspace {
-                grid-template-columns: 1fr;
-            }
-        }
-        
-        .section {
-            background: #fff;
-            border-radius: var(--border-radius);
-            overflow: hidden;
-            display: flex;
-            flex-direction: column;
-            border: 1px solid var(--border-color);
-            box-shadow: 0 2px 5px rgba(0, 0, 0, 0.05);
-        }
-        
-        .section-header {
-            background: #f8f9fa;
-            padding: 12px 16px;
-            border-bottom: 1px solid var(--border-color);
-            display: flex;
-            justify-content: space-between;
-            align-items: center;
-        }
-        
-        .section-title {
-            display: flex;
-            align-items: center;
-            gap: 8px;
-            font-size: 1rem;
-            font-weight: 500;
-        }
-        
-        /* Code editor styles */
-        .editor-content {
-            flex: 1;
-            position: relative;
-            overflow: hidden;
-        }
-        
-        .code-area {
-            position: absolute;
-            left: 40px;
-            right: 0;
-            top: 0;
-            bottom: 0;
-            padding: 12px 16px;
-            color: #333;
-            font-family: 'JetBrains Mono', 'Fira Code', monospace;
-            font-size: 14px;
-            line-height: 1.6;
-            overflow: auto;
-        }
-        
-        .code-area pre {
-            margin: 0;
-            padding: 0;
-            background: none;
-            border: none;
-        }
-        
-        .code-area code {
-            display: block;
-            padding: 0;
-            tab-size: 4;
-            font-family: 'JetBrains Mono', 'Fira Code', monospace;
-            outline: none;
-            position: relative;
-            min-height: 100%;
-            white-space: pre !important;
-            word-wrap: normal !important;
-        }
-        
-        .line-numbers {
-            position: absolute;
-            left: 0;
-            top: 0;
-            bottom: 0;
-            width: 40px;
-            padding: 12px 0;
-            background: #f5f7fa;
-            border-right: 1px solid #e9ecef;
-            text-align: center;
-            color: #6c757d;
-            font-family: 'JetBrains Mono', 'Fira Code', monospace;
-            font-size: 14px;
-            line-height: 1.6;
-            user-select: none;
-        }
-        
-        /* Terminal styles */
-        .output-content {
-            flex: 1;
-            background: #f8f9fa;
-            position: relative;
-            overflow: hidden;
-            display: flex;
-            flex-direction: column;
-        }
-        
-        .terminal-window {
-            flex: 1;
-            overflow-y: auto;
-            background: #212529;
-            color: #f8f9fa;
-        }
-        
-        #output {
-            color: #f8f9fa;
-            margin: 0;
-            padding: 12px 16px;
-            background: transparent;
-            border: none;
-            white-space: pre-wrap;
-            word-wrap: break-word;
-            line-height: 1.6;
-            font-size: 14px;
-            font-family: 'JetBrains Mono', 'Fira Code', monospace;
-        }
-        
-        .console-input-container {
-            display: flex;
-            align-items: center;
-            background: #343a40;
-            border-top: 1px solid #495057;
-            padding: 8px 12px;
-        }
-        
-        .console-prompt {
-            color: #4caf50;
-            font-family: 'JetBrains Mono', 'Fira Code', monospace;
-            margin-right: 8px;
-            font-size: 14px;
-            user-select: none;
-        }
-        
-        .console-input {
-            flex: 1;
-            background: transparent;
-            border: none;
-            color: #f8f9fa;
-            font-family: 'JetBrains Mono', 'Fira Code', monospace;
-            font-size: 14px;
-            line-height: 1.5;
-            padding: 4px 0;
-        }
-        
-        .console-input:focus {
-            outline: none;
-        }
-        
-        /* Chat section */
-        .chat-section {
-            display: flex;
-            flex-direction: column;
-            height: 100%;
-        }
-        
-        .chat-messages {
-            flex: 1;
-            overflow-y: auto;
-            padding: 16px;
-        }
-        
-        .message {
-            margin-bottom: 16px;
-            padding: 12px;
-            border-radius: var(--border-radius);
-            max-width: 85%;
-            position: relative;
-        }
-        
-        .message.user {
-            background-color: #e3f2fd;
-            color: #0d47a1;
-            align-self: flex-end;
-            margin-left: auto;
-        }
-        
-        .message.bot {
-            background-color: #f5f5f5;
-            color: #333;
-            align-self: flex-start;
-            border-left: 3px solid var(--primary-color);
-        }
-        
-        .chat-input-container {
-            padding: 16px;
-            border-top: 1px solid var(--border-color);
-            background-color: #f9f9f9;
-        }
-        
-        .input-row {
-            display: flex;
-            gap: 8px;
-        }
-        
-        .chat-input {
-            flex: 1;
-            padding: 12px;
-            border: 1px solid var(--border-color);
-            border-radius: var(--border-radius);
-            resize: none;
-            font-size: 14px;
-            height: 100px;
-        }
-        
-        .chat-input:focus {
-            outline: none;
-            border-color: var(--primary-color);
-            box-shadow: 0 0 0 3px rgba(67, 97, 238, 0.1);
-        }
-        
-        /* Button styles */
-        .btn {
-            padding: 8px 16px;
-            border: none;
-            border-radius: var(--border-radius);
-            cursor: pointer;
-            font-weight: 500;
-            font-size: 14px;
-            display: flex;
-            align-items: center;
-            gap: 8px;
-            transition: all 0.2s ease;
-        }
-        
-        .btn-primary {
-            background-color: var(--primary-color);
-            color: white;
-        }
-        
-        .btn-primary:hover {
-            background-color: var(--secondary-color);
-        }
-        
-        .btn-secondary {
-            background-color: #6c757d;
-            color: white;
-        }
-        
-        .btn-secondary:hover {
-            background-color: #5a6268;
-        }
-        
-        .btn-danger {
-            background-color: var(--danger-color);
-            color: white;
-        }
-        
-        .btn-danger:hover {
-            background-color: #d32f2f;
-        }
-        
-        /* Utilities */
-        .loading {
-            display: none;
-            align-items: center;
-            gap: 8px;
-            color: #6c757d;
-            font-size: 14px;
-        }
-        
-        .loading.active {
-            display: flex;
-        }
-        
-        @keyframes spin {
-            to { transform: rotate(360deg); }
-        }
-        
-        .loading i {
-            animation: spin 1s linear infinite;
-        }
-        
-        /* Custom scrollbar */
-        ::-webkit-scrollbar {
-            width: 8px;
-            height: 8px;
-        }
-        
-        ::-webkit-scrollbar-track {
-            background: #f1f1f1;
-        }
-        
-        ::-webkit-scrollbar-thumb {
-            background: #c1c1c1;
-            border-radius: 4px;
-        }
-        
-        ::-webkit-scrollbar-thumb:hover {
-            background: #a8a8a8;
-        }
-        
-        /* Terminal text styles */
-        .term-input {
-            color: #4caf50;
-        }
-        
-        .term-output {
-            color: #f8f9fa;
-        }
-        
-        .term-error {
-            color: #f44336;
-        }
-        
-        .term-warning {
-            color: #ff9800;
-        }
-        
-        .term-system {
-            color: #2196f3;
-        }
-        
-        /* Header */
-        .header {
-            background-color: #fff;
-            border-bottom: 1px solid var(--border-color);
-            padding: 1rem;
-            box-shadow: 0 2px 4px rgba(0, 0, 0, 0.05);
-        }
-        
-        .header-content {
-            max-width: 1200px;
-            margin: 0 auto;
-            display: flex;
-            justify-content: space-between;
-            align-items: center;
-        }
-        
-        .header h1 {
-            margin: 0;
-            font-size: 1.5rem;
-            color: var(--primary-color);
-        }
-    </style>
-</head>
-<body>
-    <header class="header">
-        <div class="header-content">
-            <h1>AI代码助手</h1>
-            <div>
-                <span class="badge bg-primary">Python编程环境</span>
-            </div>
-        </div>
-    </header>
-    
-    <div class="workspace">
-        <div class="section">
-            <div class="section-header">
-                <div class="section-title">
-                    <i class="bi bi-code-square"></i>
-                    代码编辑器
-                </div>
-                <div style="display: flex; gap: 8px;">
-                    <button class="btn btn-primary" id="runCode">
-                        <i class="bi bi-play-fill"></i>
-                        运行
-                    </button>
-                    <button class="btn btn-danger" id="stopCode" style="display: none;">
-                        <i class="bi bi-stop-fill"></i>
-                        停止
-                    </button>
-                    <button class="btn btn-secondary" id="clearCode">
-                        <i class="bi bi-trash"></i>
-                        清除
-                    </button>
-                </div>
-            </div>
-            <div class="editor-content">
-                <div class="line-numbers" id="lineNumbers">1</div>
-                <div class="code-area" id="codeArea">
-                    <pre><code class="language-python" contenteditable="true" spellcheck="false" autocorrect="off" autocapitalize="off"># 您的代码将在这里显示</code></pre>
-                </div>
-            </div>
-        </div>
-        
-        <div class="section">
-            <div class="section-header">
-                <div class="section-title">
-                    <i class="bi bi-terminal"></i>
-                    终端输出
-                </div>
-                <div style="display: flex; gap: 8px;">
-                    <button class="btn btn-secondary" id="clearTerminal">
-                        <i class="bi bi-eraser"></i>
-                        清除
-                    </button>
-                </div>
-            </div>
-            <div class="output-content">
-                <div class="terminal-window">
-                    <pre id="output"></pre>
-                </div>
-                <div class="console-input-container" id="consoleInputContainer" style="display: none;">
-                    <div class="console-prompt">>></div>
-                    <input type="text" id="consoleInput" class="console-input" autocomplete="off" spellcheck="false" />
-                </div>
-            </div>
-        </div>
-    </div>
-    
-    <script>
-        document.addEventListener('DOMContentLoaded', () => {
-            // Terminal elements
-            const terminalOutput = document.getElementById('output');
-            const consoleInputContainer = document.getElementById('consoleInputContainer');
-            const consoleInput = document.getElementById('consoleInput');
-            const clearTerminalBtn = document.getElementById('clearTerminal');
-            const stopCodeBtn = document.getElementById('stopCode');
-            
-            // Editor elements
-            const codeArea = document.querySelector('#codeArea code');
-            const lineNumbers = document.getElementById('lineNumbers');
-            const runButton = document.getElementById('runCode');
-            const clearButton = document.getElementById('clearCode');
-            
-            // State variables
-            let executionContext = null;
-            let isExecuting = false;
-            let executionStartTime = null;
-            
-            // Initialize terminal
-            initializeTerminal();
-            
-            // Initialize editor
-            updateLineNumbers();
-            
-            // Function to update line numbers
-            function updateLineNumbers() {
-                const lines = codeArea.textContent.split('\n').length;
-                lineNumbers.innerHTML = Array.from({length: lines}, (_, i) => i + 1).join('<br>');
-            }
-            
-            // Initialize terminal with welcome message
-            function initializeTerminal() {
-                clearTerminalOutput();
-                appendToTerminal("欢迎使用Python交互式终端", "term-system");
-                appendToTerminal("使用'运行'按钮执行您的代码", "term-system");
-                appendToTerminal("", "term-system"); // 空行
-                appendToTerminal(">>> 准备执行代码...", "term-system");
-            }
-            
-            // Append text to terminal
-            function appendToTerminal(text, type = null) {
-                const lines = text.split('\n');
-                let html = '';
-                
-                for (const line of lines) {
-                    if (type) {
-                        html += `<span class="${type}">${escapeHtml(line)}</span>\n`;
-                    } else {
-                        html += escapeHtml(line) + '\n';
-                    }
-                }
-                
-                terminalOutput.innerHTML += html;
-                terminalOutput.scrollTop = terminalOutput.scrollHeight;
-            }
-            
-            // Clear terminal output
-            function clearTerminalOutput() {
-                terminalOutput.innerHTML = '';
-            }
-            
-            // Escape HTML to prevent XSS
-            function escapeHtml(text) {
-                return text
-                    .replace(/&/g, "&amp;")
-                    .replace(/</g, "&lt;")
-                    .replace(/>/g, "&gt;")
-                    .replace(/"/g, "&quot;")
-                    .replace(/'/g, "&#039;");
-            }
-            
-            // Run code
-            runButton.addEventListener('click', async () => {
-                if (isExecuting) return; // Prevent multiple executions
-                
-                const code = codeArea.textContent;
-                
-                // Clear terminal and show execution start
-                clearTerminalOutput();
-                appendToTerminal("开始执行Python代码...", "term-system");
-                
-                // Update UI state
-                isExecuting = true;
-                executionStartTime = performance.now();
-                runButton.style.display = 'none';
-                stopCodeBtn.style.display = 'flex';
-                
-                try {
-                    const response = await fetch('/api/code/execute', {
-                        method: 'POST',
-                        headers: { 'Content-Type': 'application/json' },
-                        body: JSON.stringify({ code })
-                    });
-                    
-                    const data = await response.json();
-                    
-                    if (data.success) {
-                        // Show output (if any)
-                        if (data.output && data.output.trim()) {
-                            appendToTerminal(data.output, "term-output");
-                        }
-                        
-                        if (data.needsInput) {
-                            // Code is waiting for input
-                            executionContext = data.context_id;
-                            consoleInputContainer.style.display = 'flex';
-                            consoleInput.focus();
-                        } else {
-                            // Code has completed, no input needed
-                            appendToTerminal("程序执行完成", "term-system");
-                            finishExecution();
-                        }
-                    } else {
-                        // Handle error
-                        appendToTerminal(`错误: ${data.error}`, "term-error");
-                        if (data.traceback) {
-                            appendToTerminal(data.traceback, "term-error");
-                        }
-                        appendToTerminal("执行失败", "term-system");
-                        finishExecution();
-                    }
-                } catch (error) {
-                    appendToTerminal(`系统错误: ${error.message}`, "term-error");
-                    appendToTerminal("执行失败", "term-system");
-                    finishExecution();
-                }
-            });
-            
-            // Submit console input
-            consoleInput.addEventListener('keydown', async (e) => {
-                if (e.key === 'Enter' && executionContext) {
-                    const input = consoleInput.value;
-                    consoleInput.value = '';
-                    
-                    // Add input to terminal
-                    appendToTerminal(`>> ${input}`, "term-input");
-                    
-                    try {
-                        // Send input to backend
-                        const response = await fetch('/api/code/input', {
-                            method: 'POST',
-                            headers: { 'Content-Type': 'application/json' },
-                            body: JSON.stringify({
-                                input: input,
-                                context_id: executionContext
-                            })
-                        });
-                        
-                        const data = await response.json();
-                        
-                        if (data.success) {
-                            // Show new output
-                            if (data.output && data.output.trim()) {
-                                appendToTerminal(data.output, "term-output");
-                            }
-                            
-                            if (data.needsInput) {
-                                // Still waiting for more input
-                                consoleInput.focus();
-                            } else {
-                                // Execution completed
-                                appendToTerminal(">>> 程序执行完成", "term-system");
-                                finishExecution();
-                            }
-                        } else {
-                            // Handle error
-                            appendToTerminal(`错误: ${data.error}`, "term-error");
-                            if (data.traceback) {
-                                appendToTerminal(data.traceback, "term-error");
-                            }
-                            finishExecution();
-                        }
-                    } catch (error) {
-                        appendToTerminal(`系统错误: ${error.message}`, "term-error");
-                        finishExecution();
-                    }
-                }
-            });
-            
-            // Stop code execution
-            stopCodeBtn.addEventListener('click', async () => {
-                if (!executionContext) return;
-                
-                try {
-                    // Send stop request to server
-                    const response = await fetch('/api/code/stop', {
-                        method: 'POST',
-                        headers: { 'Content-Type': 'application/json' },
-                        body: JSON.stringify({ context_id: executionContext })
-                    });
-                    
-                    // Clean up UI regardless of response
-                    appendToTerminal("用户终止了执行", "term-warning");
-                    finishExecution();
-                } catch (error) {
-                    console.error('停止执行时出错:', error);
-                    finishExecution();
-                }
-            });
-            
-            // Clear code
-            clearButton.addEventListener('click', () => {
-                codeArea.textContent = '# 您的代码将在这里显示';
-                updateLineNumbers();
-                clearTerminalOutput();
-                initializeTerminal();
-                consoleInputContainer.style.display = 'none';
-                executionContext = null;
-                isExecuting = false;
-                runButton.style.display = 'flex';
-                stopCodeBtn.style.display = 'none';
-            });
-            
-            // Clear terminal
-            clearTerminalBtn.addEventListener('click', () => {
-                if (!isExecuting) {
-                    clearTerminalOutput();
-                    initializeTerminal();
-                } else {
-                    // If execution is in progress, just add a separator
-                    appendToTerminal("\n--- 已清除终端 ---\n", "term-system");
-                }
-            });
-            
-            // Update line numbers on code changes
-            codeArea.addEventListener('input', updateLineNumbers);
-            
-            // Handle tab key
-            codeArea.addEventListener('keydown', (e) => {
-                if (e.key === 'Tab') {
-                    e.preventDefault();
-                    document.execCommand('insertText', false, '    ');
-                }
-            });
-            
-            // Function to clean up after execution completes
-            function finishExecution() {
-                consoleInputContainer.style.display = 'none';
-                executionContext = null;
-                isExecuting = false;
-                runButton.style.display = 'flex';
-                stopCodeBtn.style.display = 'none';
-            }
-            
-            // Check if code was provided via URL parameters
-            const urlParams = new URLSearchParams(window.location.search);
-            const initialCode = urlParams.get('code');
-            if (initialCode) {
-                try {
-                    codeArea.textContent = decodeURIComponent(initialCode);
-                    updateLineNumbers();
-                } catch (e) {
-                    console.error('Failed to decode initial code:', e);
-                }
-            }
-            
-            // Check for messages from parent frame
-            window.addEventListener('message', (event) => {
-                if (event.data && event.data.type === 'setCode') {
-                    codeArea.textContent = event.data.code;
-                    updateLineNumbers();
-                }
-            });
-        });
-    </script>
-</body>
+<!DOCTYPE html>
+<html lang="zh-CN">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>AI代码助手 - Python执行环境</title>
+    <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/[email protected]/dist/css/bootstrap.min.css">
+    <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/[email protected]/font/bootstrap-icons.css">
+    <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/highlight.js/11.8.0/styles/vs2015.min.css">
+    <script src="https://cdnjs.cloudflare.com/ajax/libs/highlight.js/11.8.0/highlight.min.js"></script>
+    <script src="https://cdnjs.cloudflare.com/ajax/libs/highlight.js/11.8.0/languages/python.min.js"></script>
+    <style>
+        /* Base styles */
+        :root {
+            --primary-color: #4361ee;
+            --secondary-color: #3f37c9;
+            --accent-color: #4cc9f0;
+            --success-color: #4caf50;
+            --warning-color: #ff9800;
+            --danger-color: #f44336;
+            --light-color: #f8f9fa;
+            --dark-color: #212529;
+            --border-color: #dee2e6;
+            --border-radius: 0.375rem;
+        }
+        
+        body {
+            font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
+            margin: 0;
+            padding: 0;
+            height: 100vh;
+            background-color: #f5f7fa;
+            color: #333;
+            display: flex;
+            flex-direction: column;
+        }
+        
+        /* Layout structure */
+        .workspace {
+            display: grid;
+            grid-template-columns: 1fr 1fr;
+            gap: 16px;
+            flex: 1;
+            padding: 16px;
+        }
+        
+        @media (max-width: 992px) {
+            .workspace {
+                grid-template-columns: 1fr;
+            }
+        }
+        
+        .section {
+            background: #fff;
+            border-radius: var(--border-radius);
+            overflow: hidden;
+            display: flex;
+            flex-direction: column;
+            border: 1px solid var(--border-color);
+            box-shadow: 0 2px 5px rgba(0, 0, 0, 0.05);
+        }
+        
+        .section-header {
+            background: #f8f9fa;
+            padding: 12px 16px;
+            border-bottom: 1px solid var(--border-color);
+            display: flex;
+            justify-content: space-between;
+            align-items: center;
+        }
+        
+        .section-title {
+            display: flex;
+            align-items: center;
+            gap: 8px;
+            font-size: 1rem;
+            font-weight: 500;
+        }
+        
+        /* Code editor styles */
+        .editor-content {
+            flex: 1;
+            position: relative;
+            overflow: hidden;
+        }
+        
+        .code-area {
+            position: absolute;
+            left: 40px;
+            right: 0;
+            top: 0;
+            bottom: 0;
+            padding: 12px 16px;
+            color: #333;
+            font-family: 'JetBrains Mono', 'Fira Code', monospace;
+            font-size: 14px;
+            line-height: 1.6;
+            overflow: auto;
+        }
+        
+        .code-area pre {
+            margin: 0;
+            padding: 0;
+            background: none;
+            border: none;
+        }
+        
+        .code-area code {
+            display: block;
+            padding: 0;
+            tab-size: 4;
+            font-family: 'JetBrains Mono', 'Fira Code', monospace;
+            outline: none;
+            position: relative;
+            min-height: 100%;
+            white-space: pre !important;
+            word-wrap: normal !important;
+        }
+        
+        .line-numbers {
+            position: absolute;
+            left: 0;
+            top: 0;
+            bottom: 0;
+            width: 40px;
+            padding: 12px 0;
+            background: #f5f7fa;
+            border-right: 1px solid #e9ecef;
+            text-align: center;
+            color: #6c757d;
+            font-family: 'JetBrains Mono', 'Fira Code', monospace;
+            font-size: 14px;
+            line-height: 1.6;
+            user-select: none;
+        }
+        
+        /* Terminal styles */
+        .output-content {
+            flex: 1;
+            background: #f8f9fa;
+            position: relative;
+            overflow: hidden;
+            display: flex;
+            flex-direction: column;
+        }
+        
+        .terminal-window {
+            flex: 1;
+            overflow-y: auto;
+            background: #212529;
+            color: #f8f9fa;
+        }
+        
+        #output {
+            color: #f8f9fa;
+            margin: 0;
+            padding: 12px 16px;
+            background: transparent;
+            border: none;
+            white-space: pre-wrap;
+            word-wrap: break-word;
+            line-height: 1.6;
+            font-size: 14px;
+            font-family: 'JetBrains Mono', 'Fira Code', monospace;
+        }
+        
+        .console-input-container {
+            display: flex;
+            align-items: center;
+            background: #343a40;
+            border-top: 1px solid #495057;
+            padding: 8px 12px;
+        }
+        
+        .console-prompt {
+            color: #4caf50;
+            font-family: 'JetBrains Mono', 'Fira Code', monospace;
+            margin-right: 8px;
+            font-size: 14px;
+            user-select: none;
+        }
+        
+        .console-input {
+            flex: 1;
+            background: transparent;
+            border: none;
+            color: #f8f9fa;
+            font-family: 'JetBrains Mono', 'Fira Code', monospace;
+            font-size: 14px;
+            line-height: 1.5;
+            padding: 4px 0;
+        }
+        
+        .console-input:focus {
+            outline: none;
+        }
+        
+        /* Chat section */
+        .chat-section {
+            display: flex;
+            flex-direction: column;
+            height: 100%;
+        }
+        
+        .chat-messages {
+            flex: 1;
+            overflow-y: auto;
+            padding: 16px;
+        }
+        
+        .message {
+            margin-bottom: 16px;
+            padding: 12px;
+            border-radius: var(--border-radius);
+            max-width: 85%;
+            position: relative;
+        }
+        
+        .message.user {
+            background-color: #e3f2fd;
+            color: #0d47a1;
+            align-self: flex-end;
+            margin-left: auto;
+        }
+        
+        .message.bot {
+            background-color: #f5f5f5;
+            color: #333;
+            align-self: flex-start;
+            border-left: 3px solid var(--primary-color);
+        }
+        
+        .chat-input-container {
+            padding: 16px;
+            border-top: 1px solid var(--border-color);
+            background-color: #f9f9f9;
+        }
+        
+        .input-row {
+            display: flex;
+            gap: 8px;
+        }
+        
+        .chat-input {
+            flex: 1;
+            padding: 12px;
+            border: 1px solid var(--border-color);
+            border-radius: var(--border-radius);
+            resize: none;
+            font-size: 14px;
+            height: 100px;
+        }
+        
+        .chat-input:focus {
+            outline: none;
+            border-color: var(--primary-color);
+            box-shadow: 0 0 0 3px rgba(67, 97, 238, 0.1);
+        }
+        
+        /* Button styles */
+        .btn {
+            padding: 8px 16px;
+            border: none;
+            border-radius: var(--border-radius);
+            cursor: pointer;
+            font-weight: 500;
+            font-size: 14px;
+            display: flex;
+            align-items: center;
+            gap: 8px;
+            transition: all 0.2s ease;
+        }
+        
+        .btn-primary {
+            background-color: var(--primary-color);
+            color: white;
+        }
+        
+        .btn-primary:hover {
+            background-color: var(--secondary-color);
+        }
+        
+        .btn-secondary {
+            background-color: #6c757d;
+            color: white;
+        }
+        
+        .btn-secondary:hover {
+            background-color: #5a6268;
+        }
+        
+        .btn-danger {
+            background-color: var(--danger-color);
+            color: white;
+        }
+        
+        .btn-danger:hover {
+            background-color: #d32f2f;
+        }
+        
+        /* Utilities */
+        .loading {
+            display: none;
+            align-items: center;
+            gap: 8px;
+            color: #6c757d;
+            font-size: 14px;
+        }
+        
+        .loading.active {
+            display: flex;
+        }
+        
+        @keyframes spin {
+            to { transform: rotate(360deg); }
+        }
+        
+        .loading i {
+            animation: spin 1s linear infinite;
+        }
+        
+        /* Custom scrollbar */
+        ::-webkit-scrollbar {
+            width: 8px;
+            height: 8px;
+        }
+        
+        ::-webkit-scrollbar-track {
+            background: #f1f1f1;
+        }
+        
+        ::-webkit-scrollbar-thumb {
+            background: #c1c1c1;
+            border-radius: 4px;
+        }
+        
+        ::-webkit-scrollbar-thumb:hover {
+            background: #a8a8a8;
+        }
+        
+        /* Terminal text styles */
+        .term-input {
+            color: #4caf50;
+        }
+        
+        .term-output {
+            color: #f8f9fa;
+        }
+        
+        .term-error {
+            color: #f44336;
+        }
+        
+        .term-warning {
+            color: #ff9800;
+        }
+        
+        .term-system {
+            color: #2196f3;
+        }
+        
+        /* Header */
+        .header {
+            background-color: #fff;
+            border-bottom: 1px solid var(--border-color);
+            padding: 1rem;
+            box-shadow: 0 2px 4px rgba(0, 0, 0, 0.05);
+        }
+        
+        .header-content {
+            max-width: 1200px;
+            margin: 0 auto;
+            display: flex;
+            justify-content: space-between;
+            align-items: center;
+        }
+        
+        .header h1 {
+            margin: 0;
+            font-size: 1.5rem;
+            color: var(--primary-color);
+        }
+    </style>
+</head>
+<body>
+    <header class="header">
+        <div class="header-content">
+            <h1>AI代码助手</h1>
+            <div>
+                <span class="badge bg-primary">Python编程环境</span>
+            </div>
+        </div>
+    </header>
+    
+    <div class="workspace">
+        <div class="section">
+            <div class="section-header">
+                <div class="section-title">
+                    <i class="bi bi-code-square"></i>
+                    代码编辑器
+                </div>
+                <div style="display: flex; gap: 8px;">
+                    <button class="btn btn-primary" id="runCode">
+                        <i class="bi bi-play-fill"></i>
+                        运行
+                    </button>
+                    <button class="btn btn-danger" id="stopCode" style="display: none;">
+                        <i class="bi bi-stop-fill"></i>
+                        停止
+                    </button>
+                    <button class="btn btn-secondary" id="clearCode">
+                        <i class="bi bi-trash"></i>
+                        清除
+                    </button>
+                </div>
+            </div>
+            <div class="editor-content">
+                <div class="line-numbers" id="lineNumbers">1</div>
+                <div class="code-area" id="codeArea">
+                    <pre><code class="language-python" contenteditable="true" spellcheck="false" autocorrect="off" autocapitalize="off"># 您的代码将在这里显示</code></pre>
+                </div>
+            </div>
+        </div>
+        
+        <div class="section">
+            <div class="section-header">
+                <div class="section-title">
+                    <i class="bi bi-terminal"></i>
+                    终端输出
+                </div>
+                <div style="display: flex; gap: 8px;">
+                    <button class="btn btn-secondary" id="clearTerminal">
+                        <i class="bi bi-eraser"></i>
+                        清除
+                    </button>
+                </div>
+            </div>
+            <div class="output-content">
+                <div class="terminal-window">
+                    <pre id="output"></pre>
+                </div>
+                <div class="console-input-container" id="consoleInputContainer" style="display: none;">
+                    <div class="console-prompt">>></div>
+                    <input type="text" id="consoleInput" class="console-input" autocomplete="off" spellcheck="false" />
+                </div>
+            </div>
+        </div>
+    </div>
+    
+    <script>
+        document.addEventListener('DOMContentLoaded', () => {
+            // Terminal elements
+            const terminalOutput = document.getElementById('output');
+            const consoleInputContainer = document.getElementById('consoleInputContainer');
+            const consoleInput = document.getElementById('consoleInput');
+            const clearTerminalBtn = document.getElementById('clearTerminal');
+            const stopCodeBtn = document.getElementById('stopCode');
+            
+            // Editor elements
+            const codeArea = document.querySelector('#codeArea code');
+            const lineNumbers = document.getElementById('lineNumbers');
+            const runButton = document.getElementById('runCode');
+            const clearButton = document.getElementById('clearCode');
+            
+            // State variables
+            let executionContext = null;
+            let isExecuting = false;
+            let executionStartTime = null;
+            
+            // Initialize terminal
+            initializeTerminal();
+            
+            // Initialize editor
+            updateLineNumbers();
+            
+            // Function to update line numbers
+            function updateLineNumbers() {
+                const lines = codeArea.textContent.split('\n').length;
+                lineNumbers.innerHTML = Array.from({length: lines}, (_, i) => i + 1).join('<br>');
+            }
+            
+            // Initialize terminal with welcome message
+            function initializeTerminal() {
+                clearTerminalOutput();
+                appendToTerminal("欢迎使用Python交互式终端", "term-system");
+                appendToTerminal("使用'运行'按钮执行您的代码", "term-system");
+                appendToTerminal("", "term-system"); // 空行
+                appendToTerminal(">>> 准备执行代码...", "term-system");
+            }
+            
+            // Append text to terminal
+            function appendToTerminal(text, type = null) {
+                const lines = text.split('\n');
+                let html = '';
+                
+                for (const line of lines) {
+                    if (type) {
+                        html += `<span class="${type}">${escapeHtml(line)}</span>\n`;
+                    } else {
+                        html += escapeHtml(line) + '\n';
+                    }
+                }
+                
+                terminalOutput.innerHTML += html;
+                terminalOutput.scrollTop = terminalOutput.scrollHeight;
+            }
+            
+            // Clear terminal output
+            function clearTerminalOutput() {
+                terminalOutput.innerHTML = '';
+            }
+            
+            // Escape HTML to prevent XSS
+            function escapeHtml(text) {
+                return text
+                    .replace(/&/g, "&amp;")
+                    .replace(/</g, "&lt;")
+                    .replace(/>/g, "&gt;")
+                    .replace(/"/g, "&quot;")
+                    .replace(/'/g, "&#039;");
+            }
+            
+            // Run code
+            runButton.addEventListener('click', async () => {
+                if (isExecuting) return; // Prevent multiple executions
+                
+                const code = codeArea.textContent;
+                
+                // Clear terminal and show execution start
+                clearTerminalOutput();
+                appendToTerminal("开始执行Python代码...", "term-system");
+                
+                // Update UI state
+                isExecuting = true;
+                executionStartTime = performance.now();
+                runButton.style.display = 'none';
+                stopCodeBtn.style.display = 'flex';
+                
+                try {
+                    const response = await fetch('/api/code/execute', {
+                        method: 'POST',
+                        headers: { 'Content-Type': 'application/json' },
+                        body: JSON.stringify({ code })
+                    });
+                    
+                    const data = await response.json();
+                    
+                    if (data.success) {
+                        // Show output (if any)
+                        if (data.output && data.output.trim()) {
+                            appendToTerminal(data.output, "term-output");
+                        }
+                        
+                        if (data.needsInput) {
+                            // Code is waiting for input
+                            executionContext = data.context_id;
+                            consoleInputContainer.style.display = 'flex';
+                            consoleInput.focus();
+                        } else {
+                            // Code has completed, no input needed
+                            appendToTerminal("程序执行完成", "term-system");
+                            finishExecution();
+                        }
+                    } else {
+                        // Handle error
+                        appendToTerminal(`错误: ${data.error}`, "term-error");
+                        if (data.traceback) {
+                            appendToTerminal(data.traceback, "term-error");
+                        }
+                        appendToTerminal("执行失败", "term-system");
+                        finishExecution();
+                    }
+                } catch (error) {
+                    appendToTerminal(`系统错误: ${error.message}`, "term-error");
+                    appendToTerminal("执行失败", "term-system");
+                    finishExecution();
+                }
+            });
+            
+            // Submit console input
+            consoleInput.addEventListener('keydown', async (e) => {
+                if (e.key === 'Enter' && executionContext) {
+                    const input = consoleInput.value;
+                    consoleInput.value = '';
+                    
+                    // Add input to terminal
+                    appendToTerminal(`>> ${input}`, "term-input");
+                    
+                    try {
+                        // Send input to backend
+                        const response = await fetch('/api/code/input', {
+                            method: 'POST',
+                            headers: { 'Content-Type': 'application/json' },
+                            body: JSON.stringify({
+                                input: input,
+                                context_id: executionContext
+                            })
+                        });
+                        
+                        const data = await response.json();
+                        
+                        if (data.success) {
+                            // Show new output
+                            if (data.output && data.output.trim()) {
+                                appendToTerminal(data.output, "term-output");
+                            }
+                            
+                            if (data.needsInput) {
+                                // Still waiting for more input
+                                consoleInput.focus();
+                            } else {
+                                // Execution completed
+                                appendToTerminal(">>> 程序执行完成", "term-system");
+                                finishExecution();
+                            }
+                        } else {
+                            // Handle error
+                            appendToTerminal(`错误: ${data.error}`, "term-error");
+                            if (data.traceback) {
+                                appendToTerminal(data.traceback, "term-error");
+                            }
+                            finishExecution();
+                        }
+                    } catch (error) {
+                        appendToTerminal(`系统错误: ${error.message}`, "term-error");
+                        finishExecution();
+                    }
+                }
+            });
+            
+            // Stop code execution
+            stopCodeBtn.addEventListener('click', async () => {
+                if (!executionContext) return;
+                
+                try {
+                    // Send stop request to server
+                    const response = await fetch('/api/code/stop', {
+                        method: 'POST',
+                        headers: { 'Content-Type': 'application/json' },
+                        body: JSON.stringify({ context_id: executionContext })
+                    });
+                    
+                    // Clean up UI regardless of response
+                    appendToTerminal("用户终止了执行", "term-warning");
+                    finishExecution();
+                } catch (error) {
+                    console.error('停止执行时出错:', error);
+                    finishExecution();
+                }
+            });
+            
+            // Clear code
+            clearButton.addEventListener('click', () => {
+                codeArea.textContent = '# 您的代码将在这里显示';
+                updateLineNumbers();
+                clearTerminalOutput();
+                initializeTerminal();
+                consoleInputContainer.style.display = 'none';
+                executionContext = null;
+                isExecuting = false;
+                runButton.style.display = 'flex';
+                stopCodeBtn.style.display = 'none';
+            });
+            
+            // Clear terminal
+            clearTerminalBtn.addEventListener('click', () => {
+                if (!isExecuting) {
+                    clearTerminalOutput();
+                    initializeTerminal();
+                } else {
+                    // If execution is in progress, just add a separator
+                    appendToTerminal("\n--- 已清除终端 ---\n", "term-system");
+                }
+            });
+            
+            // Update line numbers on code changes
+            codeArea.addEventListener('input', updateLineNumbers);
+            
+            // Handle tab key
+            codeArea.addEventListener('keydown', (e) => {
+                if (e.key === 'Tab') {
+                    e.preventDefault();
+                    document.execCommand('insertText', false, '    ');
+                }
+            });
+            
+            // Function to clean up after execution completes
+            function finishExecution() {
+                consoleInputContainer.style.display = 'none';
+                executionContext = null;
+                isExecuting = false;
+                runButton.style.display = 'flex';
+                stopCodeBtn.style.display = 'none';
+            }
+            
+            // Check if code was provided via URL parameters
+            const urlParams = new URLSearchParams(window.location.search);
+            const initialCode = urlParams.get('code');
+            if (initialCode) {
+                try {
+                    codeArea.textContent = decodeURIComponent(initialCode);
+                    updateLineNumbers();
+                } catch (e) {
+                    console.error('Failed to decode initial code:', e);
+                }
+            }
+            
+            // Check for messages from parent frame
+            window.addEventListener('message', (event) => {
+                if (event.data && event.data.type === 'setCode') {
+                    codeArea.textContent = event.data.code;
+                    updateLineNumbers();
+                }
+            });
+        });
+    </script>
+</body>
 </html>

templates/login.html

@@ -0,0 +1,505 @@
+<!DOCTYPE html>
+<html lang="zh-CN">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <meta http-equiv="Content-Security-Policy" content="upgrade-insecure-requests">
+    <title>教育AI助手平台 - 登录</title>
+    <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/[email protected]/dist/css/bootstrap.min.css">
+    <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/[email protected]/font/bootstrap-icons.css">
+    <style>
+        @import url('https://fonts.googleapis.com/css2?family=Inter:wght@300;400;500;600;700&display=swap');
+        
+        :root {
+            /* 优雅的配色方案 */
+            --primary-color: #0f2d49;
+            --primary-light: #234a70;
+            --secondary-color: #4a6cfd;
+            --secondary-light: #7b91ff;
+            --tertiary-color: #f7f9fe;
+            --success-color: #10b981;
+            --success-light: rgba(16, 185, 129, 0.1);
+            --warning-color: #f59e0b;
+            --warning-light: rgba(245, 158, 11, 0.1);
+            --info-color: #0ea5e9;
+            --info-light: rgba(14, 165, 233, 0.1);
+            --danger-color: #ef4444;
+            --danger-light: rgba(239, 68, 68, 0.1);
+            --neutral-50: #f9fafb;
+            --neutral-100: #f3f4f6;
+            --neutral-200: #e5e7eb;
+            --neutral-300: #d1d5db;
+            --neutral-400: #9ca3af;
+            --neutral-500: #6b7280;
+            --neutral-600: #4b5563;
+            --neutral-700: #374151;
+            --neutral-800: #1f2937;
+            --neutral-900: #111827;
+            
+            /* 样式变量 */
+            --border-radius-sm: 0.25rem;
+            --border-radius: 0.375rem;
+            --border-radius-lg: 0.5rem;
+            --border-radius-xl: 0.75rem;
+            --border-radius-2xl: 1rem;
+            --card-shadow: 0 1px 3px rgba(0, 0, 0, 0.05), 0 1px 2px rgba(0, 0, 0, 0.1);
+            --card-shadow-hover: 0 10px 20px rgba(0, 0, 0, 0.05), 0 6px 6px rgba(0, 0, 0, 0.1);
+            --card-shadow-lg: 0 10px 25px -5px rgba(0, 0, 0, 0.1), 0 10px 10px -5px rgba(0, 0, 0, 0.04);
+            --transition-base: all 0.2s ease-in-out;
+            --transition-smooth: all 0.3s cubic-bezier(0.4, 0, 0.2, 1);
+            --font-family: 'Inter', 'PingFang SC', 'Helvetica Neue', 'Microsoft YaHei', sans-serif;
+        }
+        
+        /* 基础样式 */
+        body {
+            font-family: var(--font-family);
+            background-color: var(--neutral-50);
+            color: var(--neutral-800);
+            margin: 0;
+            padding: 0;
+            min-height: 100vh;
+            display: flex;
+            align-items: center;
+            justify-content: center;
+            -webkit-font-smoothing: antialiased;
+            -moz-osx-font-smoothing: grayscale;
+        }
+        
+        h1, h2, h3, h4, h5, h6 {
+            font-weight: 600;
+            color: var(--neutral-900);
+        }
+        
+        .text-gradient {
+            background: linear-gradient(135deg, var(--secondary-color), var(--secondary-light));
+            -webkit-background-clip: text;
+            -webkit-text-fill-color: transparent;
+            background-clip: text;
+            color: transparent;
+        }
+        
+        /* 登录容器样式 */
+        .login-container {
+            width: 100%;
+            max-width: 450px;
+            padding: 2.5rem;
+            background-color: white;
+            border-radius: var(--border-radius-xl);
+            box-shadow: var(--card-shadow-lg);
+            transition: var(--transition-smooth);
+            position: relative;
+            overflow: hidden;
+        }
+        
+        .login-container::before {
+            content: '';
+            position: absolute;
+            top: 0;
+            left: 0;
+            width: 100%;
+            height: 4px;
+            background: linear-gradient(to right, var(--secondary-color), var(--secondary-light));
+            border-radius: 4px 4px 0 0;
+        }
+        
+        .login-header {
+            text-align: center;
+            margin-bottom: 2.5rem;
+        }
+        
+        .login-header h1 {
+            font-size: 1.75rem;
+            font-weight: 700;
+            margin-bottom: 0.5rem;
+            background: linear-gradient(45deg, var(--primary-color), var(--secondary-color));
+            -webkit-background-clip: text;
+            -webkit-text-fill-color: transparent;
+            letter-spacing: -0.01em;
+        }
+        
+        .login-header p {
+            color: var(--neutral-600);
+            margin-bottom: 0;
+            font-size: 0.95rem;
+        }
+        
+        .user-type-switch {
+            margin-bottom: 2rem;
+        }
+        
+        .user-type-switch .btn-group {
+            width: 100%;
+            box-shadow: var(--card-shadow);
+            border-radius: var(--border-radius-lg);
+            padding: 0.25rem;
+            background-color: var(--neutral-100);
+        }
+        
+        .user-type-switch .btn {
+            flex: 1;
+            background: transparent;
+            border: none;
+            padding: 0.75rem 1rem;
+            font-weight: 500;
+            color: var(--neutral-700);
+            border-radius: var(--border-radius);
+            transition: var(--transition-base);
+        }
+        
+        .user-type-switch .btn:hover {
+            color: var(--primary-color);
+        }
+        
+        .user-type-switch .btn.active {
+            background: white;
+            color: var(--secondary-color);
+            box-shadow: var(--card-shadow);
+        }
+        
+        /* 表单样式 */
+        .form-floating {
+            margin-bottom: 1.25rem;
+        }
+        
+        .form-floating > .form-control {
+            padding: 1rem 1rem;
+            height: calc(3.5rem + 2px);
+            border-radius: var(--border-radius-lg);
+            border: 1px solid var(--neutral-200);
+            font-size: 0.95rem;
+            transition: var(--transition-base);
+        }
+        
+        .form-floating > .form-control:focus {
+            border-color: var(--secondary-color);
+            box-shadow: 0 0 0 3px rgba(74, 108, 253, 0.1);
+        }
+        
+        .form-floating > label {
+            padding: 1rem;
+            color: var(--neutral-500);
+        }
+        
+        .form-check {
+            display: flex;
+            align-items: center;
+            margin-bottom: 1.5rem;
+        }
+        
+        .form-check-input {
+            width: 1.25em;
+            height: 1.25em;
+            margin-right: 0.75rem;
+            background-color: white;
+            border: 1px solid var(--neutral-300);
+            border-radius: 0.25em;
+            transition: all 0.15s ease-in-out;
+        }
+        
+        .form-check-input:checked {
+            background-color: var(--secondary-color);
+            border-color: var(--secondary-color);
+        }
+        
+        .form-check-label {
+            color: var(--neutral-700);
+            font-size: 0.95rem;
+        }
+        
+        /* 按钮样式 */
+        .btn-primary {
+            background: linear-gradient(to right, var(--secondary-color), var(--secondary-light));
+            border: none;
+            width: 100%;
+            padding: 0.85rem;
+            font-weight: 500;
+            border-radius: var(--border-radius-lg);
+            transition: var(--transition-base);
+        }
+        
+        .btn-primary:hover, .btn-primary:focus {
+            background: linear-gradient(to right, var(--secondary-light), var(--secondary-color));
+            box-shadow: 0 4px 10px rgba(74, 108, 253, 0.3);
+            transform: translateY(-1px);
+        }
+        
+        .btn-outline-primary {
+            color: var(--secondary-color);
+            border-color: var(--secondary-color);
+            background-color: transparent;
+        }
+        
+        .btn-outline-primary:hover, .btn-outline-primary:focus {
+            background-color: var(--secondary-color);
+            border-color: var(--secondary-color);
+            color: white;
+            box-shadow: 0 4px 10px rgba(74, 108, 253, 0.2);
+        }
+        
+        /* 学生说明部分 */
+        .student-instructions {
+            margin-top: 1.5rem;
+            padding: 1.25rem;
+            background-color: var(--neutral-50);
+            border-radius: var(--border-radius-lg);
+            border: 1px solid var(--neutral-200);
+            font-size: 0.95rem;
+            color: var(--neutral-700);
+            transition: var(--transition-smooth);
+        }
+        
+        .student-instructions p {
+            margin-bottom: 0.75rem;
+        }
+        
+        .student-instructions p:last-child {
+            margin-bottom: 0;
+        }
+        
+        .student-instructions strong {
+            color: var(--neutral-900);
+        }
+        
+        /* 页脚样式 */
+        .login-footer {
+            text-align: center;
+            margin-top: 2rem;
+            color: var(--neutral-500);
+            font-size: 0.85rem;
+        }
+        
+        /* 动画效果 */
+        @keyframes fadeIn {
+            from { opacity: 0; transform: translateY(8px); }
+            to { opacity: 1; transform: translateY(0); }
+        }
+        
+        .fade-in {
+            opacity: 0;
+            animation: fadeIn 0.4s ease-out forwards;
+        }
+        
+        /* 输入组样式 */
+        .input-group {
+            position: relative;
+        }
+        
+        .input-group .form-control {
+            padding-right: 3.5rem;
+            border-radius: var(--border-radius-lg);
+        }
+        
+        .input-group .btn {
+            position: absolute;
+            right: 0;
+            top: 0;
+            bottom: 0;
+            z-index: 5;
+            border-radius: 0 var(--border-radius-lg) var(--border-radius-lg) 0;
+            color: var(--neutral-700);
+            background-color: var(--neutral-100);
+            border: 1px solid var(--neutral-200);
+            border-left: none;
+            transition: var(--transition-base);
+        }
+        
+        .input-group .btn:hover, .input-group .btn:focus {
+            background-color: var(--neutral-200);
+        }
+    </style>
+</head>
+<body>
+    <div class="login-container fade-in">
+        <div class="login-header">
+            <h1>AI 助教平台</h1>
+            <p id="login-subtitle">教师端登录</p>
+        </div>
+        
+        <div class="user-type-switch">
+            <div class="btn-group" role="group" id="user-type-buttons">
+                <button type="button" class="btn active" id="teacher-button">教师</button>
+                <button type="button" class="btn" id="student-button">学生</button>
+            </div>
+        </div>
+        
+        <form id="login-form">
+            <div class="form-floating">
+                <input type="text" class="form-control" id="username" placeholder="用户名" required>
+                <label for="username">用户名</label>
+            </div>
+            
+            <div class="form-floating">
+                <input type="password" class="form-control" id="password" placeholder="密码" required>
+                <label for="password">密码</label>
+            </div>
+            
+            <div class="form-check">
+                <input class="form-check-input" type="checkbox" id="remember-me">
+                <label class="form-check-label" for="remember-me">
+                    记住我
+                </label>
+            </div>
+            
+            <div id="student-instructions" class="student-instructions" style="display: none;">
+                <p><strong>学生登录说明：</strong></p>
+                <p>您也可以直接使用教师分享的链接访问AI助手，无需登录。</p>
+                <div class="input-group mt-3">
+                    <input type="text" class="form-control" id="access-token" placeholder="输入访问令牌">
+                    <button class="btn" type="button" id="access-button">
+                        <i class="bi bi-arrow-right"></i>
+                    </button>
+                </div>
+            </div>
+            
+            <button type="submit" class="btn btn-primary mt-3">
+                <i class="bi bi-box-arrow-in-right me-2"></i>登录
+            </button>
+        </form>
+        
+        <div class="login-footer">
+            © 2025 教育AI助手平台 版权所有
+        </div>
+    </div>
+    
+    <script>
+        // 硬编码用户
+        const users = {
+            teachers: [
+                { username: 'teacher', password: '123456', name: '李志刚' },
+                { username: 'admin', password: 'admin123', name: '管理员' }
+            ],
+            students: [
+                { username: 'student1', password: '123456', name: '张三' },
+                { username: 'student2', password: '123456', name: '李四' }
+            ]
+        };
+        
+        // DOM元素
+        const teacherButton = document.getElementById('teacher-button');
+        const studentButton = document.getElementById('student-button');
+        const loginSubtitle = document.getElementById('login-subtitle');
+        const studentInstructions = document.getElementById('student-instructions');
+        const loginForm = document.getElementById('login-form');
+        const accessButton = document.getElementById('access-button');
+        
+        // 切换用户类型
+        teacherButton.addEventListener('click', function() {
+            teacherButton.classList.add('active');
+            studentButton.classList.remove('active');
+            loginSubtitle.textContent = '教师端登录';
+            studentInstructions.style.display = 'none';
+        });
+        
+        studentButton.addEventListener('click', function() {
+            studentButton.classList.add('active');
+            teacherButton.classList.remove('active');
+            loginSubtitle.textContent = '学生端登录';
+            studentInstructions.style.display = 'block';
+        });
+        
+        // 处理登录表单提交
+        // 处理登录表单提交
+loginForm.addEventListener('submit', async function(e) {
+    e.preventDefault();
+    
+    const username = document.getElementById('username').value;
+    const password = document.getElementById('password').value;
+    const isTeacher = teacherButton.classList.contains('active');
+    
+    try {
+        const response = await fetch('/api/auth/login', {
+            method: 'POST',
+            headers: {
+                'Content-Type': 'application/json'
+            },
+            body: JSON.stringify({
+                username: username,
+                password: password,
+                type: isTeacher ? 'teacher' : 'student'
+            })
+        });
+        
+        const data = await response.json();
+        
+        if (data.success) {
+            // 登录成功，重定向到相应页面
+            window.location.href = isTeacher ? '/index.html' : '/student_portal.html';
+        } else {
+            showAlert('用户名或密码错误', 'danger');
+        }
+    } catch (error) {
+        console.error('登录失败:', error);
+        showAlert('登录请求失败，请重试', 'danger');
+    }
+});
+        // 处理访问令牌
+        accessButton.addEventListener('click', function() {
+            const token = document.getElementById('access-token').value.trim();
+            
+            if (token) {
+                // 简单验证令牌格式
+                if (token.length >= 32) {
+                    // 这里应该向服务器验证令牌有效性，这里简化处理
+                    // 直接从URL中提取agent_id (假设格式为: [agent_id]?token=[token])
+                    if (token.includes('?token=')) {
+                        window.location.href = '/student/' + token;
+                    } else {
+                        // 假设这是一个纯token，需要输入Agent ID
+                        const agentId = prompt('请输入Agent ID');
+                        if (agentId) {
+                            window.location.href = `/student/${agentId}?token=${token}`;
+                        }
+                    }
+                } else {
+                    showAlert('无效的访问令牌格式', 'warning');
+                }
+            } else {
+                showAlert('请输入访问令牌', 'warning');
+            }
+        });
+        
+        // 显示提示信息
+        function showAlert(message, type) {
+            // 移除现有的提示
+            const existingAlert = document.querySelector('.alert');
+            if (existingAlert) {
+                existingAlert.remove();
+            }
+            
+            // 创建新提示
+            const alert = document.createElement('div');
+            alert.className = `alert alert-${type} fade-in`;
+            alert.role = 'alert';
+            alert.style.marginBottom = '1.5rem';
+            alert.innerHTML = `
+                <i class="bi ${type === 'danger' ? 'bi-exclamation-triangle' : 'bi-info-circle'} me-2"></i>
+                ${message}
+            `;
+            
+            // 添加到表单上方
+            loginForm.insertAdjacentElement('beforebegin', alert);
+            
+            // 3秒后自动移除
+            setTimeout(() => {
+                alert.style.opacity = '0';
+                setTimeout(() => alert.remove(), 300);
+            }, 3000);
+        }
+        
+        // 页面加载时检查是否已经登录
+        // window.addEventListener('DOMContentLoaded', function() {
+        //     const currentUser = localStorage.getItem('currentUser');
+            
+        //     if (currentUser) {
+        //         const userData = JSON.parse(currentUser);
+                
+        //         if (userData.type === 'teacher') {
+        //             window.location.href = '/index.html';
+        //         } else if (userData.type === 'student') {
+        //             window.location.href = '/student_portal.html';
+        //         }
+        //     }
+        // });
+    </script>
+</body>
+</html>

templates/student_portal.html

@@ -0,0 +1,1005 @@
+<!DOCTYPE html>
+<html lang="zh-CN">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <meta http-equiv="Content-Security-Policy" content="upgrade-insecure-requests">
+    <title>学生端门户 - 教育AI助手平台</title>
+    <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/[email protected]/dist/css/bootstrap.min.css">
+    <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/[email protected]/font/bootstrap-icons.css">
+    <style>
+        @import url('https://fonts.googleapis.com/css2?family=Inter:wght@300;400;500;600;700&display=swap');
+        
+        :root {
+            /* 优雅的配色方案 */
+            --primary-color: #0f2d49;
+            --primary-light: #234a70;
+            --secondary-color: #4a6cfd;
+            --secondary-light: #7b91ff;
+            --tertiary-color: #f7f9fe;
+            --success-color: #10b981;
+            --success-light: rgba(16, 185, 129, 0.1);
+            --warning-color: #f59e0b;
+            --warning-light: rgba(245, 158, 11, 0.1);
+            --info-color: #0ea5e9;
+            --info-light: rgba(14, 165, 233, 0.1);
+            --danger-color: #ef4444;
+            --danger-light: rgba(239, 68, 68, 0.1);
+            --neutral-50: #f9fafb;
+            --neutral-100: #f3f4f6;
+            --neutral-200: #e5e7eb;
+            --neutral-300: #d1d5db;
+            --neutral-400: #9ca3af;
+            --neutral-500: #6b7280;
+            --neutral-600: #4b5563;
+            --neutral-700: #374151;
+            --neutral-800: #1f2937;
+            --neutral-900: #111827;
+            
+            /* 样式变量 */
+            --border-radius-sm: 0.25rem;
+            --border-radius: 0.375rem;
+            --border-radius-lg: 0.5rem;
+            --border-radius-xl: 0.75rem;
+            --border-radius-2xl: 1rem;
+            --card-shadow: 0 1px 3px rgba(0, 0, 0, 0.05), 0 1px 2px rgba(0, 0, 0, 0.1);
+            --card-shadow-hover: 0 10px 20px rgba(0, 0, 0, 0.05), 0 6px 6px rgba(0, 0, 0, 0.1);
+            --card-shadow-lg: 0 10px 25px -5px rgba(0, 0, 0, 0.1), 0 10px 10px -5px rgba(0, 0, 0, 0.04);
+            --transition-base: all 0.2s ease-in-out;
+            --transition-smooth: all 0.3s cubic-bezier(0.4, 0, 0.2, 1);
+            --font-family: 'Inter', 'PingFang SC', 'Helvetica Neue', 'Microsoft YaHei', sans-serif;
+        }
+        
+        /* 基础样式 */
+        body {
+            font-family: var(--font-family);
+            background-color: var(--neutral-50);
+            color: var(--neutral-800);
+            margin: 0;
+            padding: 0;
+            min-height: 100vh;
+            display: flex;
+            flex-direction: column;
+            -webkit-font-smoothing: antialiased;
+            -moz-osx-font-smoothing: grayscale;
+        }
+        
+        h1, h2, h3, h4, h5, h6 {
+            font-weight: 600;
+            color: var(--neutral-900);
+        }
+        
+        .text-gradient {
+            background: linear-gradient(135deg, var(--secondary-color), var(--secondary-light));
+            -webkit-background-clip: text;
+            -webkit-text-fill-color: transparent;
+            background-clip: text;
+            color: transparent;
+        }
+        
+        /* 滚动条样式 */
+        ::-webkit-scrollbar {
+            width: 6px;
+            height: 6px;
+        }
+        
+        ::-webkit-scrollbar-track {
+            background: var(--neutral-100);
+            border-radius: 10px;
+        }
+        
+        ::-webkit-scrollbar-thumb {
+            background: var(--neutral-300);
+            border-radius: 10px;
+        }
+        
+        ::-webkit-scrollbar-thumb:hover {
+            background: var(--neutral-400);
+        }
+        
+        /* 头部样式 */
+        .header {
+            background-color: white;
+            border-bottom: 1px solid var(--neutral-200);
+            padding: 1rem 1.5rem;
+            box-shadow: 0 2px 4px rgba(0, 0, 0, 0.05);
+            position: sticky;
+            top: 0;
+            z-index: 100;
+        }
+        
+        .header-content {
+            max-width: 1400px;
+            margin: 0 auto;
+            display: flex;
+            justify-content: space-between;
+            align-items: center;
+        }
+        
+        .header h1 {
+            margin: 0;
+            font-size: 1.5rem;
+            font-weight: 700;
+            background: linear-gradient(45deg, var(--primary-color), var(--secondary-color));
+            -webkit-background-clip: text;
+            -webkit-text-fill-color: transparent;
+            letter-spacing: -0.01em;
+        }
+        
+        .user-info {
+            display: flex;
+            align-items: center;
+            gap: 0.75rem;
+        }
+        
+        .user-info .badge {
+            font-size: 0.85rem;
+            padding: 0.4rem 0.75rem;
+            border-radius: var(--border-radius-lg);
+            background: linear-gradient(to right, var(--secondary-color), var(--secondary-light));
+            color: white;
+            font-weight: 500;
+        }
+        
+        .user-info button {
+            color: var(--neutral-600);
+            text-decoration: none;
+            font-size: 0.9rem;
+            transition: var(--transition-base);
+        }
+        
+        .user-info button:hover {
+            color: var(--secondary-color);
+        }
+        
+        /* 主容器样式 */
+        .main-container {
+            flex: 1;
+            max-width: 1400px;
+            margin: 2rem auto;
+            padding: 0 1.5rem;
+            width: 100%;
+            box-sizing: border-box;
+        }
+        
+        /* 欢迎消息样式 */
+        .welcome-message {
+            background-color: white;
+            border-radius: var(--border-radius-xl);
+            box-shadow: var(--card-shadow);
+            padding: 2rem;
+            margin-bottom: 2.5rem;
+            position: relative;
+            overflow: hidden;
+            border: none;
+            transition: var(--transition-smooth);
+        }
+        
+        .welcome-message:hover {
+            box-shadow: var(--card-shadow-hover);
+            transform: translateY(-3px);
+        }
+        
+        .welcome-message::before {
+            content: '';
+            position: absolute;
+            top: 0;
+            bottom: 0;
+            left: 0;
+            width: 4px;
+            background: linear-gradient(to bottom, var(--info-color), var(--secondary-light));
+            border-radius: 4px 0 0 4px;
+        }
+        
+        .welcome-message h2 {
+            margin-top: 0;
+            margin-bottom: 1rem;
+            font-size: 1.6rem;
+            font-weight: 700;
+            color: var(--primary-color);
+        }
+        
+        .welcome-message p {
+            color: var(--neutral-600);
+            margin-bottom: 0.75rem;
+            font-size: 0.95rem;
+            line-height: 1.5;
+        }
+        
+        .welcome-message p:last-child {
+            margin-bottom: 0;
+        }
+        
+        /* 令牌输入样式 */
+        .token-input {
+            background-color: white;
+            border-radius: var(--border-radius-xl);
+            box-shadow: var(--card-shadow);
+            padding: 2rem;
+            margin-bottom: 2.5rem;
+            position: relative;
+            overflow: hidden;
+            transition: var(--transition-smooth);
+        }
+        
+        .token-input:hover {
+            box-shadow: var(--card-shadow-hover);
+            transform: translateY(-3px);
+        }
+        
+        .token-input::before {
+            content: '';
+            position: absolute;
+            top: 0;
+            bottom: 0;
+            left: 0;
+            width: 4px;
+            background: linear-gradient(to bottom, var(--warning-color), var(--secondary-light));
+            border-radius: 4px 0 0 4px;
+        }
+        
+        .token-input h3 {
+            margin-top: 0;
+            margin-bottom: 1rem;
+            font-size: 1.35rem;
+            font-weight: 600;
+            color: var(--neutral-900);
+        }
+        
+        .token-input p {
+            color: var(--neutral-600);
+            margin-bottom: 1.25rem;
+        }
+        
+        .token-input .input-group {
+            box-shadow: 0 2px 5px rgba(0, 0, 0, 0.05);
+            border-radius: var(--border-radius-lg);
+            margin-bottom: 0;
+        }
+        
+        .token-input .form-control {
+            border-radius: var(--border-radius-lg) 0 0 var(--border-radius-lg);
+            border: 1px solid var(--neutral-200);
+            padding: 0.75rem 1rem;
+            font-size: 0.95rem;
+        }
+        
+        .token-input .form-control:focus {
+            box-shadow: none;
+            border-color: var(--secondary-color);
+        }
+        
+        .token-input .btn {
+            background: linear-gradient(to right, var(--secondary-color), var(--secondary-light));
+            color: white;
+            border: none;
+            border-radius: 0 var(--border-radius-lg) var(--border-radius-lg) 0;
+            padding: 0.75rem 1.25rem;
+            transition: var(--transition-base);
+        }
+        
+        .token-input .btn:hover {
+            background: linear-gradient(to right, var(--secondary-light), var(--secondary-color));
+            box-shadow: 0 4px 10px rgba(74, 108, 253, 0.2);
+        }
+        
+        /* 标题样式 */
+        .section-header {
+            margin-bottom: 1.5rem;
+            display: flex;
+            flex-direction: column;
+        }
+        
+        .section-header h2 {
+            margin: 0;
+            font-size: 1.5rem;
+            font-weight: 700;
+            color: var(--primary-color);
+            margin-bottom: 0.5rem;
+        }
+        
+        .section-header p {
+            color: var(--neutral-600);
+            margin: 0;
+            font-size: 0.95rem;
+        }
+        
+        /* Agent卡片样式 */
+        .agent-card {
+            background-color: white;
+            border-radius: var(--border-radius-xl);
+            margin-bottom: 1.5rem;
+            box-shadow: var(--card-shadow);
+            overflow: hidden;
+            transition: var(--transition-smooth);
+            position: relative;
+            border: none;
+        }
+        
+        .agent-card:hover {
+            transform: translateY(-5px);
+            box-shadow: var(--card-shadow-hover);
+        }
+        
+        .agent-card::before {
+            content: '';
+            position: absolute;
+            top: 0;
+            bottom: 0;
+            left: 0;
+            width: 4px;
+            background: linear-gradient(to bottom, var(--secondary-color), var(--secondary-light));
+            border-radius: 4px 0 0 4px;
+        }
+        
+        .agent-card-header {
+            padding: 1.5rem;
+            background-color: rgba(74, 108, 253, 0.03);
+            border-bottom: 1px solid var(--neutral-100);
+        }
+        
+        .agent-card-title {
+            margin: 0;
+            font-size: 1.25rem;
+            font-weight: 600;
+            color: var(--primary-color);
+            display: flex;
+            align-items: center;
+        }
+        
+        .agent-card-title i {
+            margin-right: 0.75rem;
+            font-size: 1.2rem;
+            color: var(--secondary-color);
+        }
+        
+        .agent-card-content {
+            padding: 1.5rem;
+        }
+        
+        .agent-card-description {
+            margin-bottom: 1.25rem;
+            color: var(--neutral-700);
+            font-size: 0.95rem;
+            line-height: 1.5;
+        }
+        
+        .agent-meta {
+            display: flex;
+            flex-wrap: wrap;
+            margin-bottom: 1.25rem;
+            gap: 1.5rem;
+        }
+        
+        .agent-meta-item {
+            display: flex;
+            align-items: center;
+            font-size: 0.9rem;
+            color: var(--neutral-600);
+        }
+        
+        .agent-meta-item i {
+            margin-right: 0.5rem;
+            font-size: 0.95rem;
+            color: var(--secondary-color);
+        }
+        
+        .agent-tags {
+            display: flex;
+            flex-wrap: wrap;
+            gap: 0.5rem;
+            margin-bottom: 1.5rem;
+        }
+        
+        .agent-tag {
+            display: inline-flex;
+            align-items: center;
+            padding: 0.35em 0.75em;
+            background-color: rgba(74, 108, 253, 0.1);
+            color: var(--secondary-color);
+            border-radius: 20px;
+            font-size: 0.85rem;
+            font-weight: 500;
+        }
+        
+        .agent-tag i {
+            margin-right: 0.4rem;
+        }
+        
+        .agent-actions {
+            display: flex;
+            justify-content: flex-end;
+        }
+        
+        .agent-actions .btn {
+            padding: 0.65em 1.25em;
+            font-weight: 500;
+            background: linear-gradient(to right, var(--secondary-color), var(--secondary-light));
+            border: none;
+            color: white;
+            border-radius: var(--border-radius-lg);
+            transition: var(--transition-base);
+        }
+        
+        .agent-actions .btn:hover {
+            background: linear-gradient(to right, var(--secondary-light), var(--secondary-color));
+            transform: translateY(-2px);
+            box-shadow: 0 4px 10px rgba(74, 108, 253, 0.2);
+        }
+        
+        .agent-actions .btn i {
+            margin-right: 0.5rem;
+        }
+        
+        /* 活动记录样式 */
+        .recent-activity {
+            background-color: white;
+            border-radius: var(--border-radius-xl);
+            box-shadow: var(--card-shadow);
+            padding: 1.5rem;
+            margin-bottom: 2rem;
+            position: relative;
+            overflow: hidden;
+            transition: var(--transition-smooth);
+        }
+        
+        .recent-activity:hover {
+            box-shadow: var(--card-shadow-hover);
+            transform: translateY(-3px);
+        }
+        
+        .recent-activity::before {
+            content: '';
+            position: absolute;
+            top: 0;
+            bottom: 0;
+            left: 0;
+            width: 4px;
+            background: linear-gradient(to bottom, var(--success-color), var(--secondary-light));
+            border-radius: 4px 0 0 4px;
+        }
+        
+        .recent-activity h3 {
+            margin-top: 0;
+            margin-bottom: 1.5rem;
+            font-size: 1.35rem;
+            font-weight: 600;
+            color: var(--neutral-900);
+        }
+        
+        .activity-list {
+            list-style: none;
+            padding: 0;
+            margin: 0;
+        }
+        
+        .activity-item {
+            display: flex;
+            padding: 1rem 0;
+            border-bottom: 1px solid var(--neutral-100);
+            align-items: center;
+        }
+        
+        .activity-item:last-child {
+            border-bottom: none;
+            padding-bottom: 0;
+        }
+        
+        .activity-icon {
+            width: 40px;
+            height: 40px;
+            border-radius: 50%;
+            background-color: rgba(74, 108, 253, 0.1);
+            display: flex;
+            align-items: center;
+            justify-content: center;
+            margin-right: 1rem;
+            flex-shrink: 0;
+        }
+        
+        .activity-icon i {
+            color: var(--secondary-color);
+            font-size: 1.1rem;
+        }
+        
+        .activity-content {
+            flex: 1;
+        }
+        
+        .activity-title {
+            margin: 0;
+            margin-bottom: 0.35rem;
+            font-size: 0.95rem;
+            font-weight: 500;
+            color: var(--neutral-800);
+        }
+        
+        .activity-time {
+            font-size: 0.85rem;
+            color: var(--neutral-500);
+        }
+        
+        /* 动画效果 */
+        @keyframes fadeIn {
+            from { opacity: 0; transform: translateY(10px); }
+            to { opacity: 1; transform: translateY(0); }
+        }
+        
+        .fade-in {
+            opacity: 0;
+            animation: fadeIn 0.4s ease-out forwards;
+        }
+        
+        .animate-delay-1 { animation-delay: 0.1s; }
+        .animate-delay-2 { animation-delay: 0.2s; }
+        .animate-delay-3 { animation-delay: 0.3s; }
+        .animate-delay-4 { animation-delay: 0.4s; }
+        .animate-delay-5 { animation-delay: 0.5s; }
+        
+        /* 响应式样式 */
+        @media (max-width: 992px) {
+            .main-container {
+                padding: 0 1rem;
+            }
+            
+            .agent-meta {
+                flex-direction: column;
+                gap: 0.75rem;
+            }
+        }
+        
+        @media (max-width: 768px) {
+            .header h1 {
+                font-size: 1.25rem;
+            }
+            
+            .welcome-message h2 {
+                font-size: 1.35rem;
+            }
+            
+            .agent-card-title {
+                font-size: 1.1rem;
+            }
+            
+            .section-header h2 {
+                font-size: 1.3rem;
+            }
+            
+            .recent-activity h3,
+            .token-input h3 {
+                font-size: 1.2rem;
+            }
+        }
+    </style>
+</head>
+<body>
+    <header class="header">
+        <div class="header-content">
+            <h1>教育AI助手平台 - 学生端</h1>
+            <div class="user-info">
+                <span class="badge" id="user-name">张三</span>
+                <button class="btn btn-sm btn-link text-decoration-none" id="logout-btn">登出</button>
+            </div>
+        </div>
+    </header>
+    
+    <div class="main-container">
+        <div class="welcome-message fade-in">
+            <h2>欢迎回来，<span id="welcome-name">张三</span>！</h2>
+            <p>教育AI助手平台为您提供智能学习辅助工具，帮助您更高效地学习和解决问题。</p>
+            <p>下方是您的教师分享给您的AI助手，点击"开始对话"与它们互动。</p>
+        </div>
+        
+        <div class="token-input fade-in animate-delay-1">
+            <h3>访问新的AI助手</h3>
+            <p>输入教师分享的访问令牌，即可使用新的AI助手。</p>
+            <div class="input-group">
+                <input type="text" class="form-control" id="access-token" placeholder="输入访问令牌">
+                <button class="btn" id="access-button">
+                    <i class="bi bi-arrow-right-circle me-1"></i> 访问
+                </button>
+            </div>
+        </div>
+        
+        <div class="section-header fade-in animate-delay-2">
+            <h2>我的AI助手</h2>
+            <p>教师分享给您的智能学习助手</p>
+        </div>
+        
+        <div class="row" id="agents-container">
+            <!-- 代码将在JavaScript中动态生成 -->
+        </div>
+        
+        <div class="recent-activity fade-in animate-delay-4">
+            <h3>最近活动</h3>
+            <ul class="activity-list" id="activity-list">
+                <!-- 代码将在JavaScript中动态生成 -->
+            </ul>
+        </div>
+    </div>
+    
+    <script>
+        // DOM元素
+        const userNameElement = document.getElementById('user-name');
+        const welcomeNameElement = document.getElementById('welcome-name');
+        const logoutBtn = document.getElementById('logout-btn');
+        const accessButton = document.getElementById('access-button');
+        const agentsContainer = document.getElementById('agents-container');
+        const activityList = document.getElementById('activity-list');
+        
+        // 页面加载函数
+        document.addEventListener('DOMContentLoaded', function() {
+            // 检查登录状态
+            checkAuthStatus();
+            
+            // 加载用户的AI助手列表
+            loadAgents();
+            
+            // 初始化登出按钮
+            logoutBtn.addEventListener('click', logout);
+            
+            // 初始化访问令牌按钮
+            accessButton.addEventListener('click', accessWithToken);
+        });
+        
+        // 检查认证状态
+        async function checkAuthStatus() {
+            try {
+                const response = await fetch('/api/auth/check');
+                const data = await response.json();
+                
+                if (!data.success) {
+                    // 未登录，跳转到登录页面
+                    window.location.href = '/login.html';
+                    return;
+                }
+                
+                // 更新用户信息显示
+                userNameElement.textContent = data.user.name;
+                welcomeNameElement.textContent = data.user.name;
+                
+                // 如果不是学生，跳转到教师端
+                if (data.user.type !== 'student') {
+                    window.location.href = '/index.html';
+                }
+            } catch (error) {
+                console.error('验证登录状态出错:', error);
+                window.location.href = '/login.html';
+            }
+        }
+        
+        // 登出
+        async function logout() {
+            try {
+                const response = await fetch('/api/auth/logout', {
+                    method: 'POST'
+                });
+                
+                const data = await response.json();
+                
+                if (data.success) {
+                    window.location.href = '/login.html';
+                }
+            } catch (error) {
+                console.error('登出出错:', error);
+                alert('登出失败，请重试');
+            }
+        }
+        
+        // 加载Agent列表
+        async function loadAgents() {
+            try {
+                // 显示加载状态
+                agentsContainer.innerHTML = `
+                    <div class="col-12 text-center py-4">
+                        <div class="spinner-border text-primary" role="status">
+                            <span class="visually-hidden">加载中...</span>
+                        </div>
+                        <p class="mt-3">正在加载AI助手列表...</p>
+                    </div>
+                `;
+                
+                // 获取Agent列表
+                const response = await fetch('/api/student/agents');
+                const result = await response.json();
+                
+                if (result.success) {
+                    const agents = result.agents || [];
+                    
+                    if (agents.length === 0) {
+                        agentsContainer.innerHTML = `
+                            <div class="col-12">
+                                <div class="alert alert-info">
+                                    <i class="bi bi-info-circle me-2"></i>
+                                    您还没有可用的AI助手。请向您的教师获取访问令牌。
+                                </div>
+                            </div>
+                        `;
+                        
+                        // 无Agent时，清空活动列表
+                        activityList.innerHTML = `
+                            <li class="text-center py-3 text-muted">
+                                暂无活动记录
+                            </li>
+                        `;
+                        
+                        return;
+                    }
+                    
+                    // 渲染Agent列表
+                    agentsContainer.innerHTML = '';
+                    
+                    agents.forEach(agent => {
+                        const col = document.createElement('div');
+                        col.className = 'col-lg-6 col-12 fade-in';
+                        
+                        // 格式化最后使用时间
+                        let lastUsedText = '从未使用';
+                        if (agent.last_used) {
+                            const lastUsedDate = new Date(agent.last_used * 1000);
+                            lastUsedText = lastUsedDate.toLocaleString();
+                        }
+                        
+                        // 构建插件标签
+                        let pluginsHtml = '';
+                        if (agent.plugins && agent.plugins.length > 0) {
+                            pluginsHtml = '<div class="agent-tags">';
+                            
+                            agent.plugins.forEach(plugin => {
+                                let pluginName = '未知插件';
+                                let pluginIcon = 'puzzle';
+                                
+                                if (plugin === 'code') {
+                                    pluginName = '代码执行';
+                                    pluginIcon = 'code-square';
+                                } else if (plugin === 'visualization') {
+                                    pluginName = '3D可视化';
+                                    pluginIcon = 'bar-chart';
+                                } else if (plugin === 'mindmap') {
+                                    pluginName = '思维导图';
+                                    pluginIcon = 'diagram-3';
+                                }
+                                
+                                pluginsHtml += `
+                                    <span class="agent-tag">
+                                        <i class="bi bi-${pluginIcon}"></i> ${pluginName}
+                                    </span>
+                                `;
+                            });
+                            
+                            pluginsHtml += '</div>';
+                        }
+                        
+                        // 构建主题和教师标签
+                        let metaHtml = '<div class="agent-meta">';
+                        
+                        if (agent.subject) {
+                            metaHtml += `
+                                <div class="agent-meta-item">
+                                    <i class="bi bi-book"></i> ${agent.subject}
+                                </div>
+                            `;
+                        }
+                        
+                        if (agent.instructor) {
+                            metaHtml += `
+                                <div class="agent-meta-item">
+                                    <i class="bi bi-person"></i> ${agent.instructor}
+                                </div>
+                            `;
+                        }
+                        
+                        metaHtml += `
+                            <div class="agent-meta-item">
+                                <i class="bi bi-clock"></i> 最后使用: ${lastUsedText}
+                            </div>
+                        </div>`;
+                        
+                        // 构建Agent卡片
+                        col.innerHTML = `
+                            <div class="agent-card">
+                                <div class="agent-card-header">
+                                    <h3 class="agent-card-title">
+                                        <i class="bi bi-robot"></i> ${agent.name}
+                                    </h3>
+                                </div>
+                                <div class="agent-card-content">
+                                    <div class="agent-card-description">
+                                        ${agent.description || '暂无描述'}
+                                    </div>
+                                    ${metaHtml}
+                                    ${pluginsHtml}
+                                    <div class="agent-actions">
+                                        <a href="/student/${agent.id}?token=${agent.token}" class="btn">
+                                            <i class="bi bi-chat-dots me-1"></i> 开始对话
+                                        </a>
+                                    </div>
+                                </div>
+                            </div>
+                        `;
+                        
+                        agentsContainer.appendChild(col);
+                    });
+                    
+                    // 生成简单的活动记录（实际中应从API获取）
+                    loadActivityRecords();
+                } else {
+                    agentsContainer.innerHTML = `
+                        <div class="col-12">
+                            <div class="alert alert-danger">
+                                <i class="bi bi-exclamation-triangle me-2"></i>
+                                加载Agent列表失败: ${result.message}
+                            </div>
+                        </div>
+                    `;
+                    
+                    // 加载失败时，清空活动列表
+                    activityList.innerHTML = `
+                        <li class="text-center py-3 text-muted">
+                            暂无活动记录
+                        </li>
+                    `;
+                }
+            } catch (error) {
+                console.error('加载Agent列表出错:', error);
+                agentsContainer.innerHTML = `
+                    <div class="col-12">
+                        <div class="alert alert-danger">
+                            <i class="bi bi-exclamation-triangle me-2"></i>
+                            加载Agent列表时发生错误，请刷新页面重试
+                        </div>
+                    </div>
+                `;
+            }
+        }
+        
+        // 加载活动记录（实际中应从API获取）
+        // 加载活动记录
+async function loadActivityRecords() {
+    try {
+        // 从API获取活动记录
+        const response = await fetch('/api/student/activities');
+        const result = await response.json();
+        
+        if (result.success) {
+            const activities = result.activities || [];
+            
+            if (activities.length === 0) {
+                activityList.innerHTML = `
+                    <li class="text-center py-3 text-muted">
+                        暂无活动记录
+                    </li>
+                `;
+                return;
+            }
+            
+            // 渲染活动记录
+            activityList.innerHTML = '';
+            
+            activities.forEach(activity => {
+                let iconClass = 'bi-activity';
+                
+                switch (activity.type) {
+                    case 'chat':
+                        iconClass = 'bi-chat-dots';
+                        break;
+                    case 'code':
+                        iconClass = 'bi-code-square';
+                        break;
+                    case 'viz':
+                        iconClass = 'bi-bar-chart';
+                        break;
+                    case 'mindmap':
+                        iconClass = 'bi-diagram-3';
+                        break;
+                }
+                
+                const li = document.createElement('li');
+                li.className = 'activity-item';
+                
+                // 如果有agent_id和token，添加链接
+                let titleHtml = `<h4 class="activity-title">${activity.title}</h4>`;
+                if (activity.agent_id) {
+                    // 查找对应Agent的token
+                    const agent = agents.find(a => a.id === activity.agent_id);
+                    if (agent && agent.token) {
+                        titleHtml = `
+                            <h4 class="activity-title">
+                                <a href="/student/${activity.agent_id}?token=${agent.token}">${activity.title}</a>
+                            </h4>
+                        `;
+                    }
+                }
+                
+                li.innerHTML = `
+                    <div class="activity-icon">
+                        <i class="bi ${iconClass}"></i>
+                    </div>
+                    <div class="activity-content">
+                        ${titleHtml}
+                        <div class="activity-time">${activity.time}</div>
+                    </div>
+                `;
+                
+                activityList.appendChild(li);
+            });
+        } else {
+            activityList.innerHTML = `
+                <li class="text-center py-3 text-muted">
+                    <i class="bi bi-exclamation-circle me-2"></i>
+                    无法加载活动记录
+                </li>
+            `;
+        }
+    } catch (error) {
+        console.error('加载活动记录出错:', error);
+        activityList.innerHTML = `
+            <li class="text-center py-3 text-muted">
+                <i class="bi bi-exclamation-circle me-2"></i>
+                加载活动记录时出错
+            </li>
+        `;
+    }
+}
+        // 使用令牌访问AI助手
+        async function accessWithToken() {
+            const token = document.getElementById('access-token').value.trim();
+            
+            if (!token) {
+                showMessage('请输入访问令牌', 'warning');
+                return;
+            }
+            
+            try {
+                // 验证令牌有效性
+                const response = await fetch('/api/verify_token', {
+                    method: 'POST',
+                    headers: {
+                        'Content-Type': 'application/json'
+                    },
+                    body: JSON.stringify({ token })
+                });
+                
+                const result = await response.json();
+                
+                if (result.success) {
+                    // 令牌有效，重定向到Agent页面
+                    const agent = result.agent;
+                    window.location.href = `/student/${agent.id}?token=${token}`;
+                } else {
+                    showMessage(result.message || '无效的访问令牌', 'danger');
+                }
+            } catch (error) {
+                console.error('验证令牌出错:', error);
+                showMessage('验证令牌时发生错误，请重试', 'danger');
+            }
+        }
+        
+        // 显示提示消息
+        function showMessage(message, type) {
+            // 移除现有的提示
+            const existingAlert = document.querySelector('.token-input .alert');
+            if (existingAlert) {
+                existingAlert.remove();
+            }
+            
+            // 创建新提示
+            const alertDiv = document.createElement('div');
+            alertDiv.className = `alert alert-${type} mt-3`;
+            alertDiv.innerHTML = `
+                <i class="bi ${type === 'danger' ? 'bi-exclamation-triangle' : 'bi-info-circle'} me-2"></i>
+                ${message}
+            `;
+            
+            // 添加到令牌输入区域
+            document.querySelector('.token-input .input-group').insertAdjacentElement('afterend', alertDiv);
+            
+            // 3秒后自动移除
+            setTimeout(() => {
+                alertDiv.style.opacity = '0';
+                setTimeout(() => alertDiv.remove(), 300);
+            }, 3000);
+        }
+    </script>
+</body>
+</html>

templates/token_verification.html

@@ -0,0 +1,522 @@
+<!DOCTYPE html>
+<html lang="zh-CN">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <meta http-equiv="Content-Security-Policy" content="upgrade-insecure-requests">
+    <title>访问验证 - 教育AI助手平台</title>
+    <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/[email protected]/dist/css/bootstrap.min.css">
+    <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/[email protected]/font/bootstrap-icons.css">
+    <style>
+        @import url('https://fonts.googleapis.com/css2?family=Inter:wght@300;400;500;600;700&display=swap');
+        
+        :root {
+            /* 优雅的配色方案 */
+            --primary-color: #0f2d49;
+            --primary-light: #234a70;
+            --secondary-color: #4a6cfd;
+            --secondary-light: #7b91ff;
+            --tertiary-color: #f7f9fe;
+            --success-color: #10b981;
+            --success-light: rgba(16, 185, 129, 0.1);
+            --warning-color: #f59e0b;
+            --warning-light: rgba(245, 158, 11, 0.1);
+            --info-color: #0ea5e9;
+            --info-light: rgba(14, 165, 233, 0.1);
+            --danger-color: #ef4444;
+            --danger-light: rgba(239, 68, 68, 0.1);
+            --neutral-50: #f9fafb;
+            --neutral-100: #f3f4f6;
+            --neutral-200: #e5e7eb;
+            --neutral-300: #d1d5db;
+            --neutral-400: #9ca3af;
+            --neutral-500: #6b7280;
+            --neutral-600: #4b5563;
+            --neutral-700: #374151;
+            --neutral-800: #1f2937;
+            --neutral-900: #111827;
+            
+            /* 样式变量 */
+            --border-radius-sm: 0.25rem;
+            --border-radius: 0.375rem;
+            --border-radius-lg: 0.5rem;
+            --border-radius-xl: 0.75rem;
+            --border-radius-2xl: 1rem;
+            --card-shadow: 0 1px 3px rgba(0, 0, 0, 0.05), 0 1px 2px rgba(0, 0, 0, 0.1);
+            --card-shadow-hover: 0 10px 20px rgba(0, 0, 0, 0.05), 0 6px 6px rgba(0, 0, 0, 0.1);
+            --card-shadow-lg: 0 10px 25px -5px rgba(0, 0, 0, 0.1), 0 10px 10px -5px rgba(0, 0, 0, 0.04);
+            --transition-base: all 0.2s ease-in-out;
+            --transition-smooth: all 0.3s cubic-bezier(0.4, 0, 0.2, 1);
+            --font-family: 'Inter', 'PingFang SC', 'Helvetica Neue', 'Microsoft YaHei', sans-serif;
+        }
+        
+        /* 基础样式 */
+        body {
+            font-family: var(--font-family);
+            background-color: var(--neutral-50);
+            color: var(--neutral-800);
+            margin: 0;
+            padding: 0;
+            min-height: 100vh;
+            display: flex;
+            align-items: center;
+            justify-content: center;
+            -webkit-font-smoothing: antialiased;
+            -moz-osx-font-smoothing: grayscale;
+        }
+        
+        h1, h2, h3, h4, h5, h6 {
+            font-weight: 600;
+            color: var(--neutral-900);
+        }
+        
+        .text-gradient {
+            background: linear-gradient(135deg, var(--secondary-color), var(--secondary-light));
+            -webkit-background-clip: text;
+            -webkit-text-fill-color: transparent;
+            background-clip: text;
+            color: transparent;
+        }
+        
+        /* 验证容器样式 */
+        .verification-container {
+            width: 100%;
+            max-width: 500px;
+            padding: 2.5rem;
+            background-color: white;
+            border-radius: var(--border-radius-xl);
+            box-shadow: var(--card-shadow-lg);
+            transition: var(--transition-smooth);
+            position: relative;
+            overflow: hidden;
+        }
+        
+        .verification-container:hover {
+            box-shadow: var(--card-shadow-hover);
+        }
+        
+        .verification-container::before {
+            content: '';
+            position: absolute;
+            top: 0;
+            left: 0;
+            width: 100%;
+            height: 4px;
+            background: linear-gradient(to right, var(--secondary-color), var(--secondary-light));
+            border-radius: 4px 4px 0 0;
+        }
+        
+        .verification-header {
+            text-align: center;
+            margin-bottom: 2rem;
+        }
+        
+        .verification-header h1 {
+            font-size: 1.75rem;
+            font-weight: 700;
+            margin-bottom: 0.5rem;
+            background: linear-gradient(45deg, var(--primary-color), var(--secondary-color));
+            -webkit-background-clip: text;
+            -webkit-text-fill-color: transparent;
+            letter-spacing: -0.01em;
+        }
+        
+        .verification-header p {
+            color: var(--neutral-600);
+            margin-bottom: 0;
+            font-size: 0.95rem;
+        }
+        
+        /* 加载状态 */
+        .loading-container {
+            text-align: center;
+            margin-bottom: 1.5rem;
+        }
+        
+        .spinner-border {
+            width: 3rem;
+            height: 3rem;
+            color: var(--secondary-color);
+            margin-bottom: 1.5rem;
+        }
+        
+        .loading-container h2 {
+            margin-bottom: 0.75rem;
+            font-size: 1.4rem;
+            color: var(--primary-color);
+        }
+        
+        .loading-container p {
+            color: var(--neutral-600);
+            font-size: 0.95rem;
+        }
+        
+        /* Agent信息卡片 */
+        .agent-info {
+            padding: 1.5rem;
+            background-color: var(--neutral-50);
+            border-radius: var(--border-radius-lg);
+            margin-bottom: 1.75rem;
+            border: 1px solid var(--neutral-200);
+            position: relative;
+            overflow: hidden;
+            transition: var(--transition-base);
+        }
+        
+        .agent-info:hover {
+            border-color: var(--secondary-color);
+            box-shadow: 0 4px 10px rgba(74, 108, 253, 0.1);
+        }
+        
+        .agent-info::before {
+            content: '';
+            position: absolute;
+            top: 0;
+            bottom: 0;
+            left: 0;
+            width: 4px;
+            background: linear-gradient(to bottom, var(--secondary-color), var(--secondary-light));
+            border-radius: 4px 0 0 4px;
+        }
+        
+        .agent-title {
+            font-size: 1.25rem;
+            font-weight: 600;
+            margin-bottom: 0.75rem;
+            color: var(--primary-color);
+            display: flex;
+            align-items: center;
+        }
+        
+        .agent-title i {
+            margin-right: 0.75rem;
+            font-size: 1.1rem;
+            color: var(--secondary-color);
+        }
+        
+        .agent-description {
+            color: var(--neutral-700);
+            margin-bottom: 1rem;
+            font-size: 0.95rem;
+            line-height: 1.5;
+        }
+        
+        .agent-meta {
+            display: flex;
+            flex-wrap: wrap;
+            gap: 1rem;
+            font-size: 0.9rem;
+            color: var(--neutral-600);
+        }
+        
+        .agent-meta-item {
+            display: flex;
+            align-items: center;
+        }
+        
+        .agent-meta-item i {
+            margin-right: 0.5rem;
+            font-size: 0.95rem;
+            color: var(--secondary-color);
+        }
+        
+        /* 验证操作按钮 */
+        .verification-actions {
+            margin-bottom: 1.5rem;
+        }
+        
+        .btn {
+            font-weight: 500;
+            padding: 0.75rem 1.25rem;
+            border-radius: var(--border-radius-lg);
+            transition: var(--transition-base);
+        }
+        
+        .btn-primary {
+            background: linear-gradient(to right, var(--secondary-color), var(--secondary-light));
+            border: none;
+            color: white;
+        }
+        
+        .btn-primary:hover, .btn-primary:focus {
+            background: linear-gradient(to right, var(--secondary-light), var(--secondary-color));
+            box-shadow: 0 4px 10px rgba(74, 108, 253, 0.3);
+            transform: translateY(-2px);
+        }
+        
+        .btn-outline-secondary {
+            color: var(--neutral-700);
+            border-color: var(--neutral-300);
+            background-color: white;
+        }
+        
+        .btn-outline-secondary:hover, .btn-outline-secondary:focus {
+            background-color: var(--neutral-100);
+            border-color: var(--neutral-400);
+            color: var(--neutral-900);
+        }
+        
+        /* 过期/错误状态 */
+        .expired-container {
+            text-align: center;
+            color: var(--danger-color);
+            margin-bottom: 1.5rem;
+        }
+        
+        .expired-container i {
+            font-size: 3rem;
+            margin-bottom: 1rem;
+            display: block;
+        }
+        
+        .expired-container h2 {
+            margin-bottom: 0.75rem;
+            font-size: 1.5rem;
+            color: var(--danger-color);
+        }
+        
+        .expired-container p {
+            color: var(--neutral-700);
+            margin-bottom: 1.5rem;
+        }
+        
+        /* 页脚样式 */
+        .verification-footer {
+            text-align: center;
+            margin-top: 2rem;
+            color: var(--neutral-500);
+            font-size: 0.85rem;
+        }
+        
+        .verification-footer a {
+            color: var(--secondary-color);
+            text-decoration: none;
+            transition: var(--transition-base);
+        }
+        
+        .verification-footer a:hover {
+            text-decoration: underline;
+            color: var(--secondary-light);
+        }
+        
+        /* 动画效果 */
+        @keyframes fadeIn {
+            from { opacity: 0; transform: translateY(10px); }
+            to { opacity: 1; transform: translateY(0); }
+        }
+        
+        .fade-in {
+            opacity: 0;
+            animation: fadeIn 0.4s ease-out forwards;
+        }
+        
+        /* 响应式样式 */
+        @media (max-width: 576px) {
+            .verification-container {
+                padding: 1.5rem;
+                margin: 0 1rem;
+            }
+            
+            .verification-header h1 {
+                font-size: 1.5rem;
+            }
+            
+            .agent-title {
+                font-size: 1.15rem;
+            }
+            
+            .agent-meta {
+                flex-direction: column;
+                gap: 0.5rem;
+            }
+        }
+    </style>
+</head>
+<body>
+    <div class="verification-container fade-in" id="main-container">
+        <div class="loading-container" id="loading-container">
+            <div class="spinner-border" role="status">
+                <span class="visually-hidden">验证中...</span>
+            </div>
+            <h2>正在验证访问权限</h2>
+            <p class="text-muted">请稍候，我们正在验证您的访问令牌...</p>
+        </div>
+        
+        <div class="verification-header" id="verification-header" style="display: none;">
+            <h1>访问 AI 助手</h1>
+            <p>您正在使用访问令牌访问以下 AI 助手</p>
+        </div>
+        
+        <div class="agent-info" id="agent-info" style="display: none;">
+            <!-- 将由JavaScript填充 -->
+        </div>
+        
+        <div class="verification-actions" id="verification-actions" style="display: none;">
+            <div class="row g-3">
+                <div class="col-12">
+                    <button class="btn btn-primary w-100" id="access-btn">
+                        <i class="bi bi-robot me-2"></i>开始对话
+                    </button>
+                </div>
+                <div class="col-12">
+                    <button class="btn btn-outline-secondary w-100" id="back-btn">
+                        <i class="bi bi-arrow-left me-2"></i>返回
+                    </button>
+                </div>
+            </div>
+        </div>
+        
+        <div class="expired-container" id="expired-container" style="display: none;">
+            <i class="bi bi-exclamation-triangle"></i>
+            <h2>访问令牌无效</h2>
+            <p>您使用的访问令牌无效或已过期，请联系教师获取新的访问令牌。</p>
+            <button class="btn btn-outline-secondary mt-3" id="back-btn-expired">
+                <i class="bi bi-arrow-left me-2"></i>返回
+            </button>
+        </div>
+        
+        <div class="verification-footer" id="verification-footer" style="display: none;">
+            <p>教育 AI 助手平台 | <a href="/login.html">登录</a></p>
+        </div>
+    </div>
+    
+    <script>
+        // 解析URL参数
+        function getUrlParams() {
+            const queryString = window.location.search;
+            const urlParams = new URLSearchParams(queryString);
+            const pathParts = window.location.pathname.split('/');
+            
+            return {
+                agentId: pathParts[pathParts.length - 1] || '',
+                token: urlParams.get('token') || ''
+            };
+        }
+        
+        // 页面加载函数
+        document.addEventListener('DOMContentLoaded', async function() {
+            const { agentId, token } = getUrlParams();
+            
+            if (!token) {
+                showExpiredState('未提供访问令牌');
+                return;
+            }
+            
+            try {
+                // 验证令牌
+                const response = await fetch('/api/verify_token', {
+                    method: 'POST',
+                    headers: {
+                        'Content-Type': 'application/json'
+                    },
+                    body: JSON.stringify({ 
+                        token: token,
+                        agent_id: agentId
+                    })
+                });
+                
+                const result = await response.json();
+                
+                if (result.success) {
+                    // 显示Agent信息
+                    showAgentInfo(result.agent);
+                } else {
+                    showExpiredState(result.message);
+                }
+            } catch (error) {
+                console.error('验证出错:', error);
+                showExpiredState('验证过程中出错');
+            }
+        });
+        
+        // 显示Agent信息
+        function showAgentInfo(agent) {
+            // 隐藏加载区域
+            document.getElementById('loading-container').style.display = 'none';
+            
+            // 显示��证内容
+            document.getElementById('verification-header').style.display = 'block';
+            document.getElementById('agent-info').style.display = 'block';
+            document.getElementById('verification-actions').style.display = 'block';
+            document.getElementById('verification-footer').style.display = 'block';
+            
+            // 填充Agent信息
+            const agentInfoElement = document.getElementById('agent-info');
+            
+            // 构建主题和教师信息
+            let metaHtml = '<div class="agent-meta">';
+            
+            if (agent.subject) {
+                metaHtml += `
+                    <div class="agent-meta-item">
+                        <i class="bi bi-book"></i>
+                        ${agent.subject}
+                    </div>
+                `;
+            }
+            
+            if (agent.instructor) {
+                metaHtml += `
+                    <div class="agent-meta-item">
+                        <i class="bi bi-person"></i>
+                        ${agent.instructor}
+                    </div>
+                `;
+            }
+            
+            metaHtml += '</div>';
+            
+            agentInfoElement.innerHTML = `
+                <div class="agent-title">
+                    <i class="bi bi-robot"></i>${agent.name}
+                </div>
+                <div class="agent-description">
+                    ${agent.description || '暂无描述'}
+                </div>
+                ${metaHtml}
+            `;
+            
+            // 设置按钮链接
+            document.getElementById('access-btn').addEventListener('click', function() {
+                const { token } = getUrlParams();
+                window.location.href = `/student/${agent.id}?token=${token}`;
+            });
+            
+            document.getElementById('back-btn').addEventListener('click', function() {
+                // 检查是否从学生门户进入
+                const hasHistory = document.referrer.includes('student_portal.html');
+                if (hasHistory) {
+                    window.history.back();
+                } else {
+                    window.location.href = '/student_portal.html';
+                }
+            });
+        }
+        
+        // 显示过期状态
+        function showExpiredState(message) {
+            // 隐藏加载区域
+            document.getElementById('loading-container').style.display = 'none';
+            
+            // 显示过期内容
+            document.getElementById('expired-container').style.display = 'block';
+            document.getElementById('verification-footer').style.display = 'block';
+            
+            // 更新过期消息
+            const expiredContainer = document.getElementById('expired-container');
+            expiredContainer.querySelector('p').textContent = message || '您使用的访问令牌无效或已过期，请联系教师获取新的访问令牌。';
+            
+            // 设置返回按钮
+            document.getElementById('back-btn-expired').addEventListener('click', function() {
+                // 检查是否从学生门户进入
+                const hasHistory = document.referrer.includes('student_portal.html');
+                if (hasHistory) {
+                    window.history.back();
+                } else {
+                    window.location.href = '/login.html';
+                }
+            });
+        }
+    </script>
+</body>
+</html>