Upload 4 files

app.py

@@ -234,9 +234,10 @@ def handle_api_error(error, attempt):
 
     elif isinstance(error, generation_types.BlockedPromptException):
         try:
-            reason = error.args[0].split(':', 1)[0].strip()
-            block_reason_str = error.args[0].split(':', 1)[1].strip()
-            if reason == "block_reason": 
+            full_reason_str = error.args[0]  
+            if full_reason_str.startswith("block_reason:"):
+                block_reason_str = full_reason_str.split(":", 1)[1].strip()
+
                 if block_reason_str == "SAFETY":
                     logger.warning(f"提示因安全原因被阻止")
                     return 2, None
@@ -250,11 +251,12 @@ def handle_api_error(error, attempt):
                     logger.warning(f"提示因未知原因被阻止")
                     return 2, None
                 else:
-                    logger.warning(f"提示被阻止，原因未知: {reason}")
+                    logger.warning(f"提示被阻止，原因未知: {block_reason_str}")
                     return 2, None
             else:
-                logger.warning(f"提示被阻止，原因未知: {reason}")
+                logger.warning(f"提示被阻止，原因未知: {full_reason_str}")
                 return 2, None
+
         except (IndexError, AttributeError) as e:
             logger.error(f"Error accessing block_reason: {e}")
             logger.error(f"BlockedPromptException: {error}")