Update app.py

app.py

@@ -87,11 +87,22 @@ def extract_user_content(messages):
 def refresh_models():
     global models
     
-    models["text"] = get_all_models(keys_str[0])
+    # Find the first valid key
+    first_valid_key = None
+    for key_list in key_status.values():
+        if key_list:
+            first_valid_key = key_list[0]
+            break
+
+    if first_valid_key:
+       models["text"] = get_all_models(first_valid_key)
+    else:
+        logging.warning("No valid keys found to fetch models.")
+        models["text"] = []
+
 
     for model_type in ["text"]:
         logging.info(f"所有{model_type}模型列表：{models[model_type]}")
-        logging.info(f"免费{model_type}模型列表：{models[f'free_{model_type}']}")
 
 def load_keys():
     global key_status
@@ -106,7 +117,27 @@ def load_keys():
         logging.warning("环境变量 KEYS 未设置。")
         return
     
-    valid_keys_global = keys_str
+    keys = keys_str.split(",")
+    keys = [key.strip() for key in keys]
+    
+    global valid_keys_global, free_keys_global, unverified_keys_global
+    valid_keys_global = []
+    free_keys_global = []
+    unverified_keys_global = []
+
+
+    with concurrent.futures.ThreadPoolExecutor(max_workers=10) as executor:
+        futures = [executor.submit(process_key, key, "gpt-3.5-turbo") for key in keys]
+        for key, future in zip(keys, futures):
+            status = future.result()
+            key_status[status].append(key)
+            if status == "valid":
+                valid_keys_global.append(key)
+            elif status == "free":
+                free_keys_global.append(key)
+            elif status == "unverified":
+                unverified_keys_global.append(key)
+            logging.info(f"Key {key} status: {status}")
 
 def process_key(key, test_model):
     credit_summary = get_credit_summary(key)
@@ -500,6 +531,56 @@ def handsome_chat_completions():
     except requests.exceptions.RequestException as e:
         logging.error(f"请求转发异常: {e}")
         return jsonify({"error": str(e)}), 500
+    
+def get_credit_summary(api_key):
+    headers = {
+        "Authorization": f"Bearer {api_key}",
+        "Content-Type": "application/json"
+    }
+    try:
+        response = session.get(
+            API_ENDPOINT,
+            headers=headers
+        )
+        response.raise_for_status()
+        return response.json().get("credit_summary")
+    except requests.exceptions.RequestException as e:
+        logging.error(
+            f"获取账户余额失败，"
+            f"API Key：{api_key}，错误信息：{e}"
+        )
+        return None
+    except (KeyError, TypeError) as e:
+        logging.error(
+            f"解析账户余额失败，"
+            f"API Key：{api_key}，错误信息：{e}"
+        )
+        return None
+
+def test_model_availability(api_key, test_model):
+    headers = {
+        "Authorization": f"Bearer {api_key}",
+        "Content-Type": "application/json"
+    }
+    data = {
+        "model": test_model,
+        "messages": [{"role": "user", "content": "hello"}]
+    }
+    try:
+        response = session.post(
+            TEST_MODEL_ENDPOINT,
+            headers=headers,
+            json=data
+        )
+        response.raise_for_status()
+        return True
+    except requests.exceptions.RequestException as e:
+        logging.error(
+            f"模型测试失败，"
+            f"API Key：{api_key}，错误信息：{e}"
+        )
+        return False
+
 
 if __name__ == '__main__':
     logging.info(f"环境变量：{os.environ}")
@@ -514,4 +595,4 @@ if __name__ == '__main__':
     refresh_models()
     logging.info("首次刷新模型列表已手动触发执行")
 
-    app.run(debug=False,host='0.0.0.0',port=int(os.environ.get('PORT', 7860)))
+    app.run(debug=False,host='0.0.0.0',port=int(os.environ.get('PORT', 7860)))