Update app.py

app.py

@@ -9,11 +9,14 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def buscar_datos_gob(term: str, page_size: int, page: int, sort: str) -> dict:
-    """A tool that searches for datasets in the API of datos.gob.es.
+def buscar_datos_gob(term: str, category: str = None, format: str = "json", page_size: int = 10, page: int = 0, sort: str = "title") -> dict:
+    """
+    A tool that searches for datasets in the datos.gob.es API.
 
     Args:
         term: The search term to filter the datasets.
+        category: The category/topic to filter datasets (e.g., "hacienda").
+        format: The format of the datasets (json, xml, csv, rdf). Default is json.
         page_size: The number of results per page (maximum: 50).
         page: The page number for pagination of results.
         sort: The field by which to sort the results.
@@ -21,18 +24,25 @@ def buscar_datos_gob(term: str, page_size: int, page: int, sort: str) -> dict:
     Returns:
         A dictionary containing the search results from the datos.gob.es API.
     """
+    import requests
+
     base_url = "https://datos.gob.es/apidata/catalog/dataset"
+    
+    if category:
+        base_url += f"/theme/{category}"
+
     params = {
         "q": term,
         "_pageSize": page_size,
         "_page": page,
         "_sort": sort
     }
+
     headers = {
-        "Accept": "application/json",
+        "Accept": f"application/{format}",
         "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/110.0.0.0 Safari/537.36"
     }
-    
+
     try:
         response = requests.get(base_url, params=params, headers=headers)
         response.raise_for_status()