Upload 4 files

app.py

@@ -84,44 +84,33 @@ 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:
-            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 str(self._data)
+        except (KeyError, IndexError):
             return ""
 
     def _extract_finish_reason(self) -> Optional[str]:
         try:
-            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 self._data['candidates'][0].get('finishReason')
+        except (KeyError, IndexError):
             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'))