Update agent.py

agent.py

@@ -1,3 +1,4 @@
+import os
 import requests
 import urllib.parse
 from bs4 import BeautifulSoup
@@ -6,9 +7,8 @@ class DuckDuckGoAgent:
     def __init__(self):
         print("DuckDuckGoAgent initialized.")
         self.headers = {
-            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64)"
+            "User-Agent": "Mozilla/5.0"
         }
-        self.gemini_api_key = None  # Opcional: pon aquí tu clave si usas Gemini
 
     def get_duckduckgo_answer(self, query: str) -> str:
         search_query = urllib.parse.quote(query)
@@ -18,14 +18,16 @@ class DuckDuckGoAgent:
             response = requests.get(url, timeout=10)
             if response.status_code == 200:
                 data = response.json()
-                if data.get('AbstractText'):
+                if 'AbstractText' in data and data['AbstractText']:
                     return data['AbstractText'][:200]
                 else:
-                    return self.scrape_duckduckgo(query)  # Fallback
+                    print("No abstract found, falling back to scraping.")
+                    return self.scrape_duckduckgo(query)
             else:
+                print(f"DuckDuckGo API failed: {response.status_code}")
                 return self.scrape_duckduckgo(query)
         except Exception as e:
-            print(f"Error in API: {e}")
+            print(f"Error contacting DuckDuckGo API: {e}")
             return self.scrape_duckduckgo(query)
 
     def scrape_duckduckgo(self, query: str) -> str:
@@ -43,33 +45,49 @@ class DuckDuckGoAgent:
                 text = s.get_text().strip()
                 if text:
                     return text[:200]
-            return "No se encontró una respuesta adecuada en DuckDuckGo."
+            print("No useful snippets found, using Hugging Face LLM.")
+            return self.call_huggingface_llm(query)
         except Exception as e:
             print(f"Scraping error: {e}")
-            return self.call_gemini_backup(query)
+            return self.call_huggingface_llm(query)
+
+    def call_huggingface_llm(self, prompt: str) -> str:
+        hf_api_key = os.getenv("HF_API_TOKEN")
+        model = "mistralai/Mistral-7B-Instruct-v0.1"
+
+        if not hf_api_key:
+            return "Error: Hugging Face API Token no configurado."
+
+        url = f"https://api-inference.huggingface.co/models/{model}"
+        headers = {
+            "Authorization": f"Bearer {hf_api_key}",
+            "Content-Type": "application/json"
+        }
+
+        payload = {
+            "inputs": prompt,
+            "parameters": {
+                "max_new_tokens": 200,
+                "temperature": 0.7
+            }
+        }
 
-    def call_gemini_backup(self, prompt: str) -> str:
-        if not self.gemini_api_key:
-            return "Error al contactar con DuckDuckGo."
-        
         try:
-            response = requests.post(
-                "https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:generateContent",
-                headers={"Content-Type": "application/json"},
-                params={"key": self.gemini_api_key},
-                json={"contents": [{"parts": [{"text": prompt}]}]},
-                timeout=15
-            )
-            if response.status_code == 200:
-                data = response.json()
-                text = data["candidates"][0]["content"]["parts"][0]["text"]
-                return text.strip()[:200]
+            response = requests.post(url, headers=headers, json=payload, timeout=30)
+            response.raise_for_status()
+            output = response.json()
+            if isinstance(output, list) and "generated_text" in output[0]:
+                return output[0]["generated_text"].strip()[:200]
+            elif isinstance(output, dict) and "error" in output:
+                return f"HF LLM error: {output['error']}"
             else:
-                return "Error con Gemini 2.0."
+                return "No se pudo obtener respuesta del LLM."
         except Exception as e:
-            print(f"Gemini fallback error: {e}")
-            return "Error al contactar con DuckDuckGo."
+            print(f"Error en Hugging Face LLM: {e}")
+            return "Error al contactar con el modelo Hugging Face."
 
     def __call__(self, question: str) -> str:
-        print(f"Agent received question: {question[:50]}...")
-        return self.get_duckduckgo_answer(question)
+        print(f"Agent received question (first 50 chars): {question[:50]}...")
+        answer = self.get_duckduckgo_answer(question)
+        print(f"Agent returning answer: {answer}")
+        return answer