Upload 3 files

app.py

@@ -11,13 +11,14 @@ from functools import lru_cache, wraps
 from typing import Dict, Any, Callable, List, Tuple
 import requests
 import tiktoken
-from flask import Flask, Response, jsonify, request, stream_with_context
+from flask import Flask, Response, jsonify, request, stream_with_context, render_template
 from flask_cors import CORS
 from requests.adapters import HTTPAdapter
 from urllib3.util.connection import create_connection
 import urllib3
 from cachetools import TTLCache
 import threading
+from datetime import datetime
 
 # 新增导入
 import register_bot
@@ -55,7 +56,7 @@ if not NOTDIAMOND_IP:
     logger.error("NOTDIAMOND_IP environment variable is not set!")
     raise ValueError("NOTDIAMOND_IP must be set")
 
-# API密钥验证装饰器
+# API钥验证装饰器
 def require_api_key(f):
     @wraps(f)
     def decorated_function(*args, **kwargs):
@@ -118,6 +119,10 @@ class AuthManager:
         self._session: requests.Session = create_custom_session()
         self._logger: logging.Logger = logging.getLogger(__name__)
         self.model_status = {model: True for model in MODEL_INFO.keys()}
+        self.total_requests = 0
+        self.success_requests = 0
+        self.failed_requests = 0
+        self.last_used_time = None
 
     def login(self) -> bool:
         """使用电子邮件和密码进行用户登录,并获取用户信息。"""
@@ -252,6 +257,14 @@ class AuthManager:
     def reset_model_status(self):
         self.model_status = {model: True for model in MODEL_INFO.keys()}
 
+    def record_request(self, success: bool):
+        self.total_requests += 1
+        if success:
+            self.success_requests += 1
+        else:
+            self.failed_requests += 1
+        self.last_used_time = datetime.now()
+
 class MultiAuthManager:
     def __init__(self, credentials):
         self.auth_managers = [AuthManager(email, password) for email, password in credentials]
@@ -477,7 +490,7 @@ def get_auth_credentials():
             if data.get('status') == 'success' and data.get('content'):
                 content = data['content']
                 credentials = []
-                # 分割多个凭据（如果有的话）
+                # 分割多个凭据（如果��的话）
                 for cred in content.split(';'):
                     if '|' in cred:
                         email, password = cred.strip().split('|')
@@ -521,29 +534,69 @@ def before_request():
     else:
         multi_auth_manager = None
 
+def get_accounts_status():
+    """获取所有账号的状态信息"""
+    if not multi_auth_manager:
+        return []
+    
+    accounts_status = []
+    for auth_manager in multi_auth_manager.auth_managers:
+        account_info = {
+            "email": auth_manager._email,
+            "is_valid": auth_manager.is_token_valid(),
+            "token_expiry": datetime.fromtimestamp(auth_manager._token_expiry).strftime('%Y-%m-%d %H:%M:%S'),
+            "models_status": {
+                model: status 
+                for model, status in auth_manager.model_status.items()
+            }
+        }
+        accounts_status.append(account_info)
+    return accounts_status
+
 @app.route('/', methods=['GET'])
 def root():
-    return jsonify({
-        "service": "AI Chat Completion Proxy",
-        "usage": {
-            "endpoint": "/ai/v1/chat/completions",
-            "method": "POST",
-            "headers": {
-                "Authorization": "Bearer YOUR_API_KEY"
-            },
-            "body": {
-                "model": "One of: " + ", ".join(MODEL_INFO.keys()),
-                "messages": [
-                    {"role": "system", "content": "You are a helpful assistant."},
-                    {"role": "user", "content": "Hello, who are you?"}
-                ],
-                "stream": False,
-                "temperature": 0.7
-            }
-        },
-        "availableModels": list(MODEL_INFO.keys()),
-        "note": "API key authentication is required for other endpoints."
-    })
+    accounts_status = get_accounts_status()
+    
+    if request.headers.get('Accept') == 'application/json':
+        return get_json_status(accounts_status)
+    
+    return render_template('monitor.html', **get_template_data(accounts_status))
+
+def get_template_data(accounts_status):
+    total_accounts = len(accounts_status)
+    valid_accounts = sum(1 for acc in accounts_status if acc["is_valid"])
+    
+    accounts_data = []
+    total_requests = 0
+    
+    for auth_manager in multi_auth_manager.auth_managers:
+        success_rate = 0
+        if auth_manager.total_requests > 0:
+            success_rate = (auth_manager.success_requests / auth_manager.total_requests) * 100
+        
+        account_info = {
+            "email": auth_manager._email,
+            "is_valid": auth_manager.is_token_valid(),
+            "total_requests": auth_manager.total_requests,
+            "success_requests": auth_manager.success_requests,
+            "failed_requests": auth_manager.failed_requests,
+            "success_rate": success_rate,
+            "last_used_time": auth_manager.last_used_time.strftime('%m/%d/%Y, %I:%M:%S %p') if auth_manager.last_used_time else "从未使用"
+        }
+        accounts_data.append(account_info)
+        total_requests += auth_manager.total_requests
+    
+    return {
+        "total_accounts": total_accounts,
+        "valid_accounts": valid_accounts,
+        "total_requests": total_requests,
+        "accounts": accounts_data,
+        "last_update": datetime.now().strftime('%Y-%m-%d %H:%M:%S')
+    }
+
+def get_json_status(accounts_status):
+    template_data = get_template_data(accounts_status)
+    return jsonify(template_data)
 
 @app.route('/ai/v1/models', methods=['GET'])
 def proxy_models():
@@ -721,6 +774,9 @@ def make_request(payload, auth_manager, model_id):
                     stream=True
                 ).result()
                 
+                success = response.status_code == 200
+                auth_manager.record_request(success)
+                
                 if response.status_code == 200 and response.headers.get('Content-Type') == 'text/event-stream':
                     return response
                 
@@ -743,6 +799,7 @@ def make_request(payload, auth_manager, model_id):
                 logger.error(f"Request failed with status {response.status_code} for account {auth_manager._email}")
                 
             except Exception as e:
+                auth_manager.record_request(False)
                 logger.error(f"Request attempt {attempt + 1} failed for account {auth_manager._email}: {e}")
                 if attempt < max_retries - 1:
                     time.sleep(retry_delay)
@@ -782,4 +839,5 @@ if __name__ == "__main__":
     health_check_thread.start()
     
     port = int(os.environ.get("PORT", 3000))
-    app.run(debug=False, host='0.0.0.0', port=port, threaded=True)
+    app.run(debug=False, host='0.0.0.0', port=port, threaded=True)
+

requirements.txt

@@ -8,3 +8,4 @@ urllib3==1.26.9
 beautifulsoup4==4.11.1
 Pillow==9.2.0
 lxml==4.9.1
+jinja2>=3.0.0

templates/monitor.html

@@ -0,0 +1,112 @@
+<!DOCTYPE html>
+<html>
+<head>
+    <title>账号状态监控</title>
+    <meta charset="UTF-8">
+    <style>
+        body {
+            font-family: Arial, sans-serif;
+            margin: 20px;
+            background-color: #f5f5f5;
+        }
+        .header {
+            text-align: center;
+            margin-bottom: 20px;
+        }
+        .summary {
+            text-align: center;
+            margin-bottom: 30px;
+            font-size: 1.2em;
+        }
+        .grid {
+            display: grid;
+            grid-template-columns: repeat(auto-fill, minmax(300px, 1fr));
+            gap: 20px;
+            padding: 20px;
+        }
+        .account-card {
+            background: white;
+            border-radius: 8px;
+            padding: 15px;
+            box-shadow: 0 2px 4px rgba(0,0,0,0.1);
+        }
+        .account-card h3 {
+            margin-top: 0;
+            color: #333;
+        }
+        .status-available {
+            color: green;
+        }
+        .status-unavailable {
+            color: red;
+        }
+        .metric {
+            margin: 8px 0;
+        }
+        .refresh-time {
+            text-align: center;
+            margin-top: 20px;
+            color: #666;
+        }
+    </style>
+</head>
+<body>
+    <div class="header">
+        <h1>账号状态监控</h1>
+    </div>
+    <div class="summary">
+        <div>共 <span id="online-count">{{ valid_accounts }}/{{ total_accounts }}</span> 个账号在线</div>
+        <div>请求总数: <span id="total-requests">{{ total_requests }}</span></div>
+    </div>
+    <div class="grid">
+        {% for account in accounts %}
+        <div class="account-card">
+            <h3>账号{{ loop.index }}: {{ account.email|replace(account.email[1:-4], '*'*3) }}</h3>
+            <div class="metric">总请求数: {{ account.total_requests }}</div>
+            <div class="metric">成功请求数: {{ account.success_requests }}</div>
+            <div class="metric">失败请求数: {{ account.failed_requests }}</div>
+            <div class="metric">成功率: {{ "%.2f"|format(account.success_rate) }}%</div>
+            <div class="metric">最后使用时间: {{ account.last_used_time }}</div>
+            <div class="metric">状态: 
+                <span class="status-{{ 'available' if account.is_valid else 'unavailable' }}">
+                    {{ "可用" if account.is_valid else "不可用" }}
+                </span>
+            </div>
+        </div>
+        {% endfor %}
+    </div>
+    <div class="refresh-time">
+        最后更新时间: <span id="last-update">{{ last_update }}</span>
+    </div>
+    <script>
+        function refreshData() {
+            fetch(window.location.href)
+                .then(response => response.json())
+                .then(data => {
+                    document.getElementById('online-count').textContent = 
+                        `${data.valid_accounts}/${data.total_accounts}`;
+                    document.getElementById('total-requests').textContent = data.total_requests;
+                    document.getElementById('last-update').textContent = data.last_update;
+                    // 更新账号卡片
+                    const grid = document.querySelector('.grid');
+                    grid.innerHTML = data.accounts.map((account, index) => `
+                        <div class="account-card">
+                            <h3>账号${index + 1}: ${account.email.replace(/(?<=.).(?=.*@)/g, '*')}</h3>
+                            <div class="metric">总请求数: ${account.total_requests}</div>
+                            <div class="metric">成功请求数: ${account.success_requests}</div>
+                            <div class="metric">失败请求数: ${account.failed_requests}</div>
+                            <div class="metric">成功率: ${account.success_rate.toFixed(2)}%</div>
+                            <div class="metric">最后使用时间: ${account.last_used_time}</div>
+                            <div class="metric">状态: 
+                                <span class="status-${account.is_valid ? 'available' : 'unavailable'}">
+                                    ${account.is_valid ? '可用' : '不可用'}
+                                </span>
+                            </div>
+                        </div>
+                    `).join('');
+                });
+        }
+        setInterval(refreshData, 60000); // 每分钟刷新一次
+    </script>
+</body>
+</html>