Upload 4 files

app.py

@@ -13,10 +13,8 @@ import time
 import requests
 from collections import deque
 import random
-import codecs
-import json
 from dataclasses import dataclass
-from typing import Optional, Iterator, Dict, Any
+from typing import Optional, Dict, Any
 
 app = Flask(__name__)
 
@@ -86,33 +84,44 @@ class GeneratedText:
     text: str
     finish_reason: Optional[str] = None
 
-
 class ResponseWrapper:
     def __init__(self, data: Dict[Any, Any]):
         self._data = data
+        logger.debug("Initializing ResponseWrapper with data: %s", self._data)
         self._text = self._extract_text()
         self._finish_reason = self._extract_finish_reason()
+        logger.debug("Text extracted: %s", self._text)
+        logger.debug("Finish reason extracted: %s", self._finish_reason)
 
     def _extract_text(self) -> str:
         try:
-            return self._data['candidates'][0]['content']['parts'][0]['text']
-        except (KeyError, IndexError):
+            text = self._data['candidates'][0]['content']['parts'][0]['text']
+            logger.debug("Successfully extracted text: %s", text)
+            return text
+        except (KeyError, IndexError) as e:
+            logger.error("Failed to extract text: %s", e)
             return ""
 
     def _extract_finish_reason(self) -> Optional[str]:
         try:
-            return self._data['candidates'][0].get('finishReason')
-        except (KeyError, IndexError):
+            finish_reason = self._data['candidates'][0].get('finishReason')
+            logger.debug("Successfully extracted finish reason: %s", finish_reason)
+            return finish_reason
+        except (KeyError, IndexError) as e:
+            logger.error("Failed to extract finish reason: %s", e)
             return None
 
     @property
     def text(self) -> str:
+        logger.debug("Accessing text property: %s", self._text)
         return self._text
 
     @property
     def finish_reason(self) -> Optional[str]:
+        logger.debug("Accessing finish_reason property: %s", self._finish_reason)
         return self._finish_reason
 
+
 class APIKeyManager:
     def __init__(self):
         self.api_keys = re.findall(r"AIzaSy[a-zA-Z0-9_-]{33}", os.environ.get('KeyArray'))
@@ -485,7 +494,6 @@ def chat_completions():
     else:
         try:
             text_content = response.text
-            logger.info(response)
         except (AttributeError, IndexError, TypeError, ValueError) as e:
             if "response.candidates" in str(e) or "response.text" in str(e):
                 logger.error(f"用户输入被AI安全过滤器阻止")