Upload 4 files

app (1).py

@@ -0,0 +1,268 @@
+import os
+import json
+import requests
+import smtplib
+from email.mime.text import MIMEText
+from email.mime.multipart import MIMEMultipart
+from flask import Flask, request, jsonify, send_from_directory
+from openai import OpenAI
+from duckduckgo_search import DDGS
+from functions import FUNCTIONS_GROUP_1, FUNCTIONS_GROUP_2, get_function_descriptions
+
+app = Flask(__name__)
+API_KEY = os.getenv("OPENAI_API_KEY")
+BASE_URL = os.getenv("OPENAI_BASE_URL")
+emailkey = os.getenv("EMAIL_KEY")
+client = OpenAI(api_key=API_KEY, base_url=BASE_URL)
+
+def search_duckduckgo(keywords):
+    search_term = " ".join(keywords)
+    with DDGS() as ddgs:
+        return list(ddgs.text(keywords=search_term, region="cn-zh", safesearch="on", max_results=5))
+
+def search_papers(query):
+    url = f"https://api.crossref.org/works?query={query}"
+    response = requests.get(url)
+    if response.status_code == 200:
+        data = response.json()
+        papers = data['message']['items']
+        processed_papers = []
+        for paper in papers:
+            processed_paper = {
+                "标题": paper.get('title', [''])[0],
+                "作者": ", ".join([f"{author.get('given', '')} {author.get('family', '')}" for author in paper.get('author', [])]),
+                "DOI": paper.get('DOI', ''),
+                "ISBN": ", ".join(paper.get('ISBN', [])),
+                "摘要": paper.get('abstract', '').replace('<p>', '').replace('</p>', '').replace('<italic>', '').replace('</italic>', '')
+            }
+            processed_papers.append(processed_paper)
+        return processed_papers
+    else:
+        return []
+
+def send_email(to, subject, content):
+    try:
+        with smtplib.SMTP('106.15.184.28', 8025) as smtp:
+            smtp.login("jwt", emailkey)
+            message = MIMEMultipart()
+            message['From'] = "Me <[email protected]>"
+            message['To'] = to
+            message['Subject'] = subject
+            message.attach(MIMEText(content, 'html'))
+            smtp.sendmail("[email protected]", to, message.as_string())
+        return True
+    except Exception as e:
+        print(f"发送邮件时出错: {str(e)}")
+        return False
+
+def get_openai_response(messages, model="gpt-4o-mini", functions=None, function_call=None):
+    try:
+        response = client.chat.completions.create(
+            model=model,
+            messages=messages,
+            functions=functions,
+            function_call=function_call
+        )
+        return response.choices[0].message
+    except Exception as e:
+        print(f"调用OpenAI API时出错: {str(e)}")
+        return None
+
+def process_function_call(function_name, function_args):
+    if function_name == "search_duckduckgo":
+        keywords = function_args.get('keywords', [])
+        if not keywords:
+            return "搜索关键词为空，无法执行搜索。"
+        return json.dumps(search_duckduckgo(keywords), ensure_ascii=False)
+    elif function_name == "search_papers":
+        query = function_args.get('query', '')
+        if not query:
+            return "搜索查询为空，无法执行论文搜索。"
+        return json.dumps(search_papers(query), ensure_ascii=False)
+    elif function_name == "send_email":
+        to = function_args.get('to', '')
+        subject = function_args.get('subject', '')
+        content = function_args.get('content', '')
+        if not to or not subject or not content:
+            return "邮件信息不完整，无法发送邮件。"
+        success = send_email(to, subject, content)
+        return json.dumps({
+            "success": success,
+            "message": "邮件发送成功" if success else "邮件发送失败",
+            "to": to,
+            "subject": subject,
+            "content": content,
+            "is_email": True  # 添加这个标记来识别邮件发送功能
+        }, ensure_ascii=False)
+    else:
+        return "未知的函数调用。"
+
+@app.route('/')
+def index():
+    return send_from_directory('.', 'index.html')
+
+# @app.route('/chat', methods=['POST'])
+# def chat():
+#     data = request.json
+#     question = data['question']
+#     history = data.get('history', [])
+#     messages = history + [{"role": "user", "content": question}]
+    
+#     status_log = []
+    
+#     # 次级模型1: 处理搜索相关函数
+#     status_log.append("次级模型1：正在判断是否需要选调第一组函数")
+#     sub_model_1_response = get_openai_response(messages, model="gpt-4o-mini", functions=FUNCTIONS_GROUP_1, function_call="auto")
+    
+#     # 次级模型2: 处理邮件发送相关函数
+#     status_log.append("次级模型2：正在判断是否需要选调第二组函数")
+#     sub_model_2_response = get_openai_response(messages, model="gpt-4o-mini", functions=FUNCTIONS_GROUP_2, function_call="auto")
+    
+#     function_call_1 = sub_model_1_response.function_call if sub_model_1_response and sub_model_1_response.function_call else None
+#     function_call_2 = sub_model_2_response.function_call if sub_model_2_response and sub_model_2_response.function_call else None
+    
+#     final_function_call = None
+#     response = None
+#     if function_call_1 and function_call_2:
+#         # 裁决模型: 决定使用哪个函数调用
+#         status_log.append("裁决模型：正在决定使用哪个函数调用")
+#         arbitration_messages = messages + [
+#             {"role": "system", "content": "两个次级模型都建议使用函数。请决定使用哪个函数更合适。"},
+#             {"role": "assistant", "content": f"次级模型1建议使用函数：{function_call_1.name}"},
+#             {"role": "assistant", "content": f"次级模型2建议使用函数：{function_call_2.name}"}
+#         ]
+#         arbitration_response = get_openai_response(arbitration_messages, model="gpt-4o-mini")
+#         if "模型1" in arbitration_response.content or function_call_1.name in arbitration_response.content:
+#             final_function_call = function_call_1
+#             status_log.append(f"裁决模型：决定使用函数 {function_call_1.name}")
+#         else:
+#             final_function_call = function_call_2
+#             status_log.append(f"裁决模型：决定使用函数 {function_call_2.name}")
+#     elif function_call_1:
+#         final_function_call = function_call_1
+#         status_log.append(f"次级模型1：决定使用函数 {function_call_1.name}")
+#     elif function_call_2:
+#         final_function_call = function_call_2
+    #      status_log.append(f"次级模型2：决定使用函数 {function_call_2.name}")
+    #  else:
+    #      status_log.append("次级模型：判断不需要进行搜索或发送邮件")
+    
+    # if final_function_call:
+    #     function_name = final_function_call.name
+    #     function_args = json.loads(final_function_call.arguments)
+    #     status_log.append(f"正在执行函数 {function_name}")
+    #     result = process_function_call(function_name, function_args)
+    #     status_log.append(f"函数 {function_name} 执行完成")
+        
+    #     # 检查是否为邮件发送功能
+    #     result_dict = json.loads(result)
+    #     if result_dict.get("is_email", False):
+    #         response = f"邮件{'已成功' if result_dict['success'] else '未能成功'}发送到 {result_dict['to']}。\n\n主题：{result_dict['subject']}\n\n内容：\n{result_dict['content']}"
+    #     else:
+    #         messages.append({
+    #             "role": "function",
+    #             "name": function_name,
+    #             "content": result
+    #         })
+    
+    # # 只有在没有邮件发送结果时才调用主模型
+    # if not response:
+    #     status_log.append("主模型：正在生成回答")
+    #     final_response = get_openai_response(messages, model="gpt-4o-mini")
+    #     response = final_response.content if final_response else "Error occurred"
+    #     status_log.append("主模型：回答生成完成")
+    
+    # return jsonify({
+    #     "response": response,
+    #     "status_log": status_log
+    # })
+
+@app.route('/chat', methods=['POST'])
+def chat():
+    data = request.json
+    question = data['question']
+    history = data.get('history', [])
+    messages = history + [{"role": "user", "content": question}]
+    
+    status_log = []
+    
+    # 次级模型1: 处理搜索相关函数
+    status_log.append("次级模型1：正在判断是否需要选调第一组函数")
+    sub_model_1_response = get_openai_response(messages, model="gpt-4o-mini", functions=FUNCTIONS_GROUP_1, function_call="auto")
+    
+    # 次级模型2: 处理邮件发送相关函数
+    status_log.append("次级模型2：正在判断是否需要选调第二组函数")
+    sub_model_2_response = get_openai_response(messages, model="gpt-4o-mini", functions=FUNCTIONS_GROUP_2, function_call="auto")
+    
+    function_call_1 = sub_model_1_response.function_call if sub_model_1_response and sub_model_1_response.function_call else None
+    function_call_2 = sub_model_2_response.function_call if sub_model_2_response and sub_model_2_response.function_call else None
+    
+    if not function_call_1:
+        status_log.append("次级模型1：判断不需要选调第一组函数")
+    if not function_call_2:
+        status_log.append("次级模型2：判断不需要选调第二组函数")
+    
+    final_function_call = None
+    response = None
+    if function_call_1 and function_call_2:
+        # 裁决模型: 决定使用哪个函数调用
+        status_log.append("裁决模型：正在决定使用哪个函数调用")
+        arbitration_messages = messages + [
+            {"role": "system", "content": "两个次级模型都建议使用函数。请决定使用哪个函数更合适。"},
+            {"role": "assistant", "content": f"次级模型1建议使用函数：{function_call_1.name}"},
+            {"role": "assistant", "content": f"次级模型2建议使用函数：{function_call_2.name}"}
+        ]
+        arbitration_response = get_openai_response(arbitration_messages, model="gpt-4o-mini")
+        if "模型1" in arbitration_response.content or function_call_1.name in arbitration_response.content:
+            final_function_call = function_call_1
+            status_log.append(f"裁决模型：决定使用函数 {function_call_1.name}")
+        else:
+            final_function_call = function_call_2
+            status_log.append(f"裁决模型：决定使用函数 {function_call_2.name}")
+    elif function_call_1:
+        final_function_call = function_call_1
+        status_log.append(f"次级模型1：决定使用函数 {function_call_1.name}")
+    elif function_call_2:
+        final_function_call = function_call_2
+        status_log.append(f"次级模型2：决定使用函数 {function_call_2.name}")
+    else:
+        status_log.append("所有次级模型：判断不需要进行任何函数调用")
+    
+    if final_function_call:
+        function_name = final_function_call.name
+        function_args = json.loads(final_function_call.arguments)
+        status_log.append(f"正在执行函数 {function_name}")
+        result = process_function_call(function_name, function_args)
+        status_log.append(f"函数 {function_name} 执行完成")
+        
+        # 检查是否为邮件发送功能
+        result_dict = json.loads(result)
+        if result_dict.get("is_email", False):
+            response = f"邮件{'已成功' if result_dict['success'] else '未能成功'}发送到 {result_dict['to']}。\n\n主题：{result_dict['subject']}\n\n内容：\n{result_dict['content']}"
+        else:
+            messages.append({
+                "role": "function",
+                "name": function_name,
+                "content": result
+            })
+    
+    # 只有在没有邮件发送结果时才调用主模型
+    if not response:
+        status_log.append("主模型：正在生成回答")
+        final_response = get_openai_response(messages, model="gpt-4o-mini")
+        response = final_response.content if final_response else "Error occurred"
+        status_log.append("主模型：回答生成完成")
+    
+    return jsonify({
+        "response": response,
+        "status_log": status_log
+    })
+
+@app.route('/settings', methods=['POST'])
+def update_settings():
+    data = request.json
+    max_history = data.get('max_history', 10)
+    return jsonify({"status": "success", "max_history": max_history})
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=7860, debug=True)

function_implementations.py

@@ -0,0 +1,76 @@
+import json
+import requests
+import smtplib
+from email.mime.text import MIMEText
+from email.mime.multipart import MIMEMultipart
+from duckduckgo_search import DDGS
+import os
+
+emailkey = os.getenv("EMAIL_KEY")
+
+def search_duckduckgo(keywords):
+    search_term = " ".join(keywords)
+    with DDGS() as ddgs:
+        return list(ddgs.text(keywords=search_term, region="cn-zh", safesearch="on", max_results=5))
+
+def search_papers(query):
+    url = f"https://api.crossref.org/works?query={query}"
+    response = requests.get(url)
+    if response.status_code == 200:
+        data = response.json()
+        papers = data['message']['items']
+        processed_papers = []
+        for paper in papers:
+            processed_paper = {
+                "标题": paper.get('title', [''])[0],
+                "作者": ", ".join([f"{author.get('given', '')} {author.get('family', '')}" for author in paper.get('author', [])]),
+                "DOI": paper.get('DOI', ''),
+                "ISBN": ", ".join(paper.get('ISBN', [])),
+                "摘要": paper.get('abstract', '').replace('<p>', '').replace('</p>', '').replace('<italic>', '').replace('</italic>', '')
+            }
+            processed_papers.append(processed_paper)
+        return processed_papers
+    else:
+        return []
+
+def send_email(to, subject, content):
+    try:
+        with smtplib.SMTP('106.15.184.28', 8025) as smtp:
+            smtp.login("jwt", emailkey)
+            message = MIMEMultipart()
+            message['From'] = "Me <[email protected]>"
+            message['To'] = to
+            message['Subject'] = subject
+            message.attach(MIMEText(content, 'html'))
+            smtp.sendmail("[email protected]", to, message.as_string())
+        return True
+    except Exception as e:
+        print(f"发送邮件时出错: {str(e)}")
+        return False
+
+def process_function_call(function_name, function_args):
+    if function_name == "search_duckduckgo":
+        keywords = function_args.get('keywords', [])
+        if not keywords:
+            return None
+        return search_duckduckgo(keywords)
+    elif function_name == "search_papers":
+        query = function_args.get('query', '')
+        if not query:
+            return None
+        return search_papers(query)
+    elif function_name == "send_email":
+        to = function_args.get('to', '')
+        subject = function_args.get('subject', '')
+        content = function_args.get('content', '')
+        if not to or not subject or not content:
+            return None
+        success = send_email(to, subject, content)
+        return {
+            "success": success,
+            "to": to,
+            "subject": subject,
+            "content": content
+        }
+    else:
+        return None

functions.py

@@ -0,0 +1,61 @@
+import json
+
+FUNCTIONS = [
+    {
+        "name": "search_duckduckgo",
+        "description": "使用DuckDuckGo搜索引擎查询信息。可以搜索最新新闻、文章、博客等内容。",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "keywords": {
+                    "type": "array",
+                    "items": {"type": "string"},
+                    "description": "搜索的关键词列表。例如：['Python', '机器学习', '最新进展']。"
+                }
+            },
+            "required": ["keywords"]
+        }
+    },
+    {
+        "name": "search_papers",
+        "description": "使用Crossref API搜索学术论文。",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "query": {
+                    "type": "string",
+                    "description": "搜索查询字符串。例如：'climate change'。"
+                }
+            },
+            "required": ["query"]
+        }
+    },
+    {
+        "name": "send_email",
+        "description": "发送电子邮件。",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "to": {
+                    "type": "string",
+                    "description": "收件人邮箱地址"
+                },
+                "subject": {
+                    "type": "string",
+                    "description": "邮件主题"
+                },
+                "content": {
+                    "type": "string",
+                    "description": "邮件内容"
+                }
+            },
+            "required": ["to", "subject", "content"]
+        }
+    }
+]
+
+FUNCTIONS_GROUP_1 = [FUNCTIONS[0], FUNCTIONS[1]]  # search_duckduckgo, search_papers
+FUNCTIONS_GROUP_2 = [FUNCTIONS[2]]  # send_email
+
+def get_function_descriptions(functions):
+    return [{"name": f["name"], "description": f["description"]} for f in functions]

index (1).html

@@ -0,0 +1,539 @@
+<!DOCTYPE html>
+<html lang="zh-CN">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>AI 对话系统</title>
+    <link href="https://cdnjs.cloudflare.com/ajax/libs/tailwindcss/2.2.19/tailwind.min.css" rel="stylesheet">
+    <link href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/5.15.4/css/all.min.css" rel="stylesheet">
+    <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/highlight.js/11.7.0/styles/atom-one-dark.min.css">
+   <style>
+    :root {
+        --primary-color: #3498db;
+        --secondary-color: #2ecc71;
+        --background-color: #f5f5f5;
+        --chat-background: #ffffff;
+        --text-color: #333333;
+        --chat-bubble-user: #e8f5fe;
+        --chat-bubble-ai: #f0f0f0;
+        --sidebar-background: #ffffff;
+        --input-background: #ffffff;
+        --send-button-color: #4CAF50;
+        --console-background: #f8f9fa;
+        --console-text: #495057;
+        --fold-background: #f8f9fa;
+        --fold-text: #495057;
+        --border-color: transparent;
+        --search-result-background: #f8f9fa;
+    }
+    .dark-mode {
+        --primary-color: #3498db;
+        --secondary-color: #2ecc71;
+        --background-color: #1e2124;
+        --chat-background: #36393f;
+        --text-color: #dcddde;
+        --chat-bubble-user: #4e5d94;
+        --chat-bubble-ai: #40444b;
+        --sidebar-background: #2f3136;
+        --input-background: #40444b;
+        --send-button-color: #7289da;
+        --console-background: #2f3136;
+        --console-text: #dcddde;
+        --fold-background: #2f3136;
+        --fold-text: #dcddde;
+        --border-color: #ffffff;
+        --search-result-background: #2f3136;
+    }
+    body {
+        background-color: var(--background-color);
+        color: var(--text-color);
+        transition: all 0.3s ease;
+    }
+    .chat-container {
+        background-color: var(--chat-background);
+        transition: all 0.3s ease;
+        border-radius: 8px;
+        box-shadow: 0 2px 10px rgba(0,0,0,0.1);
+    }
+    .chat-bubble {
+        max-width: 80%;
+        padding: 12px;
+        border-radius: 12px;
+        margin-bottom: 12px;
+        box-shadow: 0 1px 2px rgba(0,0,0,0.1);
+        transition: all 0.3s ease;
+    }
+    .user-bubble {
+        background-color: var(--chat-bubble-user);
+        color: var(--text-color);
+        margin-left: auto;
+        border-bottom-right-radius: 4px;
+    }
+    .ai-bubble {
+        background-color: var(--chat-bubble-ai);
+        color: var(--text-color);
+        margin-right: auto;
+        border-bottom-left-radius: 4px;
+    }
+    .dark-mode .chat-bubble {
+        border: 1px solid var(--border-color);
+    }
+    .status-indicator {
+        width: 12px;
+        height: 12px;
+        border-radius: 50%;
+        display: inline-block;
+        margin-right: 8px;
+        transition: all 0.3s ease;
+    }
+    .status-active {
+        background-color: var(--secondary-color);
+    }
+    .status-inactive {
+        background-color: #95a5a6;
+    }
+    .sidebar {
+        background-color: var(--sidebar-background);
+        transition: all 0.3s ease;
+        border-radius: 8px;
+        box-shadow: 0 2px 10px rgba(0,0,0,0.1);
+    }
+    #user-input {
+        background-color: var(--input-background);
+        color: var(--text-color);
+        transition: all 0.3s ease;
+        border-radius: 8px;
+        padding-right: 40px;
+    }
+    #send-btn {
+        position: absolute;
+        right: 10px;
+        top: 50%;
+        transform: translateY(-50%);
+        background: none;
+        border: none;
+        color: var(--send-button-color);
+        font-size: 20px;
+        cursor: pointer;
+        transition: all 0.3s ease;
+    }
+    #send-btn:hover {
+        opacity: 0.8;
+    }
+    .console {
+        background-color: var(--console-background);
+        color: var(--console-text);
+        transition: all 0.3s ease;
+        border-radius: 8px;
+        box-shadow: 0 2px 10px rgba(0,0,0,0.1);
+    }
+    #status-log {
+        background-color: var(--console-background);
+        color: var(--console-text);
+        transition: all 0.3s ease;
+        border: 1px solid var(--border-color);
+    }
+    /* 滚动条样式 */
+    ::-webkit-scrollbar {
+        width: 8px;
+    }
+    ::-webkit-scrollbar-track {
+        background: var(--chat-background);
+    }
+    ::-webkit-scrollbar-thumb {
+        background: var(--primary-color);
+        border-radius: 4px;
+    }
+    ::-webkit-scrollbar-thumb:hover {
+        background: var(--secondary-color);
+    }
+    /* 代码块样式 */
+    .hljs {
+        background: var(--fold-background) !important;
+        color: var(--fold-text) !important;
+    }
+
+     .search-results {
+        background-color: var(--search-result-background);
+        color: var(--text-color);
+        border: 1px solid var(--border-color);
+    }
+    .search-results summary {
+        cursor: pointer;
+        font-weight: bold;
+    }
+    .search-results ul {
+        list-style-type: none;
+        padding-left: 0;
+    }
+    .search-results li {
+        background-color: var(--chat-bubble-ai);
+        border: 1px solid var(--border-color);
+    }
+    /* 控制台消息条目样式 */
+    #status-log div {
+        background-color: var(--console-background);
+        color: var(--console-text);
+        border: 1px solid var(--border-color);
+    }
+    @media (max-width: 768px) {
+        .sidebar {
+            position: fixed;
+            right: -300px;
+            top: 0;
+            bottom: 0;
+            width: 300px;
+            z-index: 1000;
+            transition: right 0.3s ease-in-out;
+        }
+        .sidebar.open {
+            right: 0;
+        }
+        .sidebar-overlay {
+            display: none;
+            position: fixed;
+            top: 0;
+            left: 0;
+            right: 0;
+            bottom: 0;
+            background-color: rgba(0, 0, 0, 0.5);
+            z-index: 999;
+        }
+        .sidebar-overlay.show {
+            display: block;
+        }
+    }
+</style>
+<body class="h-screen flex flex-col md:flex-row p-4">
+    <!-- 主聊天窗口 -->
+    <div class="chat-container flex-grow md:w-3/4 p-4 flex flex-col h-screen mr-4">
+        <div id="chat-window" class="flex-grow overflow-y-auto mb-4 p-4 rounded-lg shadow-inner"></div>
+        <div class="relative">
+            <input id="user-input" type="text" class="w-full border p-2" placeholder="输入您的问题...">
+            <button id="send-btn" class="text-2xl">
+                <i class="fas fa-paper-plane"></i>
+            </button>
+        </div>
+    </div>
+
+    <!-- 右侧控制台 -->
+    <div class="sidebar console md:w-1/4 p-4 h-screen overflow-y-auto">
+        <h2 class="text-xl font-bold mb-4">运行状态</h2>
+        <div class="mb-4">
+            <div class="model-status">
+                <span class="status-indicator" id="main-model-status"></span>
+                <span class="model-name">主模型</span>
+                <span class="model-action" id="main-model-action"></span>
+            </div>
+            <div class="model-status">
+                <span class="status-indicator" id="sub-model-1-status"></span>
+                <span class="model-name">次级模型1</span>
+                <span class="model-action" id="sub-model-1-action"></span>
+            </div>
+            <div class="model-status">
+                <span class="status-indicator" id="sub-model-2-status"></span>
+                <span class="model-name">次级模型2</span>
+                <span class="model-action" id="sub-model-2-action"></span>
+            </div>
+            <div class="model-status">
+                <span class="status-indicator" id="arbitration-model-status"></span>
+                <span class="model-name">裁决模型</span>
+                <span class="model-action" id="arbitration-model-action"></span>
+            </div>
+            <div class="model-status">
+                <span class="status-indicator" id="search-status"></span>
+                <span class="model-name">搜索</span>
+                <span class="model-action" id="search-action"></span>
+            </div>
+            <div class="model-status">
+                <span class="status-indicator" id="email-status"></span>
+                <span class="model-name">邮件发送</span>
+                <span class="model-action" id="email-action"></span>
+            </div>
+        </div>
+        <div id="status-log" class="p-2 rounded h-64 overflow-y-auto mb-4"></div>
+        <button id="settings-btn" class="w-full bg-gray-300 dark:bg-gray-600 text-gray-800 dark:text-white px-4 py-2 rounded hover:bg-gray-400 dark:hover:bg-gray-500 transition duration-200">
+            <i class="fas fa-cog"></i> 设置
+        </button>
+    </div>
+
+    <!-- 移动端侧边栏切换按钮 -->
+    <button id="sidebar-toggle" class="md:hidden fixed top-4 right-4 bg-gray-300 dark:bg-gray-600 text-gray-800 dark:text-white p-2 rounded-full shadow-lg hover:bg-gray-400 dark:hover:bg-gray-500 transition duration-200 z-50">
+        <i class="fas fa-bars"></i>
+    </button>
+
+    <!-- 移动端侧边栏遮罩 -->
+    <div id="sidebar-overlay" class="sidebar-overlay"></div>
+
+    <!-- 暗色模式切换按钮 -->
+    <button id="dark-mode-toggle" class="fixed top-4 left-4 bg-gray-300 dark:bg-gray-600 text-gray-800 dark:text-white p-2 rounded-full shadow-lg hover:bg-gray-400 dark:hover:bg-gray-500 transition duration-200">
+        <i class="fas fa-moon"></i>
+    </button>
+
+    <!-- 设置对话框 -->
+    <div id="settings-modal" class="fixed inset-0 bg-gray-600 bg-opacity-50 hidden flex items-center justify-center z-50">
+        <div class="bg-white dark:bg-gray-800 p-6 rounded-lg shadow-xl">
+            <h2 class="text-xl font-bold mb-4">设置</h2>
+            <div class="mb-4">
+                <label for="max-history" class="block mb-2">保留对话轮数:</label>
+                <input type="number" id="max-history" class="border rounded p-2 w-full dark:bg-gray-700 dark:text-white" min="1" max="50" value="10">
+            </div>
+            <div class="flex justify-end">
+                <button id="save-settings" class="bg-blue-500 text-white px-4 py-2 rounded mr-2 hover:bg-blue-600 transition duration-200">保存</button>
+                <button id="cancel-settings" class="bg-gray-300 dark:bg-gray-600 text-gray-800 dark:text-white px-4 py-2 rounded hover:bg-gray-400 dark:hover:bg-gray-500 transition duration-200">取消</button>
+            </div>
+        </div>
+    </div>
+
+    <script src="https://cdnjs.cloudflare.com/ajax/libs/marked/4.0.2/marked.min.js"></script>
+    <script src="https://cdnjs.cloudflare.com/ajax/libs/highlight.js/11.7.0/highlight.min.js"></script>
+    <script>
+        const chatWindow = document.getElementById('chat-window');
+        const userInput = document.getElementById('user-input');
+        const sendBtn = document.getElementById('send-btn');
+        const statusLog = document.getElementById('status-log');
+        const settingsBtn = document.getElementById('settings-btn');
+        const settingsModal = document.getElementById('settings-modal');
+        const saveSettingsBtn = document.getElementById('save-settings');
+        const cancelSettingsBtn = document.getElementById('cancel-settings');
+        const maxHistoryInput = document.getElementById('max-history');
+        const darkModeToggle = document.getElementById('dark-mode-toggle');
+        const sidebarToggle = document.getElementById('sidebar-toggle');
+        const sidebar = document.querySelector('.sidebar');
+        const sidebarOverlay = document.getElementById('sidebar-overlay');
+
+        let conversationHistory = [];
+        let maxHistory = 10;
+
+        function addMessage(content, isUser) {
+            const bubble = document.createElement('div');
+            bubble.className = `chat-bubble ${isUser ? 'user-bubble' : 'ai-bubble'}`;
+            bubble.innerHTML = isUser ? content : marked.parse(content);
+            chatWindow.appendChild(bubble);
+            chatWindow.scrollTop = chatWindow.scrollHeight;
+            
+            bubble.querySelectorAll('pre code').forEach((block) => {
+                hljs.highlightBlock(block);
+            });
+            
+            conversationHistory.push({
+                role: isUser ? 'user' : 'assistant',
+                content: content
+            });
+            
+            if (conversationHistory.length > maxHistory * 2) {
+                conversationHistory = conversationHistory.slice(-maxHistory * 2);
+            }
+        }
+
+        function updateStatus(main, sub1, sub2, arbitration, search, email) {
+            updateModelStatus('main-model', main);
+            updateModelStatus('sub-model-1', sub1);
+            updateModelStatus('sub-model-2', sub2);
+            updateModelStatus('arbitration-model', arbitration);
+            updateModelStatus('search', search);
+            updateModelStatus('email', email);
+        }
+
+        function updateModelStatus(modelId, isActive, action = '') {
+            const statusElement = document.getElementById(`${modelId}-status`);
+            const actionElement = document.getElementById(`${modelId}-action`);
+            statusElement.className = `status-indicator ${isActive ? 'status-active' : 'status-inactive'}`;
+            actionElement.textContent = action;
+        }
+
+        function addStatusLog(message) {
+            const logEntry = document.createElement('div');
+            logEntry.className = 'mb-2 p-2 bg-white dark:bg-gray-700 rounded shadow';
+            logEntry.textContent = message;
+            statusLog.appendChild(logEntry);
+            statusLog.scrollTop = statusLog.scrollHeight;
+        }
+
+        function displaySearchResults(results, type) {
+            const searchResults = document.createElement('details');
+            searchResults.className = 'search-results mb-2 p-2 rounded';
+            const summary = document.createElement('summary');
+            summary.textContent = type === 'papers' ? '论文搜索结果' : '搜索结果';
+            searchResults.appendChild(summary);
+            const resultsList = document.createElement('ul');
+            resultsList.className = 'mt-2';
+            results.forEach(result => {
+                const li = document.createElement('li');
+                li.className = 'mb-2 p-2 rounded';
+                if (type === 'papers') {
+                    li.innerHTML = `
+                        <strong>标题:</strong> ${result.标题}<br>
+                        <strong>作者:</strong> ${result.作者}<br>
+                        <strong>DOI:</strong> ${result.DOI}<br>
+                        <strong>ISBN:</strong> ${result.ISBN}<br>
+                        <strong>摘要:</strong> ${result.摘要}
+                    `;
+                } else {
+                    li.innerHTML = `<strong>${result.title}</strong><br>${result.body}`;
+                }
+                resultsList.appendChild(li);
+            });
+            searchResults.appendChild(resultsList);
+            chatWindow.appendChild(searchResults);
+        }
+
+        function updateStatusFromLog(log) {
+            if (log.includes("主模型：")) {
+                updateModelStatus('main-model', true, log.split("主模型：")[1]);
+            } else if (log.includes("次级模型1：")) {
+                updateModelStatus('sub-model-1', true, log.split("次级模型1：")[1]);
+            } else if (log.includes("次级模型2：")) {
+                updateModelStatus('sub-model-2', true, log.split("次级模型2：")[1]);
+            } else if (log.includes("裁决模型：")) {
+                updateModelStatus('arbitration-model', true, log.split("裁决模型：")[1]);
+            } else if (log.includes("搜索：")) {
+                updateModelStatus('search', true, log.split("搜索：")[1]);
+            } else if (log.includes("邮件：")) {
+                updateModelStatus('email', true, log.split("邮件：")[1]);
+            }
+        }
+
+        async function sendMessage() {
+            const question = userInput.value.trim();
+            if (!question) return;
+            addMessage(question, true);
+            userInput.value = '';
+            updateStatus(false, false, false, false, false, false);
+            statusLog.innerHTML = '';
+            addStatusLog('开始处理问题...');
+            try {
+                const response = await fetch('/chat', {
+                    method: 'POST',
+                    headers: {
+                        'Content-Type': 'application/json'
+                    },
+                    body: JSON.stringify({ 
+                        question,
+                        history: conversationHistory.slice(-maxHistory * 2 + 1)
+                    })
+                });
+                const data = await response.json();
+                data.status_log.forEach(log => {
+                    addStatusLog(log);
+                    updateStatusFromLog(log);
+                });
+                if (data.search_results) {
+                    if (Array.isArray(data.search_results) && data.search_results.length > 0 && 'DOI' in data.search_results[0]) {
+                        displaySearchResults(data.search_results, 'papers');
+                    } else {
+                        displaySearchResults(data.search_results, 'web');
+                    }
+                }
+                addMessage(data.response, false);
+                addStatusLog('回答完成');
+            } catch (error) {
+                console.error('Error:', error);
+                addMessage('发生错误，请稍后再试。', false);
+                updateStatus(false, false, false, false, false, false);
+                addStatusLog('发生错误');
+            }
+        }
+
+        sendBtn.addEventListener('click', sendMessage);
+        
+        userInput.addEventListener('keypress', (e) => {
+            if (e.key === 'Enter') {
+                sendMessage();
+            }
+        });
+
+        settingsBtn.addEventListener('click', () => {
+            settingsModal.classList.remove('hidden');
+            maxHistoryInput.value = maxHistory;
+        });
+
+        saveSettingsBtn.addEventListener('click', async () => {
+            const newMaxHistory = parseInt(maxHistoryInput.value);
+            if (newMaxHistory >= 1 && newMaxHistory <= 50) {
+                maxHistory = newMaxHistory;
+                try {
+                    const response = await fetch('/settings', {
+                        method: 'POST',
+                        headers: {
+                            'Content-Type': 'application/json'
+                        },
+                        body: JSON.stringify({ max_history: maxHistory })
+                    });
+                    const data = await response.json();
+                    if (data.status === 'success') {
+                        alert('设置已保存');
+                    } else {
+                        alert('保存设置时出错');
+                    }
+                } catch (error) {
+                    console.error('Error:', error);
+                    alert('保存设置时出错');
+                }
+                settingsModal.classList.add('hidden');
+            } else {
+                alert('请输入1到50之间的数字');
+            }
+        });
+
+        cancelSettingsBtn.addEventListener('click', () => {
+            settingsModal.classList.add('hidden');
+        });
+
+        darkModeToggle.addEventListener('click', () => {
+            document.body.classList.toggle('dark-mode');
+            updateDarkModeIcon();
+        });
+
+        function updateDarkModeIcon() {
+            const icon = darkModeToggle.querySelector('i');
+            if (document.body.classList.contains('dark-mode')) {
+                icon.classList.remove('fa-moon');
+                icon.classList.add('fa-sun');
+            } else {
+                icon.classList.remove('fa-sun');
+                icon.classList.add('fa-moon');
+            }
+        }
+
+        function initDarkMode() {
+            if (window.matchMedia && window.matchMedia('(prefers-color-scheme: dark)').matches) {
+                document.body.classList.add('dark-mode');
+            }
+            updateDarkModeIcon();
+        }
+
+        window.matchMedia('(prefers-color-scheme: dark)').addEventListener('change', e => {
+            if (e.matches) {
+                document.body.classList.add('dark-mode');
+            } else {
+                document.body.classList.remove('dark-mode');
+            }
+            updateDarkModeIcon();
+        });
+
+        sidebarToggle.addEventListener('click', () => {
+            sidebar.classList.toggle('open');
+            sidebarOverlay.classList.toggle('show');
+        });
+
+        sidebarOverlay.addEventListener('click', () => {
+            sidebar.classList.remove('open');
+            sidebarOverlay.classList.remove('show');
+        });
+
+        window.addEventListener('load', () => {
+            initDarkMode();
+        });
+
+        window.addEventListener('resize', () => {
+            if (window.innerWidth >= 768) {
+                sidebar.classList.remove('open');
+                sidebarOverlay.classList.remove('show');
+            }
+        });
+
+        // 可以在这里添加任何其他需要的 JavaScript 功能
+    </script>
+</body>
+</html>