Update app.py

app.py

@@ -1,10 +1,6 @@
 import requests
 from fastapi import FastAPI
 from pydantic import BaseModel
-from googlesearch import search
-import uvicorn
-
-# Inicializa la aplicación FastAPI
 
 app = FastAPI()
 
@@ -12,41 +8,54 @@ app = FastAPI()
 class ConsultaRequest(BaseModel):
     consulta: str
 
+# --- Configuración de Google CSE ---
+API_KEY = "AIzaSyAej8rDfcGTBKos8UtOSJ8KYyfwK9QLa1k"
+CX = "d453e209e2d3042bd"
+
 def busqueda_google(consulta: str):
     """
-    Busca en Google y devuelve los dos primeros resultados.
+    Busca en Google CSE y devuelve los dos primeros resultados.
     """
-    resultados = []
     try:
-        enlaces_encontrados = search(consulta, num_results=2, lang='es')
-
-        for enlace in enlaces_encontrados:
-            resultados.append({
-                "texto": "Resultado encontrado en Google",
-                "url": enlace
-            })
-
-        if resultados:
-            return {
-                "resultado_1": resultados[0],
-                "resultado_2": resultados[1] if len(resultados) > 1 else None
-            }
-        else:
+        url = "https://www.googleapis.com/customsearch/v1"
+        params = {
+            "q": consulta,
+            "key": API_KEY,
+            "cx": CX,
+            "hl": "es"
+        }
+        response = requests.get(url, params=params)
+        data = response.json()
+
+        if "items" not in data:
             return None
+
+        items = data["items"][:2]
+
+        resultados = {
+            f"resultado_{i+1}": {
+                "texto": it.get("title"),
+                "url": it.get("link")
+            }
+            for i, it in enumerate(items)
+        }
+
+        return resultados if resultados else None
     except Exception as e:
-        print(f"Error en la búsqueda de Google: {e}")
+        print(f"Error en la búsqueda de Google CSE: {e}")
         return None
 
-# --- API FastAPI ---
 @app.post("/leer")
 async def buscar(request_body: ConsultaRequest):
+    """
+    Endpoint principal para la búsqueda.
+    """
     consulta = request_body.consulta
     if not consulta:
-        return {"error": "Falta la consulta"}
+        return {"error": "Falta la consulta"}, 400
 
     print(f"Consulta recibida: {consulta}")
 
-    # Buscar en Google
     resultado = busqueda_google(consulta)
 
     if resultado:
@@ -55,7 +64,8 @@ async def buscar(request_body: ConsultaRequest):
             "resultados": resultado
         }
 
-    return {"error": "No se encontraron resultados"}
+    return {"error": "No se encontraron resultados"}, 404
+
 
 # Para Hugging Face Spaces
 if __name__ == "__main__":