Update app.py

app.py

@@ -103,18 +103,6 @@ def get_usd_to_cny_rate():
         logging.error(f"获取美元兑人民币汇率失败，错误信息：{e}")
         return 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 test_model_availability(api_key, model_name):
     """
     测试指定的模型是否可用。
@@ -151,15 +139,9 @@ def refresh_models():
     刷新模型列表和免费模型列表。
     """
     global text_models, free_text_models
-    global embedding_models, free_embedding_models
-    global image_models, free_image_models
 
     text_models = get_all_models(FREE_MODEL_TEST_KEY, "chat")
-    embedding_models = get_all_models(FREE_MODEL_TEST_KEY, "embedding")
-    image_models = get_all_models(FREE_MODEL_TEST_KEY, "text-to-image")
     free_text_models = []
-    free_embedding_models = []
-    free_image_models = []
 
     ban_models_str = os.environ.get("BAN_MODELS")
     ban_models = []
@@ -178,8 +160,6 @@ def refresh_models():
             ban_models = []
     
     text_models = [model for model in text_models if model not in ban_models]
-    embedding_models = [model for model in embedding_models if model not in ban_models]
-    image_models = [model for model in image_models if model not in ban_models]
 
     with concurrent.futures.ThreadPoolExecutor(
         max_workers=1000
@@ -200,84 +180,8 @@ def refresh_models():
             except Exception as exc:
                 logging.error(f"模型 {model} 测试生成异常: {exc}")
 
-    with concurrent.futures.ThreadPoolExecutor(
-        max_workers=1000
-    ) as executor:
-        future_to_model = {
-            executor.submit(
-                test_embedding_model_availability,
-                FREE_MODEL_TEST_KEY, model
-            ): model for model in embedding_models
-        }
-        for future in concurrent.futures.as_completed(future_to_model):
-            model = future_to_model[future]
-            try:
-                is_free = future.result()
-                if is_free:
-                    free_embedding_models.append(model)
-            except Exception as exc:
-                logging.error(f"模型 {model} 测试生成异常: {exc}")
-                
-    with concurrent.futures.ThreadPoolExecutor(
-        max_workers=1000
-    ) as executor:
-        future_to_model = {
-            executor.submit(
-                test_image_model_availability,
-                FREE_MODEL_TEST_KEY, model
-            ): model for model in image_models
-        }
-        for future in concurrent.futures.as_completed(future_to_model):
-            model = future_to_model[future]
-            try:
-                is_free = future.result()
-                if is_free:
-                    free_image_models.append(model)
-            except Exception as exc:
-                logging.error(f"模型 {model} 测试生成异常: {exc}")
-
     logging.info(f"所有文本模型列表：{text_models}")
     logging.info(f"免费文本模型列表：{free_text_models}")
-    logging.info(f"所有向量模型列表：{embedding_models}")
-    logging.info(f"免费向量模型列表：{free_embedding_models}")
-    logging.info(f"所有生图模型列表：{image_models}")
-    logging.info(f"免费生图模型列表：{free_image_models}")
-
-def test_embedding_model_availability(api_key, model_name):
-    """
-    测试指定的向量模型是否可用。
-    """
-    headers = {
-        "Authorization": f"Bearer {api_key}",
-        "Content-Type": "application/json"
-    }
-    try:
-        response = requests.post(
-            EMBEDDINGS_ENDPOINT,
-            headers=headers,
-            json={
-                "model": model_name,
-                "input": ["hi"],
-            },
-            timeout=10
-        )
-        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 load_keys():
     """
@@ -310,8 +214,6 @@ def load_keys():
             }
 
             invalid_keys = []
-            free_keys = []
-            unverified_keys = []
             valid_keys = []
 
             for future in concurrent.futures.as_completed(
@@ -322,25 +224,16 @@ def load_keys():
                     key_type = future.result()
                     if key_type == "invalid":
                         invalid_keys.append(key)
-                    elif key_type == "free":
-                        free_keys.append(key)
-                    elif key_type == "unverified":
-                        unverified_keys.append(key)
                     elif key_type == "valid":
                         valid_keys.append(key)
                 except Exception as exc:
                     logging.error(f"处理 KEY {key} 生成异常: {exc}")
 
         logging.info(f"无效 KEY：{invalid_keys}")
-        logging.info(f"免费 KEY：{free_keys}")
-        logging.info(f"未实名 KEY：{unverified_keys}")
         logging.info(f"有效 KEY：{valid_keys}")
 
-        global invalid_keys_global, free_keys_global
-        global unverified_keys_global, valid_keys_global
+        global invalid_keys_global, valid_keys_global
         invalid_keys_global = invalid_keys
-        free_keys_global = free_keys
-        unverified_keys_global = unverified_keys
         valid_keys_global = valid_keys
 
     else:
@@ -356,12 +249,9 @@ def process_key(key, test_model):
     else:
         total_balance = credit_summary.get("total_balance", 0)
         if total_balance <= 0:
-            return "free"
+            return "invalid"
         else:
-            if test_model_availability(key, test_model):
-                return "valid"
-            else:
-                return "unverified"
+            return "valid"
 
 def get_all_models(api_key, sub_type):
     """