Update app.py

app.py

@@ -13,10 +13,15 @@ from langchain_core.language_models import LLM
 from typing import Optional, List
 import requests
 
+import os
+import requests
+from typing import Optional, List
+
 class GeminiLLM(LLM):
     """Wrapper para usar Google Gemini como un LLM de LangChain."""
 
     api_key: str = os.getenv("GEMINI")
+    fallback_api_key: str = os.getenv("GEMINI_API_KEY")
     model_name: str = "gemini-2.0-flash"
     temperature: float = 0.1
 
@@ -24,15 +29,11 @@ class GeminiLLM(LLM):
     def _llm_type(self) -> str:
         return "google-gemini-llm"
 
-    def _call(self, prompt: str, stop: Optional[List[str]] = None) -> str:
-        """Envía el prompt a la API de Gemini y devuelve la respuesta."""
-        if not self.api_key:
-            raise ValueError("Debes proporcionar una API Key válida de Gemini.")
-
+    def _make_request(self, api_key: str, prompt: str) -> requests.Response:
         url = f"https://generativelanguage.googleapis.com/v1beta/models/{self.model_name}:generateContent"
         headers = {
             "Content-Type": "application/json",
-            "X-goog-api-key": self.api_key
+            "X-goog-api-key": api_key
         }
         data = {
             "contents": [
@@ -48,9 +49,24 @@ class GeminiLLM(LLM):
                 "temperature": self.temperature
             }
         }
+        return requests.post(url, headers=headers, json=data)
+
+    def _call(self, prompt: str, stop: Optional[List[str]] = None) -> str:
+        """Envía el prompt a la API de Gemini y devuelve la respuesta.
+           Si la cuota se supera, intenta con la API key alternativa."""
+
+        if not self.api_key:
+            raise ValueError("Debes proporcionar una API Key válida de Gemini.")
+
+        response = self._make_request(self.api_key, prompt)
 
+        # Si el error es por cuota y hay fallback API key definida, intentar con la fallback
+        if response.status_code == 403 and "quota" in response.text.lower():
+            if self.fallback_api_key:
+                response = self._make_request(self.fallback_api_key, prompt)
+            else:
+                return f"Error {response.status_code}: {response.text} (no hay API key alternativa)"
 
-        response = requests.post(url, headers=headers, json=data)
         if response.status_code == 200:
             result = response.json()
             return result["candidates"][0]["content"]["parts"][0]["text"]