Update app.py

app.py

@@ -24,13 +24,13 @@ model = HfApiModel(
 @tool
 def summarize_text(text: str) -> str:
     """
-    Usa una API de Hugging Face para resumir texto.
+    Resumir texto utilizando la API de Hugging Face.
 
     Args:
-        text (str): Texto que se desea resumir.
+        text (str): El texto que se desea resumir.
 
     Returns:
-        str: Resumen del texto o un mensaje de error si falla la operación.
+        str: El resumen generado o un mensaje de error en caso de fallo.
     """
     API_URL = "https://api-inference.huggingface.co/models/facebook/bart-large-cnn"
     headers = {"Authorization": f"Bearer TU_HUGGINGFACE_API_KEY"}
@@ -49,15 +49,19 @@ def summarize_text(text: str) -> str:
 
 @tool
 def scrape_webpage(url: str, tag: str = "p", class_name: str = None) -> dict:
-    """Extrae contenido de una página web según una etiqueta HTML y clase opcional.
+    """
+    Extrae contenido de una página web según una etiqueta HTML y clase opcional.
 
     Args:
-        url: URL de la página a scrapear.
-        tag: Etiqueta HTML a extraer (por defecto <p>).
-        class_name: Clase CSS opcional para filtrar resultados.
+        url (str): URL de la página a scrapear.
+        tag (str): Etiqueta HTML a extraer (por defecto "p").
+        class_name (str, optional): Clase CSS opcional para filtrar resultados.
 
     Returns:
-        Un diccionario con el contenido extraído.
+        dict: Diccionario con el contenido extraído. Contiene las claves:
+              - "url": La URL consultada.
+              - "scraped_data": Lista de fragmentos de texto extraídos (máximo 20).
+              En caso de error, devuelve una clave "error" con el mensaje correspondiente.
     """
     try:
         headers = {"User-Agent": "Mozilla/5.0"}
@@ -80,16 +84,17 @@ def scrape_webpage(url: str, tag: str = "p", class_name: str = None) -> dict:
     except Exception as e:
         return {"error": f"Error inesperado: {str(e)}"}
 
-
 @tool
 def extract_metadata_from_url(url: str) -> dict:
-    """Extrae todos los metadatos de una página web.
+    """
+    Extrae todos los metadatos de una página web.
 
     Args:
-        url: La URL de la página web a analizar.
+        url (str): La URL de la página web a analizar.
 
     Returns:
-        Un diccionario con los metadatos encontrados.
+        dict: Diccionario con los metadatos encontrados. Si no se encuentran metadatos,
+              devuelve un diccionario con la clave "error" y el mensaje correspondiente.
     """
     try:
         headers = {"User-Agent": "Mozilla/5.0"}
@@ -117,9 +122,10 @@ def save_scraped_data_as_markdown(scraped_data: dict, filename: str = None) -> s
 
     Args:
         scraped_data (dict): Diccionario que contiene:
-            - 'url' (str): La URL de la página de la cual se extrajo el contenido.
-            - 'scraped_data' (list[str]): Lista de fragmentos de texto extraídos.
-        filename (str, optional): Nombre del archivo Markdown de salida. Si no se proporciona, se genera uno usando la fecha y hora actual.
+            - "url" (str): La URL de la página de la cual se extrajo el contenido.
+            - "scraped_data" (list[str]): Lista de fragmentos de texto extraídos.
+        filename (str, optional): Nombre del archivo Markdown de salida. Si no se proporciona,
+                                  se genera uno usando la fecha y hora actual.
 
     Returns:
         str: Mensaje indicando si el archivo se guardó correctamente o si ocurrió algún error.
@@ -151,16 +157,16 @@ def save_scraped_data_as_markdown(scraped_data: dict, filename: str = None) -> s
     except Exception as e:
         return f"Error generating Markdown file: {str(e)}"
 
-
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
-    """Devuelve la hora actual en una zona horaria específica.
+    """
+    Devuelve la hora actual en una zona horaria específica.
 
     Args:
-        timezone: Una cadena que representa una zona horaria válida (ej. 'America/New_York').
+        timezone (str): Una cadena que representa una zona horaria válida (ej. "America/New_York").
 
     Returns:
-        La hora local actual en la zona horaria especificada.
+        str: La hora local actual en la zona horaria especificada o un mensaje de error si falla.
     """
     try:
         tz = pytz.timezone(timezone)
@@ -169,10 +175,9 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error obteniendo la hora para la zona horaria '{timezone}': {str(e)}"
 
-
 final_answer = FinalAnswerTool()
 
-# Import tool from Hub
+# Importar herramienta desde Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
 with open("prompts.yaml", "r") as stream:
@@ -185,6 +190,7 @@ agent = CodeAgent(
         extract_metadata_from_url,
         scrape_webpage,
         save_scraped_data_as_markdown,  # Se añade la nueva herramienta al agente
+        # Puedes añadir también otras herramientas como summarize_text si lo deseas
     ],
     max_steps=6,
     verbosity_level=1,