Update app.py

app.py

@@ -546,195 +546,7 @@ def check_tokens():
 
     return jsonify(results)
 
-import os
-import time
-import logging
-import requests
-import json
-import uuid
-from flask import Flask, request, jsonify
-
-API_ENDPOINT = "https://api.siliconflow.cn/v1/user/info"
-TEST_MODEL_ENDPOINT = "https://api.siliconflow.cn/v1/chat/completions"
-EMBEDDINGS_ENDPOINT = "https://api.siliconflow.cn/v1/embeddings"
-
-text_models = []
-free_text_models = []
-embedding_models = []
-free_embedding_models = []
-image_models = []
-free_image_models = []
-
-invalid_keys_global = []
-free_keys_global = []
-unverified_keys_global = []
-valid_keys_global = []
-model_key_indices = {}
-
-request_timestamps = []
-token_counts = []
-data_lock = None
-
-FREE_MODEL_TEST_KEY = (
-    "sk-bmjbjzleaqfgtqfzmcnsbagxrlohriadnxqrzfocbizaxukw"
-)
-
-FREE_IMAGE_LIST = [
-    "stabilityai/stable-diffusion-3-5-large",
-    "black-forest-labs/FLUX.1-schnell",
-    "stabilityai/stable-diffusion-3-medium",
-    "stabilityai/stable-diffusion-xl-base-1.0",
-    "stabilityai/stable-diffusion-2-1"
-]
-
-def get_credit_summary(api_key):
-    """
-    使用 API 密钥获取额度信息。
-    """
-    headers = {
-        "Authorization": f"Bearer {api_key}",
-        "Content-Type": "application/json"
-    }
-    try:
-        response = requests.get(API_ENDPOINT, headers=headers)
-        response.raise_for_status()
-        data = response.json().get("data", {})
-        total_balance = data.get("totalBalance", 0)
-        return {"total_balance": float(total_balance)}
-    except requests.exceptions.RequestException as e:
-        logging.error(f"获取额度信息失败，API Key：{api_key}，错误信息：{e}")
-        return None
-
-def test_model_availability(api_key, model_name):
-    """
-    测试指定的模型是否可用。
-    """
-    headers = {
-        "Authorization": f"Bearer {api_key}",
-        "Content-Type": "application/json"
-    }
-    try:
-        response = requests.post(
-            TEST_MODEL_ENDPOINT,
-            headers=headers,
-            json={
-                "model": model_name,
-                "messages": [{"role": "user", "content": "hi"}],
-                "max_tokens": 5,
-                "stream": False
-            },
-            timeout=5
-        )
-        if response.status_code == 429 or response.status_code == 200:
-            return True
-        else:
-            return False
-    except requests.exceptions.RequestException as e:
-        logging.error(
-            f"测试模型 {model_name} 可用性失败，"
-            f"API Key：{api_key}，错误信息：{e}"
-        )
-        return False
-
-def test_image_model_availability(api_key, model_name):
-    """
-    测试指定的图像模型是否在 FREE_IMAGE_LIST 中。
-    如果在列表中，返回 True，否则返回 False。
-    """
-    return model_name in FREE_IMAGE_LIST
-
-def determine_request_type(model_name, model_list, free_model_list):
-    """
-    根据用户请求的模型判断请求类型。
-    """
-    if model_name in free_model_list:
-        return "free"
-    elif model_name in model_list:
-        return "paid"
-    else:
-        return "unknown"
-
-def select_key(request_type, model_name):
-    """
-    根据请求类型和模型名称选择合适的 KEY，
-    并实现轮询和重试机制。
-    """
-    if request_type == "free":
-        available_keys = (
-            free_keys_global +
-            unverified_keys_global +
-            valid_keys_global
-        )
-    elif request_type == "paid":
-        available_keys = unverified_keys_global + valid_keys_global
-    else:
-        available_keys = (
-            free_keys_global +
-            unverified_keys_global +
-            valid_keys_global
-        )
-
-    if not available_keys:
-        return None
-
-    current_index = model_key_indices.get(model_name, 0)
-
-    for _ in range(len(available_keys)):
-        key = available_keys[current_index % len(available_keys)]
-        current_index += 1
-
-        if key_is_valid(key, request_type):
-            model_key_indices[model_name] = current_index
-            return key
-        else:
-            logging.warning(
-                f"KEY {key} 无效或达到限制，尝试下一个 KEY"
-            )
-
-    model_key_indices[model_name] = 0
-    return None
-
-def key_is_valid(key, request_type):
-    """
-    检查 KEY 是否有效，
-    根据不同的请求类型进行不同的检查。
-    """
-    if request_type == "invalid":
-        return False
-
-    credit_summary = get_credit_summary(key)
-    if credit_summary is None:
-        return False
-
-    total_balance = credit_summary.get("total_balance", 0)
-
-    if request_type == "free":
-        return True
-    elif request_type == "paid" or request_type == "unverified":
-        return total_balance > 0
-    else:
-        return False
-
-def check_authorization(request, authorization_key):
-    """
-    检查请求头中的 Authorization 字段
-    是否匹配环境变量 AUTHORIZATION_KEY。
-    """
-    if not authorization_key:
-        logging.warning("环境变量 AUTHORIZATION_KEY 未设置，请设置后重试。")
-        return False
-
-    auth_header = request.headers.get('Authorization')
-    if not auth_header:
-        logging.warning("请求头中缺少 Authorization 字段。")
-        return False
-
-    if auth_header != f"Bearer {authorization_key}":
-        logging.warning(f"无效的 Authorization 密钥：{auth_header}")
-        return False
-
-    return True
-
+@app.route('/handsome/v1/chat/completions', methods=['POST'])
 def handsome_chat_completions(request, authorization_key):
     if not check_authorization(request, authorization_key):
         return jsonify({"error": "Unauthorized"}), 401