Update app.py

app.py

@@ -18,6 +18,7 @@ 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):
@@ -521,10 +522,51 @@ 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():
+    accounts_status = get_accounts_status()
+    
+    # 计算可用和不可用账号数量
+    total_accounts = len(accounts_status)
+    valid_accounts = sum(1 for acc in accounts_status if acc["is_valid"])
+    
+    # 计算每个模型的可用账号数量
+    models_availability = {}
+    for model in MODEL_INFO.keys():
+        available_accounts = sum(
+            1 for acc in accounts_status 
+            if acc["is_valid"] and acc["models_status"].get(model, False)
+        )
+        models_availability[model] = available_accounts
+    
     return jsonify({
         "service": "AI Chat Completion Proxy",
+        "status": {
+            "total_accounts": total_accounts,
+            "valid_accounts": valid_accounts,
+            "invalid_accounts": total_accounts - valid_accounts,
+            "accounts_details": accounts_status,
+            "models_availability": models_availability
+        },
         "usage": {
             "endpoint": "/ai/v1/chat/completions",
             "method": "POST",
@@ -782,4 +824,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)
+