Update app.py

app.py

@@ -4,7 +4,7 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
-
+import json
 from Gradio_UI import GradioUI
 from typing import Optional, Dict, Any, List
 import requests
@@ -114,14 +114,73 @@ def pdok_location_info(postal_code: Optional[str] = None, house_number: Optional
         first_result = docs[0]
         debug_info.append(f"First result data: {first_result}")
         
-        location_info = PDOKLocationInfo(**first_result)
-        return location_info.model_dump_json()
-    
+        # Create a dictionary with only the fields we need
+        location_info = {
+            "straatnaam": first_result.get("straatnaam", ""),
+            "huisnummer": first_result.get("huisnummer", ""),
+            "postcode": first_result.get("postcode", ""),
+            "woonplaatsnaam": first_result.get("woonplaatsnaam", ""),
+            "gemeentenaam": first_result.get("gemeentenaam", ""),
+            "provincienaam": first_result.get("provincienaam", ""),
+            "buurtnaam": first_result.get("buurtnaam", ""),
+            "wijknaam": first_result.get("wijknaam", ""),
+            "centroide_ll": first_result.get("centroide_ll", ""),
+            "centroide_rd": first_result.get("centroide_rd", "")
+        }
+        
+        return json.dumps(location_info)
+        
     except requests.exceptions.RequestException as e:
         return f"Error during API request: {e}\nDebug info:\n" + "\n".join(debug_info)
     except (ValueError, KeyError) as e:
         return f"Error processing API response: {e}\nDebug info:\n" + "\n".join(debug_info)
 
+# Usage code:
+location_info_str = pdok_location_info(postal_code=None, house_number="13", street_name="Emmastraat", city="Den Haag")
+
+try:
+    # Parse the JSON string into a dictionary
+    location_info = json.loads(location_info_str)
+    
+    # Create info dictionary with safe access to potentially missing fields
+    info = {}
+    
+    # Basic fields
+    for field in ["straatnaam", "huisnummer", "postcode", "woonplaatsnaam", "gemeentenaam", "provincienaam", "buurtnaam", "wijknaam"]:
+        info[field] = location_info.get(field, "N/A")
+    
+    # Coordinate fields that need parsing
+    if location_info.get("centroide_ll"):
+        try:
+            ll_parts = location_info["centroide_ll"].replace("POINT(", "").replace(")", "").split()
+            info["Latitude (WGS84)"] = ll_parts[0] if len(ll_parts) > 0 else "N/A"
+            info["Longitude (WGS84)"] = ll_parts[1] if len(ll_parts) > 1 else "N/A"
+        except:
+            info["Latitude (WGS84)"] = "N/A"
+            info["Longitude (WGS84)"] = "N/A"
+    
+    if location_info.get("centroide_rd"):
+        try:
+            rd_parts = location_info["centroide_rd"].replace("POINT(", "").replace(")", "").split()
+            info["X (RD)"] = rd_parts[0] if len(rd_parts) > 0 else "N/A"
+            info["Y (RD)"] = rd_parts[1] if len(rd_parts) > 1 else "N/A"
+        except:
+            info["X (RD)"] = "N/A"
+            info["Y (RD)"] = "N/A"
+    
+    # Print the information
+    print("Location Information:")
+    print("-" * 50)
+    for key, value in info.items():
+        print(f"{key}: {value}")
+
+except json.JSONDecodeError:
+    print("Error: Unable to parse location information")
+    print(f"Raw response: {location_info_str}")
+except Exception as e:
+    print(f"Error processing location information: {str(e)}")
+    print(f"Raw response: {location_info_str}")
+
 @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