Update app.py

app.py

@@ -33,6 +33,19 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
+@tool
+def calculate_distance(latitude1: float, longitude1: float, latitude2: float, longitude2: float) -> float:
+    """Calculate the geodesic distance in kilometers between two points given their coordinates (latitude and longitude).
+    Args:
+        latitude1: Latitude of the first point.
+        longitude1: Longitude of the first point.
+        latitude2: Latitude of the second point.
+        longitude2: Longitude of the second point.
+    Returns:
+        float: Distance in kilometers.
+    """
+    from geopy import distance
+    return distance.distance((latitude1, longitude1), (latitude2, longitude2)).km
 
 final_answer = FinalAnswerTool()
 
@@ -57,7 +70,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, web_search],
+    tools=[final_answer, web_search, calculate_distance],
     max_steps=6,
     verbosity_level=1,
     grammar=None,