Upload 4 files

app.py

@@ -1,6 +1,6 @@
 from flask import Flask, request, jsonify, Response, stream_with_context, render_template_string
 from google.generativeai.types import generation_types
-from google.api_core.exceptions import InvalidArgument, ResourceExhausted, ServiceUnavailable, InternalServerError, Aborted
+from google.api_core.exceptions import InvalidArgument, ResourceExhausted, Aborted, InternalServerError, ServiceUnavailable, PermissionDenied
 import google.generativeai as genai
 import json
 import os
@@ -203,6 +203,12 @@ def handle_api_error(error, attempt):
         switch_api_key()
         time.sleep(delay)
         return 0, None
+    
+    elif isinstance(error, PermissionDenied):
+        logger.error(f"{current_api_key[:11]} → 403 权限被拒绝，API KEY 已被禁用")
+        key_manager.blacklist_key(current_api_key)
+        switch_api_key()
+        return 0, None
 
     elif isinstance(error, generation_types.StopCandidateException):
         logger.warning(f"AI输出内容被Gemini官方阻挡，代理没有得到有效回复")
@@ -210,7 +216,7 @@ def handle_api_error(error, attempt):
         return 0, None
 
     else:
-        logger.error(f"该模型暂时不可用↙\n {error}")
+        logger.error(f"其他错误↙\n {error}")
         return 2, None
 
 @app.route('/hf/v1/chat/completions', methods=['POST'])