Upload 7 files

README .md

@@ -0,0 +1,119 @@
+---
+title: Abacus Chat Proxy
+emoji: 🤖
+colorFrom: blue
+colorTo: purple
+sdk: docker
+sdk_version: "3.9"
+app_file: app.py
+pinned: false
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+
+# Abacus Chat Proxy
+
+> 📢 本项目基于 [orbitoo/abacus_chat_proxy](https://github.com/orbitoo/abacus_chat_proxy) 改进
+> 
+> 特别感谢 orbitoo 大佬提供的原始项目！
+> 
+> 本项目增加了：Docker部署支持、Hugging Face一键部署、自动保活功能等
+
+一个用于中转API请求的代理服务器。
+
+#### 注意创建space的时候把private改为public，因为不改为public导致的错误别来找我
+
+[![Deploy to Hugging Face Spaces](https://huggingface.co/datasets/huggingface/badges/raw/main/deploy-to-spaces-lg.svg)](https://huggingface.co/spaces/malt666/abacus_chat_proxy?duplicate=true)
+
+## ⚠️ 警告
+
+**本地部署方式已失效！**为了适配hugging face，本项目的本地部署方式已不再可用。目前只能通过Hugging Face Spaces部署来使用本代理服务。请使用下方的Hugging Face一键部署方法。
+
+## 🚀 快速开始
+
+### Hugging Face一键部署
+
+#### 注意创建space的时候把private改为public，因为不改为public导致的错误别来找我
+
+1. 点击上方的"Deploy to Hugging Face Spaces"按钮
+2. 登录你的Hugging Face账号（如果还没有，需要注册一个）
+3. 在弹出的页面中设置你的Space名称，注意创建space的时候把private改为public，因为不改为为public导致的错误别来找我
+4. 创建完Space后，在Space的Settings -> Repository Secrets中添加以下配置：
+   - `cookie_1`: 你的cookies字符串
+   - `password`: （必填，最好超过8位数，不然被盗用api导致点数用光，损失自负）api调用密码，也是dashboard登录密码
+5. 等待自动部署完成即可
+6. 登录dashboard查看使用情况，space里面是没办法登录的，点击弹出的登录页面的小窗口提示“请点击 https://xyz-abacus-chat-proxy.hf.space 来登录并查看使用情况”的链接来登录，登录密码是你设置的password
+7. **获取API链接**：部署成功后，网络的登录页面会显示api接口链接；或者你点击右上角的三个点按钮，在弹出的选项卡里面点击"Embed this Space"，然后在弹出的"Embed this Space"界面里的"Direct URL"就是你的访问链接，你可以用这个链接调用API和查看使用情况
+8. api调用：api调用的时候注意密钥填写你的password，不然没办法调用
+### 本地运行（已失效）
+
+> ⚠️ 以下本地运行方法已失效，仅作参考。请使用Hugging Face部署方式。
+
+#### Windows用户
+
+1. 双击运行 `start.bat`
+2. 首次运行选择 `0` 进行配置
+3. 配置完成后选择 `Y` 直接启动，或 `N` 返回菜单
+4. 之后可直接选择 `1` 启动代理
+5. 代理服务器默认运行在 `http://127.0.0.1:9876/`
+
+#### Linux/macOS用户
+
+```bash
+# 赋予脚本执行权限
+chmod +x start.sh
+
+# 运行脚本
+./start.sh
+```
+
+选项说明同Windows。
+
+### 🌐 Hugging Face部署
+
+1. Fork本仓库到你的GitHub账号
+2. 在Hugging Face上创建新的Space（选择Docker类型）
+3. 在Space的设置中连接你的GitHub仓库
+4. 在Space的设置中添加以下Secrets：
+   - 第1组配置：
+     - `cookie_1`: 第1个cookies字符串
+   - 第2组配置（如果需要）：
+     - `cookie_2`: 第2个cookies字符串
+   - 更多配置以此类推（`cookie_3`...）
+   - `password`: （可选）访问密码
+5. Space会自动部署，服务将在 `https://你的空间名-你的用户名.hf.space` 上运行
+
+## ⚙️ 环境要求
+
+- Python 3.8+
+- pip
+
+## 📦 依赖
+
+```bash
+Flask==3.1.0
+requests==2.32.3
+PyJWT==2.8.0
+```
+
+## 📝 配置说明
+
+### 本地配置
+
+首次运行时，请选择 `0` 进行配置，按照提示填写相关信息。配置文件将保存在 `config.json` 中。
+
+### 环境变量配置
+
+在Docker或云平台部署时，需要配置以下环境变量：
+
+- 必需的配置（至少需要一组）：
+  - `cookie_1`: 第1组配置
+  - `cookie_2`: 第2组配置（可选）
+  - 以此类推...
+- 可选配置：
+  - `password`: 访问密码
+
+## 🔒 安全说明
+
+- 建议在部署到Hugging Face时设置访问密码
+- 在Hugging Face上配置时，请使用Secrets来存储敏感信息 

app.py

@@ -26,6 +26,7 @@ USER_INFO_URL = "https://abacus.ai/api/v0/_getUserInfo"
 COMPUTE_POINTS_URL = "https://apps.abacus.ai/api/_getOrganizationComputePoints"
 COMPUTE_POINTS_LOG_URL = "https://abacus.ai/api/v0/_getOrganizationComputePointLog"
 CREATE_CONVERSATION_URL = "https://apps.abacus.ai/api/createDeploymentConversation"
+DELETE_CONVERSATION_URL = "https://apps.abacus.ai/api/deleteDeploymentConversation"
 
 
 USER_AGENTS = [
@@ -38,6 +39,9 @@ USER_NUM = 0
 USER_DATA = []
 CURRENT_USER = -1
 MODELS = set()
+# 添加变量跟踪是否删除上一个对话以及上一个对话ID
+DELETE_CHAT = True # 默认启用删除功能
+PREVIOUS_CONVERSATION_IDS = {} # 用于跟踪每个用户的上一个conversation_id
 
 
 TRACE_ID = "3042e28b3abf475d8d973c7e904935af"
@@ -81,7 +85,6 @@ users_compute_points = []
 # 记录启动时间
 START_TIME = datetime.utcnow() + timedelta(hours=8)  # 北京时间
 
-
 # 自定义JSON编码器，处理datetime对象
 class DateTimeEncoder(json.JSONEncoder):
     def default(self, obj):
@@ -90,32 +93,6 @@ class DateTimeEncoder(json.JSONEncoder):
         return super(DateTimeEncoder, self).default(obj)
 
 
-# 加载模型调用记录
-def load_model_usage_records():
-    global model_usage_records
-    try:
-        if os.path.exists(MODEL_USAGE_RECORDS_FILE):
-            with open(MODEL_USAGE_RECORDS_FILE, 'r', encoding='utf-8') as f:
-                records = json.load(f)
-                if isinstance(records, list):
-                    model_usage_records = records
-                    print(f"成功加载 {len(model_usage_records)} 条模型调用记录")
-                else:
-                    print("调用记录文件格式不正确，初始化为空列表")
-    except Exception as e:
-        print(f"加载模型调用记录失败: {e}")
-        model_usage_records = []
-
-# 保存模型调用记录
-def save_model_usage_records():
-    try:
-        with open(MODEL_USAGE_RECORDS_FILE, 'w', encoding='utf-8') as f:
-            json.dump(model_usage_records, f, ensure_ascii=False, indent=2, cls=DateTimeEncoder)
-        print(f"成功保存 {len(model_usage_records)} 条模型调用记录")
-    except Exception as e:
-        print(f"保存模型调用记录失败: {e}")
-
-
 def update_conversation_id(user_index, conversation_id):
     """更新用户的conversation_id并保存到配置文件"""
     try:
@@ -136,39 +113,46 @@ def update_conversation_id(user_index, conversation_id):
         print(f"更新conversation_id失败: {e}")
 
 
-def resolve_config():
-    # 从环境变量读取多组配置
-    config_list = []
-    i = 1
-    while True:
-        cookie = os.environ.get(f"cookie_{i}")
-        if not cookie:
-            break
-        
-        # 为每个cookie创建一个配置项，conversation_id初始为空
-        config_list.append({
-            "conversation_id": "",  # 初始为空，将通过get_or_create_conversation自动创建
-            "cookies": cookie
-        })
-        i += 1
-    
-    # 如果环境变量存在配置，使用环境变量的配置
-    if config_list:
-        print(f"从环境变量加载了 {len(config_list)} 个配置")
-        return config_list
-    
-    # 如果环境变量不存在，从文件读取
+def load_config():
+    """从配置文件加载配置"""
+    global USER_DATA, CURRENT_USER
     try:
-        with open("config.json", "r") as f:
+        # 读取配置文件
+        with open('config.json', 'r') as f:
             config = json.load(f)
-        config_list = config.get("config")
-        return config_list
-    except FileNotFoundError:
-        print("未找到config.json文件")
-        return []
-    except json.JSONDecodeError:
-        print("config.json格式错误")
-        return []
+        
+        # 确保USER_DATA为空列表
+        USER_DATA = []
+        
+        # 如果配置文件中有配置项，则加载
+        if "config" in config and isinstance(config["config"], list):
+            for i, user_config in enumerate(config["config"]):
+                # 获取必要的配置项
+                cookies = user_config.get("cookies", "")
+                session_token = user_config.get("session_token", "")
+                models = user_config.get("models", {})
+                
+                # 不再从配置文件读取conversation_id，确保每次启动都使用新会话
+                conversation_id = None
+                
+                # 创建会话对象
+                session = requests.Session()
+                
+                # 添加到USER_DATA
+                USER_DATA.append((session, cookies, session_token, conversation_id, models, i))
+                
+            print(f"成功加载了 {len(USER_DATA)} 个用户配置")
+            
+            # 设置当前用户为第一个用户
+            CURRENT_USER = 0
+            
+            return True
+        else:
+            print("配置文件格式错误")
+            return False
+    except Exception as e:
+        print(f"加载配置失败: {e}")
+        return False
 
 
 def get_password():
@@ -248,11 +232,10 @@ def refresh_token(session, cookies):
     }
     
     try:
-        response = session.post(
+        response = requests.post(
             USER_INFO_URL,
             headers=headers,
-            json={},
-            cookies=None
+            json={}
         )
         
         if response.status_code == 200:
@@ -297,11 +280,10 @@ def get_model_map(session, cookies, session_token):
     models_set = set()
     
     try:
-        response = session.post(
+        response = requests.post(
             MODEL_LIST_URL,
             headers=headers,
-            json={},
-            cookies=None
+            json={}
         )
         
         if response.status_code != 200:
@@ -490,7 +472,7 @@ def create_conversation(session, cookies, session_token, external_application_id
     }
     
     try:
-        response = session.post(
+        response = requests.post(
             CREATE_CONVERSATION_URL,
             headers=headers,
             json=create_payload
@@ -511,6 +493,49 @@ def create_conversation(session, cookies, session_token, external_application_id
         return None
 
 
+def delete_conversation(session, cookies, session_token, conversation_id):
+    """删除指定的会话"""
+    if not conversation_id:
+        print("无法删除会话: conversation_id为空")
+        return False
+    
+    headers = {
+        "accept": "application/json, text/plain, */*",
+        "accept-language": "zh-CN,zh;q=0.9",
+        "content-type": "application/json",
+        "cookie": cookies,
+        "user-agent": random.choice(USER_AGENTS),
+        "x-abacus-org-host": "apps"
+    }
+    
+    if session_token:
+        headers["session-token"] = session_token
+    
+    delete_payload = {
+        "deploymentId": "14b2a314cc",
+        "deploymentConversationId": conversation_id
+    }
+    
+    try:
+        response = requests.post(
+            DELETE_CONVERSATION_URL,
+            headers=headers,
+            json=delete_payload
+        )
+        
+        if response.status_code == 200:
+            data = response.json()
+            if data.get("success", False):
+                print(f"成功删除conversation: {conversation_id}")
+                return True
+        
+        print(f"删除会话失败: {response.status_code} - {response.text[:100]}")
+        return False
+    except Exception as e:
+        print(f"删除会话时出错: {e}")
+        return False
+
+
 def is_conversation_valid(session, cookies, session_token, conversation_id, model_map, model):
     """检查会话ID是否有效"""
     if not conversation_id:
@@ -542,7 +567,8 @@ def is_conversation_valid(session, cookies, session_token, conversation_id, mode
     }
     
     try:
-        response = session.post(
+        # 直接发送请求
+        response = requests.post(
             CHAT_URL,
             headers=headers,
             data=json.dumps(payload),
@@ -566,42 +592,33 @@ def is_conversation_valid(session, cookies, session_token, conversation_id, mode
 
 def get_or_create_conversation(session, cookies, session_token, conversation_id, model_map, model, user_index):
     """获取有效的会话ID，如果无效则创建新会话"""
-    # 如果conversation_id为None或为空字符串，直接创建新会话
-    if not conversation_id:
-        print("会话ID为空，将创建新会话")
-        need_create = True
-    else:
-        # 检查现有会话ID是否有效
-        need_create = not is_conversation_valid(session, cookies, session_token, conversation_id, model_map, model)
-        if need_create:
-            print(f"会话ID {conversation_id} 无效，将创建新会话")
-    
-    # 如果需要创建新会话
-    if need_create:
-        if model in model_map and len(model_map[model]) >= 2:
-            external_app_id = model_map[model][0]
-            # 创建会话时需要deployment_id，我们先使用一个固定值
-            # 在实际应用中应从API响应中获取
-            deployment_id = "14b2a314cc"  # 这是从您提供的请求中获取的
+    # 始终创建新会话，不再检查现有会话是否有效
+    print("正在创建新会话...")
+    
+    if model in model_map and len(model_map[model]) >= 2:
+        external_app_id = model_map[model][0]
+        # 创建会话时需要deployment_id，我们先使用一个固定值
+        # 在实际应用中应从API响应中获取
+        deployment_id = "14b2a314cc"  # 这是从您提供的请求中获取的
+        
+        new_conversation_id = create_conversation(
+            session, cookies, session_token, 
+            external_application_id=external_app_id,
+            deployment_id=deployment_id
+        )
+        
+        if new_conversation_id:
+            # 更新全局存储的会话ID
+            global USER_DATA, CURRENT_USER
+            session, cookies, session_token, _, model_map, _ = USER_DATA[CURRENT_USER]
+            USER_DATA[CURRENT_USER] = (session, cookies, session_token, new_conversation_id, model_map, user_index)
             
-            new_conversation_id = create_conversation(
-                session, cookies, session_token, 
-                external_application_id=external_app_id,
-                deployment_id=deployment_id
-            )
+            # 不再保存到配置文件，确保每次都创建新会话
+            # update_conversation_id(user_index, new_conversation_id)
             
-            if new_conversation_id:
-                # 更新全局存储的会话ID
-                global USER_DATA, CURRENT_USER
-                session, cookies, session_token, _, model_map, _ = USER_DATA[CURRENT_USER]
-                USER_DATA[CURRENT_USER] = (session, cookies, session_token, new_conversation_id, model_map, user_index)
-                
-                # 保存到配置文件
-                update_conversation_id(user_index, new_conversation_id)
-                
-                return new_conversation_id
+            return new_conversation_id
     
-    # 如果无法创建，返回原始ID
+    # 如果创建失败，返回原始ID
     return conversation_id
 
 
@@ -616,8 +633,14 @@ def send_message(message, model, think=False):
     """Flua traktado kaj plusendo de mesaĝoj"""
     (session, cookies, session_token, conversation_id, model_map, user_index) = get_user_data()
     
-    # 确保有有效的会话ID
-    conversation_id = get_or_create_conversation(session, cookies, session_token, conversation_id, model_map, model, user_index)
+    # 保存当前用户的上一个对话ID用于稍后删除
+    previous_conversation_id = PREVIOUS_CONVERSATION_IDS.get(user_index)
+    
+    # 确保有有效的会话ID，现在总是会创建新的
+    new_conversation_id = get_or_create_conversation(session, cookies, session_token, conversation_id, model_map, model, user_index)
+    
+    # 更新当前用户的会话ID记录
+    PREVIOUS_CONVERSATION_IDS[user_index] = new_conversation_id
     
     trace_id, sentry_trace = generate_trace_id()
     
@@ -646,7 +669,7 @@ def send_message(message, model, think=False):
     
     payload = {
         "requestId": str(uuid.uuid4()),
-        "deploymentConversationId": conversation_id,
+        "deploymentConversationId": new_conversation_id,
         "message": message,
         "isDesktop": False,
         "chatConfig": {
@@ -663,7 +686,7 @@ def send_message(message, model, think=False):
         payload["useThinking"] = think
     
     try:
-        response = session.post(
+        response = requests.post(
             CHAT_URL,
             headers=headers,
             data=json.dumps(payload),
@@ -732,6 +755,10 @@ def send_message(message, model, think=False):
             # 在流式传输完成后计算token并更新统计
             completion_tokens = num_tokens_from_string(completion_buffer.getvalue())
             update_model_stats(model, prompt_tokens, completion_tokens)
+            
+            # 在流式传输完成后删除上一个对话（如果存在且启用了删除功能）
+            if DELETE_CHAT and previous_conversation_id and previous_conversation_id != new_conversation_id:
+                delete_conversation(session, cookies, session_token, previous_conversation_id)
         
         return Response(generate(), mimetype="text/event-stream")
     except requests.exceptions.RequestException as e:
@@ -747,8 +774,14 @@ def send_message_non_stream(message, model, think=False):
     """Ne-flua traktado de mesaĝoj"""
     (session, cookies, session_token, conversation_id, model_map, user_index) = get_user_data()
     
-    # 确保有有效的会话ID
-    conversation_id = get_or_create_conversation(session, cookies, session_token, conversation_id, model_map, model, user_index)
+    # 保存当前用户的上一个对话ID用于稍后删除
+    previous_conversation_id = PREVIOUS_CONVERSATION_IDS.get(user_index)
+    
+    # 确保有有效的会话ID，现在总是会创建新的
+    new_conversation_id = get_or_create_conversation(session, cookies, session_token, conversation_id, model_map, model, user_index)
+    
+    # 更新当前用户的会话ID记录
+    PREVIOUS_CONVERSATION_IDS[user_index] = new_conversation_id
     
     trace_id, sentry_trace = generate_trace_id()
     
@@ -776,7 +809,7 @@ def send_message_non_stream(message, model, think=False):
     
     payload = {
         "requestId": str(uuid.uuid4()),
-        "deploymentConversationId": conversation_id,
+        "deploymentConversationId": new_conversation_id,
         "message": message,
         "isDesktop": False,
         "chatConfig": {
@@ -793,7 +826,7 @@ def send_message_non_stream(message, model, think=False):
         payload["useThinking"] = think
     
     try:
-        response = session.post(
+        response = requests.post(
             CHAT_URL,
             headers=headers,
             data=json.dumps(payload),
@@ -854,6 +887,10 @@ def send_message_non_stream(message, model, think=False):
             completion_tokens = num_tokens_from_string(think_content + response_content)
             update_model_stats(model, prompt_tokens, completion_tokens)
             
+            # 删除上一个对话（如果存在且启用了删除功能）
+            if DELETE_CHAT and previous_conversation_id and previous_conversation_id != new_conversation_id:
+                delete_conversation(session, cookies, session_token, previous_conversation_id)
+            
             return jsonify({
                 "id": f"chatcmpl-{str(uuid.uuid4())}",
                 "object": "chat.completion",
@@ -887,6 +924,10 @@ def send_message_non_stream(message, model, think=False):
             completion_tokens = num_tokens_from_string(response_content)
             update_model_stats(model, prompt_tokens, completion_tokens)
             
+            # 删除上一个对话（如果存在且启用了删除功能）
+            if DELETE_CHAT and previous_conversation_id and previous_conversation_id != new_conversation_id:
+                delete_conversation(session, cookies, session_token, previous_conversation_id)
+            
             return jsonify({
                 "id": f"chatcmpl-{str(uuid.uuid4())}",
                 "object": "chat.completion",
@@ -979,10 +1020,14 @@ def keep_alive():
     """每20分钟进行一次自我健康检查"""
     while True:
         try:
-            requests.get("http://127.0.0.1:7860/health")
+            # 获取当前端口
+            port = int(os.environ.get("PORT", 9876))
+            # 请求本地健康检查端点
+            requests.get(f"http://127.0.0.1:{port}/health")
             time.sleep(1200)  # 20分钟
         except:
-            pass  # 忽略错误，保持运行
+            # 忽略错误，保持运行
+            time.sleep(1200)  # 即使失败也等待20分钟
 
 
 @app.route("/", methods=["GET"])
@@ -1096,7 +1141,7 @@ def get_compute_points():
                 "cookie": cookies
             }
             
-            response = session.get(
+            response = requests.get(
                 COMPUTE_POINTS_URL,
                 headers=headers
             )
@@ -1170,7 +1215,7 @@ def get_compute_points_log(session, cookies, session_token):
             "cookie": cookies
         }
         
-        response = session.post(
+        response = requests.post(
             COMPUTE_POINTS_LOG_URL,
             headers=headers,
             json={"byLlm": True}
@@ -1251,10 +1296,26 @@ def dashboard():
         compute_points_log=compute_points_log,
         space_url=SPACE_URL,  # 传递空间URL
         users_compute_points=users_compute_points,  # 传递用户计算点信息
-        model_usage_records=model_usage_records  # 传递模型使用记录
+        model_usage_records=model_usage_records,  # 传递模型使用记录
+        delete_chat=DELETE_CHAT  # 传递删除对话开关状态
     )
 
 
+@app.route("/api/set_delete_chat", methods=["POST"])
+@require_auth
+def set_delete_chat():
+    """设置是否启用自动删除对话功能"""
+    global DELETE_CHAT
+    try:
+        data = request.get_json()
+        if "delete_chat" in data:
+            DELETE_CHAT = bool(data["delete_chat"])
+            return jsonify({"success": True, "delete_chat": DELETE_CHAT})
+        return jsonify({"success": False, "error": "Missing delete_chat parameter"}), 400
+    except Exception as e:
+        return jsonify({"success": False, "error": str(e)}), 500
+
+
 # 获取Hugging Face Space URL
 def get_space_url():
     # 尝试从环境变量获取
@@ -1310,6 +1371,32 @@ def auto_save_stats():
             print(f"自动保存/加载模型使用记录出错: {e}")
 
 
+# 加载模型调用记录
+def load_model_usage_records():
+    global model_usage_records
+    try:
+        if os.path.exists(MODEL_USAGE_RECORDS_FILE):
+            with open(MODEL_USAGE_RECORDS_FILE, 'r', encoding='utf-8') as f:
+                records = json.load(f)
+                if isinstance(records, list):
+                    model_usage_records = records
+                    print(f"成功加载 {len(model_usage_records)} 条模型调用记录")
+                else:
+                    print("调用记录文件格式不正确，初始化为空列表")
+    except Exception as e:
+        print(f"加载模型调用记录失败: {e}")
+        model_usage_records = []
+
+# 保存模型调用记录
+def save_model_usage_records():
+    try:
+        with open(MODEL_USAGE_RECORDS_FILE, 'w', encoding='utf-8') as f:
+            json.dump(model_usage_records, f, ensure_ascii=False, indent=2, cls=DateTimeEncoder)
+        print(f"成功保存 {len(model_usage_records)} 条模型调用记录")
+    except Exception as e:
+        print(f"保存模型调用记录失败: {e}")
+
+
 if __name__ == "__main__":
     # 启动保活线程
     threading.Thread(target=keep_alive, daemon=True).start()

config_editor.py

@@ -0,0 +1,72 @@
+import json
+import os
+import hashlib
+
+if __name__ == "__main__":
+    config = None
+    path = os.path.dirname(os.path.realpath(__file__))
+    os.chdir(path)
+    if os.path.exists("config.json"):
+        with open("config.json", "r") as f:
+            config = json.load(f)
+            try:
+                config["config"]
+            except KeyError:
+                config = None
+    if config is None:
+        print(f"配置文件不存在或为空，创建新配置...")
+        config = {"config": []}
+        print(f"输入会话ID: ")
+        user_data = {"conversation_id": input()}
+        print(f"输入cookies: ")
+        user_data["cookies"] = input()
+        config["config"].append(user_data)
+    
+    again = True
+    while True:
+        if again:
+            num = len(config["config"])
+            print(f"\n当前有 {num} 个配置。")
+        print("----------")
+        print(f"1. 添加新配置")
+        print(f"2. 删除所有配置")
+        print(f"3. 设置密码")
+        print(f"4. 保存并退出")
+        choice = input()
+        
+        if choice == "1":
+            print(f"输入会话ID: ")
+            user_data = {"conversation_id": input()}
+            print(f"输入cookies: ")
+            user_data["cookies"] = input()
+            config["config"].append(user_data)
+            print("\n成功添加配置！")
+            again = True
+        
+        elif choice == "2":
+            print("确定要删除所有配置吗? (y/n)")
+            if input().lower() == 'y':
+                config["config"] = []
+                print("已删除所有配置")
+            again = True
+        
+        elif choice == "3":
+            print(f"输入新密码（留空则删除密码）: ")
+            password = input()
+            with open("password.txt", "w") as f:
+                if password != "":
+                    f.write(hashlib.sha256(password.encode()).hexdigest())
+                    print(f"密码已设置")
+                else:
+                    f.write("")
+                    print(f"密码已删除")
+        
+        elif choice == "4":
+            with open("config.json", "w") as f:
+                json.dump(config, f, indent=4)
+            print("配置已保存")
+            break
+        
+        else:
+            print(f"无效的选择")
+            again = False

templates/dashboard.html

@@ -3,7 +3,7 @@
 <head>
     <meta charset="UTF-8">
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <title>Abacus Chat Proxy - 仪表盘</title>
+    <title>Abacus Chat Service - 仪表盘</title>
     <style>
         :root {
             --primary-color: #6f42c1;
@@ -504,6 +504,64 @@
                 grid-template-columns: 1fr;
             }
         }
+        
+        /* 开关样式 */
+        .switch {
+            position: relative;
+            display: inline-block;
+            width: 50px;
+            height: 24px;
+            margin-right: 10px;
+            vertical-align: middle;
+        }
+        
+        .switch input {
+            opacity: 0;
+            width: 0;
+            height: 0;
+        }
+        
+        .slider {
+            position: absolute;
+            cursor: pointer;
+            top: 0;
+            left: 0;
+            right: 0;
+            bottom: 0;
+            background-color: rgba(255, 255, 255, 0.2);
+            transition: .4s;
+        }
+        
+        .slider:before {
+            position: absolute;
+            content: "";
+            height: 18px;
+            width: 18px;
+            left: 3px;
+            bottom: 3px;
+            background-color: white;
+            transition: .4s;
+        }
+        
+        input:checked + .slider {
+            background-color: var(--primary-color);
+        }
+        
+        input:focus + .slider {
+            box-shadow: 0 0 1px var(--primary-color);
+        }
+        
+        input:checked + .slider:before {
+            transform: translateX(26px);
+        }
+        
+        .slider.round {
+            border-radius: 24px;
+        }
+        
+        .slider.round:before {
+            border-radius: 50%;
+        }
     </style>
 </head>
 <body>
@@ -512,7 +570,7 @@
     <nav class="navbar">
         <a href="/" class="navbar-brand">
             <span class="navbar-logo">🤖</span>
-            <span class="navbar-title">Abacus Chat Proxy</span>
+            <span class="navbar-title">Abacus Chat Service</span>
         </a>
         <div class="navbar-actions">
             <a href="/logout" class="btn-logout">
@@ -526,32 +584,40 @@
         <div class="card">
             <div class="card-header">
                 <h2 class="card-title">
-                    <span class="card-icon">📊</span>
-                    系统状态
+                    <div class="card-icon">🖥️</div>
+                    系统信息
                 </h2>
             </div>
-            <div class="status-item">
-                <span class="status-label">服务状态</span>
-                <span class="status-value success">运行中</span>
-            </div>
-            <div class="status-item">
-                <span class="status-label">运行时间</span>
-                <span class="status-value">{{ uptime }}</span>
-            </div>
-            <div class="status-item">
-                <span class="status-label">健康检查次数</span>
-                <span class="status-value">{{ health_checks }}</span>
-            </div>
-            <div class="status-item">
-                <span class="status-label">已配置用户数</span>
-                <span class="status-value">{{ user_count }}</span>
-            </div>
-            <div class="status-item">
-                <span class="status-label">可用模型</span>
-                <div class="models-list">
-                    {% for model in models %}
-                    <span class="model-tag">{{ model }}</span>
-                    {% endfor %}
+            <div>
+                <div class="status-item">
+                    <div class="status-label">系统状态</div>
+                    <div class="status-value success">运行中</div>
+                </div>
+                <div class="status-item">
+                    <div class="status-label">运行时间</div>
+                    <div class="status-value">{{ uptime }}</div>
+                </div>
+                <div class="status-item">
+                    <div class="status-label">健康检查次数</div>
+                    <div class="status-value">{{ health_checks }}</div>
+                </div>
+                <div class="status-item">
+                    <div class="status-label">已配置用户数量</div>
+                    <div class="status-value">{{ user_count }}</div>
+                </div>
+                <div class="status-item">
+                    <div class="status-label">可用模型数量</div>
+                    <div class="status-value">{{ models|length }}</div>
+                </div>
+                <div class="status-item">
+                    <div class="status-label">自动删除对话</div>
+                    <div class="status-value">
+                        <label class="switch">
+                            <input type="checkbox" id="deleteChat" {% if delete_chat %}checked{% endif %}>
+                            <span class="slider round"></span>
+                        </label>
+                        <span id="deleteChatStatus">{{ "已启用" if delete_chat else "已禁用" }}</span>
+                    </div>
                 </div>
             </div>
         </div>
@@ -611,7 +677,7 @@
                     <span class="status-value token-count">{{ total_tokens.completion|int }}</span>
                 </div>
                 <div class="token-note">
-                    <small>* 以上数据仅统计通过本代理使用的token数量，不包含在Abacus官网直接使用的token。数值为粗略估计，可能与实际计费有差异。</small>
+                    <small>* 以上数据仅统计通过本服务使用的token数量，不包含在Abacus官网直接使用的token。数值为粗略估计，可能与实际计费有差异。</small>
                 </div>
                 <div class="table-container">
                     <table class="data-table token-model-table">
@@ -788,7 +854,7 @@
         </div>
         
         <div class="footer">
-            <p>© {{ year }} Abacus Chat Proxy. 保持简单，保持可靠。</p>
+            <p>© {{ year }} Abacus Chat Service. 保持简单，保持可靠。</p>
         </div>
     </div>
     
@@ -814,6 +880,41 @@
         // 初始化隐藏回到顶部按钮
         document.querySelector('.float-btn').style.opacity = '0';
         
+        // 自动删除对话开关功能
+        const deleteChatSwitch = document.getElementById('deleteChat');
+        const deleteChatStatus = document.getElementById('deleteChatStatus');
+        
+        if (deleteChatSwitch) {
+            deleteChatSwitch.addEventListener('change', () => {
+                // 发送API请求更新设置
+                fetch('/api/set_delete_chat', {
+                    method: 'POST',
+                    headers: {
+                        'Content-Type': 'application/json'
+                    },
+                    body: JSON.stringify({
+                        delete_chat: deleteChatSwitch.checked
+                    })
+                })
+                .then(response => response.json())
+                .then(data => {
+                    if (data.success) {
+                        deleteChatStatus.textContent = deleteChatSwitch.checked ? '已启用' : '已禁用';
+                    } else {
+                        alert('更新设置失败: ' + (data.error || '未知错误'));
+                        // 回滚UI状态
+                        deleteChatSwitch.checked = !deleteChatSwitch.checked;
+                    }
+                })
+                .catch(error => {
+                    console.error('更新设置出错:', error);
+                    alert('更新设置出错，请查看控制台');
+                    // 回滚UI状态
+                    deleteChatSwitch.checked = !deleteChatSwitch.checked;
+                });
+            });
+        }
+        
         // 模型统计折叠功能
         const toggleBtn = document.getElementById('toggleModelStats');
         const hiddenModels = document.querySelectorAll('.hidden-model');