Update app.py

app.py

@@ -19,6 +19,12 @@ app = Flask(__name__)
 all_models = []
 free_models = []
 
+# 全局的 key 列表
+invalid_keys_global = []
+free_keys_global = []
+unverified_keys_global = []
+valid_keys_global = []
+
 def get_credit_summary(api_key):
     """
     使用 API 密钥获取额度信息。
@@ -220,6 +226,17 @@ def check_authorization(request):
 
     return True
 
+def get_total_balance():
+    """
+    获取所有大于等于0的key的额度的和。
+    """
+    total_balance = 0
+    for key in free_keys_global + unverified_keys_global + valid_keys_global:
+        credit_summary = get_credit_summary(key)
+        if credit_summary is not None:
+            total_balance += credit_summary.get("total_balance", 0)
+    return total_balance
+
 # 创建一个后台调度器
 scheduler = BackgroundScheduler()
 
@@ -306,6 +323,27 @@ def handsome_chat_completions():
     except requests.exceptions.RequestException as e:
         return jsonify({"error": str(e)}), 500
 
+@app.route('/handsome/v1/models', methods=['GET'])
+def handsome_models():
+    """
+    处理 /handsome/v1/models 路由的请求，添加鉴权，返回所有模型列表。
+    """
+    if not check_authorization(request):
+        return jsonify({"error": "Unauthorized"}), 401
+
+    return jsonify({"data": [{"id": model} for model in all_models]})
+
+@app.route('/handsome/v1/balance', methods=['GET'])
+def handsome_balance():
+    """
+    处理 /handsome/v1/balance 路由的请求，添加鉴权，返回所有可用KEY的额度总和。
+    """
+    if not check_authorization(request):
+        return jsonify({"error": "Unauthorized"}), 401
+
+    total_balance = get_total_balance()
+    return jsonify({"total_balance": total_balance})
+
 if __name__ == '__main__':
     # 打印所有环境变量，方便调试
     logging.info(f"环境变量：{os.environ}")