Update app.py

app.py

@@ -6,7 +6,6 @@ import yaml
 from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
-
 from typing import Optional, Dict, Any, List
 import requests
 from pydantic import BaseModel, Field, validator
@@ -21,7 +20,7 @@ class PDOKLocationSearchInput(BaseModel):
     @validator("postal_code")
     def validate_postal_code(cls, v):
         if v is not None and (len(v) != 7 or not v[0:4].isdigit() or v[4] != " " or not v[5:7].isalpha()):
-            raise ValueError("Invalid postal code format.  It must be '1234 AA'.")
+            raise ValueError("Invalid postal code format. It must be '1234 AA'.")
         return v
 
     def construct_query(self) -> str:
@@ -58,8 +57,9 @@ def pdok_location_info(postal_code: Optional[str] = None, house_number: Optional
         city: City name.
 
     Returns:
-        A string containing relevant information about the location, or an error message.
+        A PDOKLocationInfo object containing the location information, or None if no results are found.
     """
+
     base_url = "https://api.pdok.nl/bzk/locatieserver/search/v3_1/free"
     headers = {"accept": "application/json"}
 
@@ -67,7 +67,7 @@ def pdok_location_info(postal_code: Optional[str] = None, house_number: Optional
     query_string = input_data.construct_query()
 
     if not query_string:
-        return "Please provide either a postal code and house number, or a street name, city, and house number."
+        return None
 
     params = {
         "q": query_string,
@@ -97,16 +97,22 @@ def pdok_location_info(postal_code: Optional[str] = None, house_number: Optional
 
         first_result = docs[0]
         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:
+        print(f"Error during API request: {e}")
+        return None
+    except (ValueError, KeyError) as e:
+        print(f"Error processing API response: {e}")
+        return None
+
 @tool
 def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
     #Keep this format for the description / args / args description but feel free to modify the tool