Update app.py

app.py

@@ -33,8 +33,22 @@ class PDOKLocationSearchInput(BaseModel):
         else:
             return ""
 
+class PDOKLocationInfo(BaseModel):
+    type: str
+    weergavenaam: str
+    straatnaam: str
+    huisnummer: int
+    huisletter: Optional[str]
+    huisnummertoevoeging: Optional[str]
+    postcode: str
+    woonplaatsnaam: str
+    gemeentenaam: str
+    provincienaam: str
+    centroide_ll: str
+    centroide_rd: str
+
 @tool
-def pdok_location_info(postal_code: Optional[str] = None, house_number: Optional[str] = None, street_name: Optional[str] = None, city: Optional[str] = None) -> str:
+def pdok_location_info(postal_code: Optional[str] = None, house_number: Optional[str] = None, street_name: Optional[str] = None, city: Optional[str] = None) -> Optional[PDOKLocationInfo]:
     """Provides information about a Dutch address or postal code.
 
     Args:
@@ -79,45 +93,23 @@ def pdok_location_info(postal_code: Optional[str] = None, house_number: Optional
 
         docs = data.get("response", {}).get("docs", [])
         if not docs:
-            return "No results found for the given query."
+        return None
 
         first_result = docs[0]
-
-        # Format the output in a more readable and comprehensive way
-        output_lines = []
-        output_lines.append(f"Information about: {first_result.get('weergavenaam', 'N/A')}")
-        output_lines.append("-" * 30)
-
-        # Key information, handle potential missing values gracefully
-        output_lines.append(f"  Type: {first_result.get('type', 'N/A')}")
-        output_lines.append(f"  Street: {first_result.get('straatnaam', 'N/A')}")
-        output_lines.append(f"  House Number: {first_result.get('huisnummer', 'N/A')}")
-        if 'huisletter' in first_result:
-            output_lines.append(f"  House Letter: {first_result.get('huisletter', 'N/A')}")
-        if 'huisnummertoevoeging' in first_result:
-             output_lines.append(f"  House Number Addition: {first_result.get('huisnummertoevoeging', 'N/A')}")
-        output_lines.append(f"  Postal Code: {first_result.get('postcode', 'N/A')}")
-        output_lines.append(f"  City: {first_result.get('woonplaatsnaam', 'N/A')}")
-        output_lines.append(f"  Municipality: {first_result.get('gemeentenaam', 'N/A')}")
-        output_lines.append(f"  Province: {first_result.get('provincienaam', 'N/A')}")
-        output_lines.append(f"  Coordinates (Lat/Lon): {first_result.get('centroide_ll', 'N/A')}")
-        output_lines.append(f"  Coordinates (RD): {first_result.get('centroide_rd', 'N/A')}")
-
-        # Add more fields if they are present and relevant
-        if 'buurtnaam' in first_result:
-            output_lines.append(f"  Neighborhood: {first_result.get('buurtnaam', 'N/A')}")
-        if 'wijknaam' in first_result:
-            output_lines.append(f"  District: {first_result.get('wijknaam', 'N/A')}")
-        if 'waterschapsnaam' in first_result:
-            output_lines.append(f"  Water Authority: {first_result.get('waterschapsnaam', 'N/A')}")
-
-
-        return "\n".join(output_lines)
-
-    except requests.exceptions.RequestException as e:
-        return f"Error during API request: {e}"
-    except (ValueError, KeyError) as e:
-        return f"Error processing API response: {e}"
+        location_info = PDOKLocationInfo(**first_result)
+    
+        # Format the output in a more user-friendly way
+        print(f"Information about: {location_info.weergavenaam}")
+        print("-" * 30)
+        for field, value in location_info.dict().items():
+            print(f"  {field.capitalize()}: {value}")
+    
+        # Add more features as needed, such as geocoding, reverse geocoding, and caching
+        return location_info
+        except requests.exceptions.RequestException as e:
+            return f"Error during API request: {e}"
+        except (ValueError, KeyError) as e:
+            return f"Error processing API response: {e}"
 
 @tool
 def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type