Upload 5 files

app.py

@@ -43,10 +43,6 @@ USER_DATA = []
 CURRENT_USER = -1
 MODELS = set()
 
-# 添加记录上一个conversation_id的变量和删除标记
-LAST_CONVERSATION_IDS = [None] * 100  # 为每个用户记录上一个conversation_id
-DELETE_CHAT = True  # 是否在对话结束后删除上一个对话
-
 TRACE_ID = "3042e28b3abf475d8d973c7e904935af"
 SENTRY_TRACE = f"{TRACE_ID}-80d9d2538b2682d0"
 
@@ -598,98 +594,47 @@ def delete_conversation(session, cookies, session_token, conversation_id, deploy
         return False
 
 
-def is_conversation_valid(session, cookies, session_token, conversation_id, model_map, model):
-    """检查会话ID是否有效"""
-    if not conversation_id:
-        return False
+def get_or_create_conversation(session, cookies, session_token, conversation_id, model_map, model, user_index):
+    """获取对话ID，如果不存在则创建；返回是否是使用现有会话"""
+    print(f"\n----- 获取会话ID (用户 {user_index+1}) -----")
+    # 如果有现有的会话ID，直接使用
+    if conversation_id:
+        print(f"使用现有会话ID: {conversation_id}")
+        return conversation_id, True
+
+    # 如果没有会话ID，创建新的
+    print("没有会话ID，创建新会话...")
+    deployment_id = "14b2a314cc"
+    
+    # 确保模型信息存在
+    if model not in model_map or len(model_map[model]) < 2:
+        print(f"错误: 无法获取模型 {model} 的信息")
+        return None, False
     
-    # 如果没有这些信息，无法验证
-    if not (model in model_map and len(model_map[model]) >= 2):
-        return False
-        
     external_app_id = model_map[model][0]
     
-    # 尝试发送一个空消息来测试会话ID是否有效
-    headers = {
-        "accept": "text/event-stream",
-        "content-type": "text/plain;charset=UTF-8",
-        "cookie": cookies,
-        "user-agent": random.choice(USER_AGENTS)
-    }
-    
-    if session_token:
-        headers["session-token"] = session_token
-    
-    payload = {
-        "requestId": str(uuid.uuid4()),
-        "deploymentConversationId": conversation_id,
-        "message": "",  # 空消息
-        "isDesktop": False,
-        "externalApplicationId": external_app_id
-    }
+    # 创建新会话
+    new_conversation_id = create_conversation(
+        session, cookies, session_token, 
+        external_application_id=external_app_id,
+        deployment_id=deployment_id
+    )
     
-    try:
-        response = session.post(
-            CHAT_URL,
-            headers=headers,
-            data=json.dumps(payload),
-            stream=False
-        )
+    if new_conversation_id:
+        print(f"成功创建新会话ID: {new_conversation_id}")
         
-        # 即使返回错误，只要不是缺少ID的错误，也说明ID是有效的
-        if response.status_code == 200:
-            return True
+        # 更新全局存储的会话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)
         
-        error_text = response.text
-        if "Missing required parameter" in error_text:
-            return False
-            
-        # 其他类型的错误，可能ID是有效的但有其他问题
-        return True
-    except:
-        # 如果请求出错，无法确定，返回False让系统创建新ID
-        return False
-
-
-def get_or_create_conversation(session, cookies, session_token, conversation_id, model_map, model, user_index):
-    """获取有效的会话ID，如果无效则创建新会话"""
-    # 修改为总是创建新的conversation_id
-    print("将为每次对话创建新会话")
-    need_create = True
-    
-    # 如果需要创建新会话
-    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"  # 这是从您提供的请求中获取的
-            
-            new_conversation_id = create_conversation(
-                session, cookies, session_token, 
-                external_application_id=external_app_id,
-                deployment_id=deployment_id
-            )
-            
-            if new_conversation_id:
-                # 获取当前用户的上一个conversation_id
-                global USER_DATA, CURRENT_USER, LAST_CONVERSATION_IDS
-                last_conversation_id = LAST_CONVERSATION_IDS[user_index]
-                
-                # 更新全局存储的会话ID
-                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)
-                
-                # 保存新的会话ID为下次调用时的"上一个ID"
-                LAST_CONVERSATION_IDS[user_index] = new_conversation_id
-                
-                return new_conversation_id
+        # 保存到配置文件
+        update_conversation_id(user_index, new_conversation_id)
+        
+        return new_conversation_id, False
     
-    # 如果无法创建，返回原始ID
-    return conversation_id
+    print("创建新会话失败")
+    return None, False
 
 
 def generate_trace_id():
@@ -708,12 +653,16 @@ def send_message(message, model, think=False):
     (session, cookies, session_token, conversation_id, model_map, user_index) = get_user_data()
     print(f"使用用户配置: {user_index + 1}")
     
-    # 获取并保存当前的conversation_id（可能是旧的，用于稍后删除）
-    last_conversation_id = conversation_id
+    # 获取会话ID，并判断是否使用现有会话
+    conversation_id, is_existing = get_or_create_conversation(
+        session, cookies, session_token, conversation_id, model_map, model, user_index
+    )
     
-    # 确保有有效的会话ID
-    conversation_id = get_or_create_conversation(session, cookies, session_token, conversation_id, model_map, model, user_index)
-    print(f"会话ID: {conversation_id}")
+    # 如果没有有效的会话ID，返回错误
+    if not conversation_id:
+        return jsonify({"error": "Failed to get a valid conversation ID"}), 500
+    
+    print(f"会话ID: {conversation_id} (是否为现有会话: {is_existing})")
     
     trace_id, sentry_trace = generate_trace_id()
     
@@ -744,6 +693,7 @@ def send_message(message, model, think=False):
     if session_token:
         headers["session-token"] = session_token
     
+    # 构建基础请求
     payload = {
         "requestId": str(uuid.uuid4()),
         "deploymentConversationId": conversation_id,
@@ -757,6 +707,12 @@ def send_message(message, model, think=False):
         "externalApplicationId": model_map[model][0]
     }
     
+    # 如果是使用现有会话，添加regenerate和editPrompt参数
+    if is_existing:
+        payload["regenerate"] = True
+        payload["editPrompt"] = True
+        print("为现有会话添加 regenerate=True 和 editPrompt=True")
+    
     if think:
         payload["useThinking"] = think
     
@@ -844,6 +800,93 @@ def send_message(message, model, think=False):
             if hasattr(e.response, 'text'):
                 error_details += f" - Response: {e.response.text[:200]}"
         print(f"发送消息失败: {error_details}")
+        
+        # 如果是使用现有会话失败，尝试创建新会话重试一次
+        if is_existing:
+            print("使用现有会话失败，尝试创建新会话...")
+            # 创建新会话
+            deployment_id = "14b2a314cc"
+            external_app_id = model_map[model][0] if model in model_map and len(model_map[model]) >= 2 else None
+            
+            if external_app_id:
+                new_conversation_id = create_conversation(
+                    session, cookies, session_token, 
+                    external_application_id=external_app_id,
+                    deployment_id=deployment_id
+                )
+                
+                if new_conversation_id:
+                    print(f"成功创建新会话ID: {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)
+                    
+                    # 修改payload使用新会话ID，并移除regenerate和editPrompt
+                    payload["deploymentConversationId"] = new_conversation_id
+                    if "regenerate" in payload:
+                        del payload["regenerate"]
+                    if "editPrompt" in payload:
+                        del payload["editPrompt"]
+                    
+                    try:
+                        # 非流式重试逻辑与流式类似，但需要重新提取响应内容
+                        response = session.post(
+                            CHAT_URL,
+                            headers=headers,
+                            data=json.dumps(payload),
+                            stream=True,
+                            cookies=None
+                        )
+                        
+                        response.raise_for_status()
+                        # 重用现有提取逻辑...
+                        # 但这里代码重复太多，实际应该重构为共享函数
+                        buffer = io.StringIO()
+                        
+                        for line in response.iter_lines():
+                            if line:
+                                decoded_line = line.decode("utf-8")
+                                segment = extract_segment(decoded_line)
+                                if segment:
+                                    buffer.write(segment)
+                        
+                        response_content = buffer.getvalue()
+                        
+                        # 计算输出token并更新统计信息
+                        completion_result, _ = num_tokens_from_string(response_content, model)
+                        
+                        # 保存对话历史并获取计算点数
+                        _, compute_points_used = save_conversation_history(session, cookies, session_token, new_conversation_id)
+                        
+                        # 更新统计信息
+                        update_model_stats(model, prompt_tokens, completion_result, calculation_method, compute_points_used)
+                        
+                        return jsonify({
+                            "id": f"chatcmpl-{str(uuid.uuid4())}",
+                            "object": "chat.completion",
+                            "created": int(time.time()),
+                            "model": model,
+                            "choices": [{
+                                "index": 0,
+                                "message": {
+                                    "role": "assistant",
+                                    "content": response_content
+                                },
+                                "finish_reason": "stop"
+                            }],
+                            "usage": {
+                                "prompt_tokens": prompt_tokens,
+                                "completion_tokens": completion_result,
+                                "total_tokens": prompt_tokens + completion_result
+                            }
+                        })
+                    except Exception as retry_e:
+                        print(f"重试失败: {retry_e}")
+        
         return jsonify({"error": f"Failed to send message: {error_details}"}), 500
 
 
@@ -856,12 +899,16 @@ def send_message_non_stream(message, model, think=False):
     (session, cookies, session_token, conversation_id, model_map, user_index) = get_user_data()
     print(f"使用用户配置: {user_index + 1}")
     
-    # 获取并保存当前的conversation_id（可能是旧的，用于稍后删除）
-    last_conversation_id = conversation_id
+    # 获取会话ID，并判断是否使用现有会话
+    conversation_id, is_existing = get_or_create_conversation(
+        session, cookies, session_token, conversation_id, model_map, model, user_index
+    )
     
-    # 确保有有效的会话ID
-    conversation_id = get_or_create_conversation(session, cookies, session_token, conversation_id, model_map, model, user_index)
-    print(f"会话ID: {conversation_id}")
+    # 如果没有有效的会话ID，返回错误
+    if not conversation_id:
+        return jsonify({"error": "Failed to get a valid conversation ID"}), 500
+    
+    print(f"会话ID: {conversation_id} (是否为现有会话: {is_existing})")
     
     trace_id, sentry_trace = generate_trace_id()
     
@@ -891,6 +938,7 @@ def send_message_non_stream(message, model, think=False):
     if session_token:
         headers["session-token"] = session_token
     
+    # 构建基础请求
     payload = {
         "requestId": str(uuid.uuid4()),
         "deploymentConversationId": conversation_id,
@@ -904,6 +952,12 @@ def send_message_non_stream(message, model, think=False):
         "externalApplicationId": model_map[model][0]
     }
     
+    # 如果是使用现有会话，添加regenerate和editPrompt参数
+    if is_existing:
+        payload["regenerate"] = True
+        payload["editPrompt"] = True
+        print("为现有会话添加 regenerate=True 和 editPrompt=True")
+    
     if think:
         payload["useThinking"] = think
     
@@ -1038,6 +1092,93 @@ def send_message_non_stream(message, model, think=False):
             if hasattr(e.response, 'text'):
                 error_details += f" - Response: {e.response.text[:200]}"
         print(f"发送消息失败: {error_details}")
+        
+        # 如果是使用现有会话失败，尝试创建新会话重试一次
+        if is_existing:
+            print("使用现有会话失败，尝试创建新会话...")
+            # 创建新会话
+            deployment_id = "14b2a314cc"
+            external_app_id = model_map[model][0] if model in model_map and len(model_map[model]) >= 2 else None
+            
+            if external_app_id:
+                new_conversation_id = create_conversation(
+                    session, cookies, session_token, 
+                    external_application_id=external_app_id,
+                    deployment_id=deployment_id
+                )
+                
+                if new_conversation_id:
+                    print(f"成功创建新会话ID: {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)
+                    
+                    # 修改payload使用新会话ID，并移除regenerate和editPrompt
+                    payload["deploymentConversationId"] = new_conversation_id
+                    if "regenerate" in payload:
+                        del payload["regenerate"]
+                    if "editPrompt" in payload:
+                        del payload["editPrompt"]
+                    
+                    try:
+                        # 非流式重试逻辑与流式类似，但需要重新提取响应内容
+                        response = session.post(
+                            CHAT_URL,
+                            headers=headers,
+                            data=json.dumps(payload),
+                            stream=True,
+                            cookies=None
+                        )
+                        
+                        response.raise_for_status()
+                        # 重用现有提取逻辑...
+                        # 但这里代码重复太多，实际应该重构为共享函数
+                        buffer = io.StringIO()
+                        
+                        for line in response.iter_lines():
+                            if line:
+                                decoded_line = line.decode("utf-8")
+                                segment = extract_segment(decoded_line)
+                                if segment:
+                                    buffer.write(segment)
+                        
+                        response_content = buffer.getvalue()
+                        
+                        # 计算输出token并更新统计信息
+                        completion_result, _ = num_tokens_from_string(response_content, model)
+                        
+                        # 保存对话历史并获取计算点数
+                        _, compute_points_used = save_conversation_history(session, cookies, session_token, new_conversation_id)
+                        
+                        # 更新统计信息
+                        update_model_stats(model, prompt_tokens, completion_result, calculation_method, compute_points_used)
+                        
+                        return jsonify({
+                            "id": f"chatcmpl-{str(uuid.uuid4())}",
+                            "object": "chat.completion",
+                            "created": int(time.time()),
+                            "model": model,
+                            "choices": [{
+                                "index": 0,
+                                "message": {
+                                    "role": "assistant",
+                                    "content": response_content
+                                },
+                                "finish_reason": "stop"
+                            }],
+                            "usage": {
+                                "prompt_tokens": prompt_tokens,
+                                "completion_tokens": completion_result,
+                                "total_tokens": prompt_tokens + completion_result
+                            }
+                        })
+                    except Exception as retry_e:
+                        print(f"重试失败: {retry_e}")
+        
         return jsonify({"error": f"Failed to send message: {error_details}"}), 500
 
 

templates/login.html

@@ -1,462 +1,463 @@
-<!DOCTYPE html>
-<html lang="zh-CN">
-<head>
-    <meta charset="UTF-8">
-    <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <title>Abacus Chat Proxy - 登录</title>
-    <style>
-        :root {
-            --primary-color: #6f42c1;
-            --secondary-color: #4a32a8;
-            --bg-color: #0a0a1a;
-            --text-color: #e6e6ff;
-            --card-bg: rgba(30, 30, 60, 0.7);
-            --input-bg: rgba(40, 40, 80, 0.6);
-            --success-color: #36d399;
-            --error-color: #f87272;
-        }
-        
-        * {
-            margin: 0;
-            padding: 0;
-            box-sizing: border-box;
-            font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
-        }
-        
-        body {
-            min-height: 100vh;
-            display: flex;
-            align-items: center;
-            justify-content: center;
-            background-color: var(--bg-color);
-            background-image: 
-                radial-gradient(circle at 20% 35%, rgba(111, 66, 193, 0.15) 0%, transparent 40%),
-                radial-gradient(circle at 80% 10%, rgba(70, 111, 171, 0.1) 0%, transparent 40%);
-            color: var(--text-color);
-            position: relative;
-            overflow: hidden;
-        }
-        
-        /* 科幻蜘蛛网动画 */
-        .web-container {
-            position: absolute;
-            top: 0;
-            left: 0;
-            width: 100%;
-            height: 100%;
-            z-index: -2;
-            opacity: 0.6;
-        }
-        
-        .web {
-            width: 100%;
-            height: 100%;
-        }
-        
-        /* 动态背景网格 */
-        .grid-background {
-            position: absolute;
-            top: 0;
-            left: 0;
-            width: 100%;
-            height: 100%;
-            background-image: linear-gradient(rgba(50, 50, 100, 0.05) 1px, transparent 1px),
-                              linear-gradient(90deg, rgba(50, 50, 100, 0.05) 1px, transparent 1px);
-            background-size: 30px 30px;
-            z-index: -1;
-            animation: grid-move 20s linear infinite;
-        }
-        
-        @keyframes grid-move {
-            0% {
-                transform: translateY(0);
-            }
-            100% {
-                transform: translateY(30px);
-            }
-        }
-        
-        /* 浮动粒子效果 */
-        .particles {
-            position: absolute;
-            top: 0;
-            left: 0;
-            width: 100%;
-            height: 100%;
-            overflow: hidden;
-            z-index: -1;
-        }
-        
-        .particle {
-            position: absolute;
-            display: block;
-            pointer-events: none;
-            width: 6px;
-            height: 6px;
-            background-color: rgba(111, 66, 193, 0.2);
-            border-radius: 50%;
-            animation: float 20s infinite ease-in-out;
-        }
-        
-        @keyframes float {
-            0%, 100% {
-                transform: translateY(0) translateX(0);
-                opacity: 0;
-            }
-            50% {
-                opacity: 0.5;
-            }
-            25%, 75% {
-                transform: translateY(-100px) translateX(50px);
-            }
-        }
-        
-        .login-card {
-            width: 420px;
-            padding: 2.5rem;
-            border-radius: 16px;
-            background: var(--card-bg);
-            box-shadow: 0 8px 32px rgba(0, 0, 0, 0.2);
-            backdrop-filter: blur(8px);
-            border: 1px solid rgba(255, 255, 255, 0.1);
-            z-index: 10;
-            animation: card-fade-in 0.6s ease-out;
-        }
-        
-        @keyframes card-fade-in {
-            from {
-                opacity: 0;
-                transform: translateY(20px);
-            }
-            to {
-                opacity: 1;
-                transform: translateY(0);
-            }
-        }
-        
-        .login-header {
-            text-align: center;
-            margin-bottom: 2rem;
-        }
-        
-        .login-header h1 {
-            font-size: 2rem;
-            font-weight: 600;
-            margin-bottom: 0.5rem;
-            background: linear-gradient(45deg, #6f42c1, #5181f1);
-            -webkit-background-clip: text;
-            -webkit-text-fill-color: transparent;
-            letter-spacing: 0.5px;
-        }
-        
-        .login-header p {
-            color: rgba(230, 230, 255, 0.7);
-            font-size: 0.95rem;
-        }
-        
-        .space-info {
-            text-align: center;
-            background: rgba(50, 50, 150, 0.2);
-            padding: 0.75rem;
-            border-radius: 8px;
-            margin-bottom: 1.5rem;
-            font-size: 0.9rem;
-            border: 1px solid rgba(111, 66, 193, 0.3);
-        }
-        
-        .space-info a {
-            color: var(--primary-color);
-            text-decoration: none;
-            font-weight: bold;
-            transition: all 0.2s;
-        }
-        
-        .space-info a:hover {
-            text-decoration: underline;
-            color: var(--secondary-color);
-        }
-        
-        .login-form {
-            display: flex;
-            flex-direction: column;
-        }
-        
-        .form-group {
-            margin-bottom: 1.5rem;
-            position: relative;
-        }
-        
-        .form-group label {
-            display: block;
-            margin-bottom: 0.5rem;
-            font-size: 0.9rem;
-            font-weight: 500;
-            color: rgba(230, 230, 255, 0.9);
-        }
-        
-        .form-control {
-            width: 100%;
-            padding: 0.75rem 1rem;
-            font-size: 1rem;
-            line-height: 1.5;
-            color: var(--text-color);
-            background-color: var(--input-bg);
-            border: 1px solid rgba(255, 255, 255, 0.1);
-            border-radius: 8px;
-            transition: all 0.2s ease;
-            outline: none;
-        }
-        
-        .form-control:focus {
-            border-color: var(--primary-color);
-            box-shadow: 0 0 0 3px rgba(111, 66, 193, 0.2);
-        }
-        
-        .btn {
-            display: inline-block;
-            font-weight: 500;
-            text-align: center;
-            vertical-align: middle;
-            cursor: pointer;
-            padding: 0.75rem 1rem;
-            font-size: 1rem;
-            line-height: 1.5;
-            border-radius: 8px;
-            transition: all 0.15s ease-in-out;
-            border: none;
-        }
-        
-        .btn-primary {
-            color: #fff;
-            background: linear-gradient(45deg, var(--primary-color), var(--secondary-color));
-            box-shadow: 0 4px 10px rgba(111, 66, 193, 0.3);
-            position: relative;
-            overflow: hidden;
-        }
-        
-        .btn-primary:hover {
-            transform: translateY(-2px);
-            box-shadow: 0 6px 15px rgba(111, 66, 193, 0.4);
-        }
-        
-        .btn-primary:active {
-            transform: translateY(0);
-        }
-        
-        /* 添加光效效果 */
-        .btn-primary::before {
-            content: '';
-            position: absolute;
-            top: -50%;
-            left: -50%;
-            width: 200%;
-            height: 200%;
-            background: linear-gradient(
-                to bottom right,
-                rgba(255, 255, 255, 0) 0%,
-                rgba(255, 255, 255, 0.1) 50%,
-                rgba(255, 255, 255, 0) 100%
-            );
-            transform: rotate(45deg);
-            animation: btn-shine 3s infinite;
-        }
-        
-        @keyframes btn-shine {
-            0% {
-                left: -50%;
-            }
-            100% {
-                left: 150%;
-            }
-        }
-        
-        .error-message {
-            background-color: rgba(248, 114, 114, 0.2);
-            color: var(--error-color);
-            padding: 0.75rem;
-            border-radius: 6px;
-            margin-bottom: 1.5rem;
-            font-size: 0.9rem;
-            border-left: 3px solid var(--error-color);
-            display: {{ 'block' if error else 'none' }};
-        }
-        
-        .logo {
-            margin-bottom: 1rem;
-            font-size: 3rem;
-            animation: glow 2s infinite alternate;
-        }
-        
-        @keyframes glow {
-            from {
-                text-shadow: 0 0 5px rgba(111, 66, 193, 0.5), 0 0 10px rgba(111, 66, 193, 0.5);
-            }
-            to {
-                text-shadow: 0 0 10px rgba(111, 66, 193, 0.8), 0 0 20px rgba(111, 66, 193, 0.8);
-            }
-        }
-    </style>
-</head>
-<body>
-    <div class="grid-background"></div>
-    <div class="particles">
-        <!-- 粒子元素会由JS生成 -->
-    </div>
-    <div class="web-container">
-        <canvas class="web" id="webCanvas"></canvas>
-    </div>
-    
-    <div class="login-card">
-        <div class="login-header">
-            <div class="logo">🤖</div>
-            <h1>Abacus Chat Proxy</h1>
-            <p>请输入访问密码</p>
-        </div>
-        
-        {% if space_url %}
-        <div class="space-info">
-            api接口为{{ space_url }}/v1，请点击 <a href="{{ space_url }}" target="_blank">{{ space_url }}</a> 来登录并查看使用情况，
-        </div>
-        {% endif %}
-        
-        <div class="error-message" id="error-message">
-            {{ error }}
-        </div>
-        
-        <form class="login-form" method="post" action="/login">
-            <div class="form-group">
-                <label for="password">密码</label>
-                <input type="password" class="form-control" id="password" name="password" placeholder="请输入访问密码" required>
-            </div>
-            
-            <button type="submit" class="btn btn-primary">登录</button>
-        </form>
-    </div>
-    
-    <script>
-        // 创建浮动粒子
-        function createParticles() {
-            const particlesContainer = document.querySelector('.particles');
-            const particleCount = 20;
-            
-            for (let i = 0; i < particleCount; i++) {
-                const particle = document.createElement('div');
-                particle.className = 'particle';
-                
-                // 随机位置和大小
-                const size = Math.random() * 5 + 2;
-                const x = Math.random() * 100;
-                const y = Math.random() * 100;
-                
-                particle.style.width = `${size}px`;
-                particle.style.height = `${size}px`;
-                particle.style.left = `${x}%`;
-                particle.style.top = `${y}%`;
-                
-                // 随机动画延迟
-                particle.style.animationDelay = `${Math.random() * 10}s`;
-                particle.style.animationDuration = `${Math.random() * 10 + 10}s`;
-                
-                // 随机透明度
-                particle.style.opacity = Math.random() * 0.5;
-                
-                particlesContainer.appendChild(particle);
-            }
-        }
-        
-        // 科幻蜘蛛网效果
-        function initWebCanvas() {
-            const canvas = document.getElementById('webCanvas');
-            const ctx = canvas.getContext('2d');
-            let width = window.innerWidth;
-            let height = window.innerHeight;
-            
-            // 设置canvas尺寸
-            canvas.width = width;
-            canvas.height = height;
-            
-            // 节点类
-            class Node {
-                constructor(x, y) {
-                    this.x = x;
-                    this.y = y;
-                    this.vx = (Math.random() - 0.5) * 0.5;
-                    this.vy = (Math.random() - 0.5) * 0.5;
-                    this.radius = Math.random() * 2 + 1;
-                }
-                
-                update() {
-                    if (this.x < 0 || this.x > width) this.vx = -this.vx;
-                    if (this.y < 0 || this.y > height) this.vy = -this.vy;
-                    
-                    this.x += this.vx;
-                    this.y += this.vy;
-                }
-                
-                draw() {
-                    ctx.beginPath();
-                    ctx.arc(this.x, this.y, this.radius, 0, Math.PI * 2);
-                    ctx.fillStyle = 'rgba(111, 66, 193, 0.4)';
-                    ctx.fill();
-                }
-            }
-            
-            // 创建节点
-            const nodeCount = Math.floor(width * height / 15000);
-            const nodes = [];
-            
-            for (let i = 0; i < nodeCount; i++) {
-                nodes.push(new Node(Math.random() * width, Math.random() * height));
-            }
-            
-            // 绘制线条
-            function drawWeb() {
-                ctx.clearRect(0, 0, width, height);
-                
-                // 更新节点
-                nodes.forEach(node => {
-                    node.update();
-                    node.draw();
-                });
-                
-                // 绘制连线
-                for (let i = 0; i < nodes.length; i++) {
-                    for (let j = i + 1; j < nodes.length; j++) {
-                        const dx = nodes[i].x - nodes[j].x;
-                        const dy = nodes[i].y - nodes[j].y;
-                        const distance = Math.sqrt(dx * dx + dy * dy);
-                        
-                        if (distance < 150) {
-                            ctx.beginPath();
-                            ctx.moveTo(nodes[i].x, nodes[i].y);
-                            ctx.lineTo(nodes[j].x, nodes[j].y);
-                            ctx.strokeStyle = `rgba(111, 66, 193, ${0.2 * (1 - distance / 150)})`;
-                            ctx.lineWidth = 0.5;
-                            ctx.stroke();
-                        }
-                    }
-                }
-                
-                requestAnimationFrame(drawWeb);
-            }
-            
-            // 监听窗口大小变化
-            window.addEventListener('resize', () => {
-                width = window.innerWidth;
-                height = window.innerHeight;
-                canvas.width = width;
-                canvas.height = height;
-            });
-            
-            // 开始动画
-            drawWeb();
-        }
-        
-        // 页面加载时初始化效果
-        window.addEventListener('load', () => {
-            createParticles();
-            initWebCanvas();
-        });
-    </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>Abacus Chat Proxy - 登录</title>
+    <style>
+        :root {
+            --primary-color: #6f42c1;
+            --secondary-color: #4a32a8;
+            --bg-color: #0a0a1a;
+            --text-color: #e6e6ff;
+            --card-bg: rgba(30, 30, 60, 0.7);
+            --input-bg: rgba(40, 40, 80, 0.6);
+            --success-color: #36d399;
+            --error-color: #f87272;
+        }
+        
+        * {
+            margin: 0;
+            padding: 0;
+            box-sizing: border-box;
+            font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
+        }
+        
+        body {
+            min-height: 100vh;
+            display: flex;
+            align-items: center;
+            justify-content: center;
+            background-color: var(--bg-color);
+            background-image: 
+                radial-gradient(circle at 20% 35%, rgba(111, 66, 193, 0.15) 0%, transparent 40%),
+                radial-gradient(circle at 80% 10%, rgba(70, 111, 171, 0.1) 0%, transparent 40%);
+            color: var(--text-color);
+            position: relative;
+            overflow: hidden;
+        }
+        
+        /* 科幻蜘蛛网动画 */
+        .web-container {
+            position: absolute;
+            top: 0;
+            left: 0;
+            width: 100%;
+            height: 100%;
+            z-index: -2;
+            opacity: 0.6;
+        }
+        
+        .web {
+            width: 100%;
+            height: 100%;
+        }
+        
+        /* 动态背景网格 */
+        .grid-background {
+            position: absolute;
+            top: 0;
+            left: 0;
+            width: 100%;
+            height: 100%;
+            background-image: linear-gradient(rgba(50, 50, 100, 0.05) 1px, transparent 1px),
+                              linear-gradient(90deg, rgba(50, 50, 100, 0.05) 1px, transparent 1px);
+            background-size: 30px 30px;
+            z-index: -1;
+            animation: grid-move 20s linear infinite;
+        }
+        
+        @keyframes grid-move {
+            0% {
+                transform: translateY(0);
+            }
+            100% {
+                transform: translateY(30px);
+            }
+        }
+        
+        /* 浮动粒子效果 */
+        .particles {
+            position: absolute;
+            top: 0;
+            left: 0;
+            width: 100%;
+            height: 100%;
+            overflow: hidden;
+            z-index: -1;
+        }
+        
+        .particle {
+            position: absolute;
+            display: block;
+            pointer-events: none;
+            width: 6px;
+            height: 6px;
+            background-color: rgba(111, 66, 193, 0.2);
+            border-radius: 50%;
+            animation: float 20s infinite ease-in-out;
+        }
+        
+        @keyframes float {
+            0%, 100% {
+                transform: translateY(0) translateX(0);
+                opacity: 0;
+            }
+            50% {
+                opacity: 0.5;
+            }
+            25%, 75% {
+                transform: translateY(-100px) translateX(50px);
+            }
+        }
+        
+        .login-card {
+            width: 420px;
+            padding: 2.5rem;
+            border-radius: 16px;
+            background: var(--card-bg);
+            box-shadow: 0 8px 32px rgba(0, 0, 0, 0.2);
+            backdrop-filter: blur(8px);
+            border: 1px solid rgba(255, 255, 255, 0.1);
+            z-index: 10;
+            animation: card-fade-in 0.6s ease-out;
+        }
+        
+        @keyframes card-fade-in {
+            from {
+                opacity: 0;
+                transform: translateY(20px);
+            }
+            to {
+                opacity: 1;
+                transform: translateY(0);
+            }
+        }
+        
+        .login-header {
+            text-align: center;
+            margin-bottom: 2rem;
+        }
+        
+        .login-header h1 {
+            font-size: 2rem;
+            font-weight: 600;
+            margin-bottom: 0.5rem;
+            background: linear-gradient(45deg, #6f42c1, #5181f1);
+            -webkit-background-clip: text;
+            -webkit-text-fill-color: transparent;
+            letter-spacing: 0.5px;
+        }
+        
+        .login-header p {
+            color: rgba(230, 230, 255, 0.7);
+            font-size: 0.95rem;
+        }
+        
+        .space-info {
+            text-align: center;
+            background: rgba(50, 50, 150, 0.2);
+            padding: 0.75rem;
+            border-radius: 8px;
+            margin-bottom: 1.5rem;
+            font-size: 0.9rem;
+            border: 1px solid rgba(111, 66, 193, 0.3);
+        }
+        
+        .space-info a {
+            color: var(--primary-color);
+            text-decoration: none;
+            font-weight: bold;
+            transition: all 0.2s;
+        }
+        
+        .space-info a:hover {
+            text-decoration: underline;
+            color: var(--secondary-color);
+        }
+        
+        .login-form {
+            display: flex;
+            flex-direction: column;
+        }
+        
+        .form-group {
+            margin-bottom: 1.5rem;
+            position: relative;
+        }
+        
+        .form-group label {
+            display: block;
+            margin-bottom: 0.5rem;
+            font-size: 0.9rem;
+            font-weight: 500;
+            color: rgba(230, 230, 255, 0.9);
+        }
+        
+        .form-control {
+            width: 100%;
+            padding: 0.75rem 1rem;
+            font-size: 1rem;
+            line-height: 1.5;
+            color: var(--text-color);
+            background-color: var(--input-bg);
+            border: 1px solid rgba(255, 255, 255, 0.1);
+            border-radius: 8px;
+            transition: all 0.2s ease;
+            outline: none;
+        }
+        
+        .form-control:focus {
+            border-color: var(--primary-color);
+            box-shadow: 0 0 0 3px rgba(111, 66, 193, 0.2);
+        }
+        
+        .btn {
+            display: inline-block;
+            font-weight: 500;
+            text-align: center;
+            vertical-align: middle;
+            cursor: pointer;
+            padding: 0.75rem 1rem;
+            font-size: 1rem;
+            line-height: 1.5;
+            border-radius: 8px;
+            transition: all 0.15s ease-in-out;
+            border: none;
+        }
+        
+        .btn-primary {
+            color: #fff;
+            background: linear-gradient(45deg, var(--primary-color), var(--secondary-color));
+            box-shadow: 0 4px 10px rgba(111, 66, 193, 0.3);
+            position: relative;
+            overflow: hidden;
+        }
+        
+        .btn-primary:hover {
+            transform: translateY(-2px);
+            box-shadow: 0 6px 15px rgba(111, 66, 193, 0.4);
+        }
+        
+        .btn-primary:active {
+            transform: translateY(0);
+        }
+        
+        /* 添加光效效果 */
+        .btn-primary::before {
+            content: '';
+            position: absolute;
+            top: -50%;
+            left: -50%;
+            width: 200%;
+            height: 200%;
+            background: linear-gradient(
+                to bottom right,
+                rgba(255, 255, 255, 0) 0%,
+                rgba(255, 255, 255, 0.1) 50%,
+                rgba(255, 255, 255, 0) 100%
+            );
+            transform: rotate(45deg);
+            animation: btn-shine 3s infinite;
+        }
+        
+        @keyframes btn-shine {
+            0% {
+                left: -50%;
+            }
+            100% {
+                left: 150%;
+            }
+        }
+        
+        .error-message {
+            background-color: rgba(248, 114, 114, 0.2);
+            color: var(--error-color);
+            padding: 0.75rem;
+            border-radius: 6px;
+            margin-bottom: 1.5rem;
+            font-size: 0.9rem;
+            border-left: 3px solid var(--error-color);
+        }
+        
+        .logo {
+            margin-bottom: 1rem;
+            font-size: 3rem;
+            animation: glow 2s infinite alternate;
+        }
+        
+        @keyframes glow {
+            from {
+                text-shadow: 0 0 5px rgba(111, 66, 193, 0.5), 0 0 10px rgba(111, 66, 193, 0.5);
+            }
+            to {
+                text-shadow: 0 0 10px rgba(111, 66, 193, 0.8), 0 0 20px rgba(111, 66, 193, 0.8);
+            }
+        }
+    </style>
+</head>
+<body>
+    <div class="grid-background"></div>
+    <div class="particles">
+        <!-- 粒子元素会由JS生成 -->
+    </div>
+    <div class="web-container">
+        <canvas class="web" id="webCanvas"></canvas>
+    </div>
+    
+    <div class="login-card">
+        <div class="login-header">
+            <div class="logo">🤖</div>
+            <h1>Abacus Chat Proxy</h1>
+            <p>请输入访问密码</p>
+        </div>
+        
+        {% if space_url %}
+        <div class="space-info">
+            api接口为{{ space_url }}/v1，请点击 <a href="{{ space_url }}" target="_blank">{{ space_url }}</a> 来登录并查看使用情况，
+        </div>
+        {% endif %}
+        
+        {% if error %}
+        <div class="error-message" id="error-message">
+            {{ error }}
+        </div>
+        {% endif %}
+        
+        <form class="login-form" method="post" action="/login">
+            <div class="form-group">
+                <label for="password">密码</label>
+                <input type="password" class="form-control" id="password" name="password" placeholder="请输入访问密码" required>
+            </div>
+            
+            <button type="submit" class="btn btn-primary">登录</button>
+        </form>
+    </div>
+    
+    <script>
+        // 创建浮动粒子
+        function createParticles() {
+            const particlesContainer = document.querySelector('.particles');
+            const particleCount = 20;
+            
+            for (let i = 0; i < particleCount; i++) {
+                const particle = document.createElement('div');
+                particle.className = 'particle';
+                
+                // 随机位置和大小
+                const size = Math.random() * 5 + 2;
+                const x = Math.random() * 100;
+                const y = Math.random() * 100;
+                
+                particle.style.width = `${size}px`;
+                particle.style.height = `${size}px`;
+                particle.style.left = `${x}%`;
+                particle.style.top = `${y}%`;
+                
+                // 随机动画延迟
+                particle.style.animationDelay = `${Math.random() * 10}s`;
+                particle.style.animationDuration = `${Math.random() * 10 + 10}s`;
+                
+                // 随机透明度
+                particle.style.opacity = Math.random() * 0.5;
+                
+                particlesContainer.appendChild(particle);
+            }
+        }
+        
+        // 科幻蜘蛛网效果
+        function initWebCanvas() {
+            const canvas = document.getElementById('webCanvas');
+            const ctx = canvas.getContext('2d');
+            let width = window.innerWidth;
+            let height = window.innerHeight;
+            
+            // 设置canvas尺寸
+            canvas.width = width;
+            canvas.height = height;
+            
+            // 节点类
+            class Node {
+                constructor(x, y) {
+                    this.x = x;
+                    this.y = y;
+                    this.vx = (Math.random() - 0.5) * 0.5;
+                    this.vy = (Math.random() - 0.5) * 0.5;
+                    this.radius = Math.random() * 2 + 1;
+                }
+                
+                update() {
+                    if (this.x < 0 || this.x > width) this.vx = -this.vx;
+                    if (this.y < 0 || this.y > height) this.vy = -this.vy;
+                    
+                    this.x += this.vx;
+                    this.y += this.vy;
+                }
+                
+                draw() {
+                    ctx.beginPath();
+                    ctx.arc(this.x, this.y, this.radius, 0, Math.PI * 2);
+                    ctx.fillStyle = 'rgba(111, 66, 193, 0.4)';
+                    ctx.fill();
+                }
+            }
+            
+            // 创建节点
+            const nodeCount = Math.floor(width * height / 15000);
+            const nodes = [];
+            
+            for (let i = 0; i < nodeCount; i++) {
+                nodes.push(new Node(Math.random() * width, Math.random() * height));
+            }
+            
+            // 绘制线条
+            function drawWeb() {
+                ctx.clearRect(0, 0, width, height);
+                
+                // 更新节点
+                nodes.forEach(node => {
+                    node.update();
+                    node.draw();
+                });
+                
+                // 绘制连线
+                for (let i = 0; i < nodes.length; i++) {
+                    for (let j = i + 1; j < nodes.length; j++) {
+                        const dx = nodes[i].x - nodes[j].x;
+                        const dy = nodes[i].y - nodes[j].y;
+                        const distance = Math.sqrt(dx * dx + dy * dy);
+                        
+                        if (distance < 150) {
+                            ctx.beginPath();
+                            ctx.moveTo(nodes[i].x, nodes[i].y);
+                            ctx.lineTo(nodes[j].x, nodes[j].y);
+                            ctx.strokeStyle = `rgba(111, 66, 193, ${0.2 * (1 - distance / 150)})`;
+                            ctx.lineWidth = 0.5;
+                            ctx.stroke();
+                        }
+                    }
+                }
+                
+                requestAnimationFrame(drawWeb);
+            }
+            
+            // 监听窗口大小变化
+            window.addEventListener('resize', () => {
+                width = window.innerWidth;
+                height = window.innerHeight;
+                canvas.width = width;
+                canvas.height = height;
+            });
+            
+            // 开始动画
+            drawWeb();
+        }
+        
+        // 页面加载时初始化效果
+        window.addEventListener('load', () => {
+            createParticles();
+            initWebCanvas();
+        });
+    </script>
+</body>
 </html>