Upload app.py

app.py

@@ -232,24 +232,10 @@ def handle_api_error(error, attempt):
         switch_api_key()
         return 0, None
 
-    elif hasattr(error, 'prompt_feedback') and error.prompt_feedback.block_reason:
-        block_reason = error.prompt_feedback.block_reason
-        safety_ratings = error.prompt_feedback.safety_ratings
-        if block_reason == generation_types.BlockReason.SAFETY:
-            logger.warning(f"用户输入被阻挡，原因：SAFETY. 详细信息：{safety_ratings}")
-            return 0, None
-        elif block_reason == generation_types.BlockReason.OTHER:
-            logger.warning(f"用户输入被阻挡，原因：OTHER")
-            return 0, None
-        elif block_reason == generation_types.BlockReason.BLOCKLIST:
-            logger.warning(f"用户输入被阻挡，原因：BLOCKLIST")
-            return 0, None
-        elif block_reason == generation_types.BlockReason.PROHIBITED_CONTENT:
-            logger.warning(f"用户输入被阻挡，原因：PROHIBITED_CONTENT")
-            return 0, None
-        else:
-            logger.warning(f"用户输入被阻挡，未知阻挡原因：{block_reason}")
-            return 0, None
+    elif "block_reason" in str(error):
+        logger.error(f"Gemini官方开始发癫，请等待官方恢复")
+        switch_api_key()
+        return 2, None
 
     else:
         logger.error(f"其他错误↙\n {error}")