Update app.py

app.py

@@ -9,45 +9,54 @@ 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, category: str = None, format: str = "json", page_size: int = 10, page: int = 0, sort: str = "title") -> dict:
+def buscar_datos_gob(term: str, page_size: int = 10, page: int = 0) -> dict:
     """
-    A tool that searches for datasets in the datos.gob.es API.
+    Search 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.
+        term: The search term for datasets.
+        page_size: Number of results per page (max 50).
+        page: The page number for pagination.
 
     Returns:
-        A dictionary containing the search results from the datos.gob.es API.
+        A dictionary with the relevant datasets, showing the title, description, publisher, and access URL.
     """
     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
+        "_page": page
     }
-
     headers = {
-        "Accept": f"application/{format}"
+        "Accept": "application/json",
+        "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()
-        return response.json()
+        data = response.json()
+
+        # Extraer los datasets de la respuesta
+        datasets = []
+        for item in data.get("result", {}).get("items", []):
+            dataset_info = {
+                "title": item.get("title", "No Title"),
+                "description": item.get("description", {}).get("text", "No Description"),
+                "publisher": item.get("publisher", "Unknown Publisher"),
+                "accessURL": item.get("distribution", {}).get("accessURL", "No URL Available"),
+                "modified": item.get("modified", "No Date Available"),
+                "license": item.get("license", "No License Available")
+            }
+            datasets.append(dataset_info)
+
+        return {"datasets": datasets}
+
     except requests.exceptions.RequestException as e:
-        return {"error": f"Error in the request: {str(e)}"}
+        return {"error": f"Request Error: {str(e)}"}
+
 
 
 final_answer = FinalAnswerTool()