Upload tool

app.py

@@ -0,0 +1,6 @@
+from smolagents import launch_gradio_demo
+from tool import SimpleTool
+
+tool = SimpleTool()
+
+launch_gradio_demo(tool)

requirements.txt

@@ -0,0 +1,2 @@
+geopy
+smolagents

tool.py

@@ -0,0 +1,54 @@
+from smolagents import Tool
+from typing import Any, Optional
+
+class SimpleTool(Tool):
+    name = "get_travel_duration"
+    description = "Gets the travel time between two places using straight-line distance."
+    inputs = {"start_location":{"type":"string","description":"The starting point of the journey."},"destination_location":{"type":"string","description":"The destination point of the journey."},"transportation_mode":{"type":"string","nullable":True,"description":"Mode of transport ('driving', 'walking', 'bicycling', or 'transit'). Defaults to 'driving'."}}
+    output_type = "string"
+
+    def forward(self, start_location: str, destination_location: str, transportation_mode: Optional[str] = None) -> str:
+        """Gets the travel time between two places using straight-line distance.
+
+        Args:
+            start_location: The starting point of the journey.
+            destination_location: The destination point of the journey.
+            transportation_mode: Mode of transport ('driving', 'walking', 'bicycling', or 'transit'). Defaults to 'driving'.
+
+        Returns:
+            str: The estimated duration of travel between the two locations.
+        """
+        # All imports must be inside the function for proper Hub sharing
+        from geopy.distance import geodesic
+        from geopy.geocoders import Nominatim
+        from datetime import datetime
+
+        try:
+            geolocator = Nominatim(user_agent="my_travel_calculator")
+
+            # Get coordinates
+            start = geolocator.geocode(start_location)
+            end = geolocator.geocode(destination_location)
+
+            if not start or not end:
+                return "Could not find one or both locations"
+
+            # Calculate distance
+            distance = geodesic((start.latitude, start.longitude), 
+                              (end.latitude, end.longitude)).kilometers
+
+            # Approximate speeds in km/h
+            speeds = {
+                'driving': 30,  # Urban average
+                'walking': 5,
+                'bicycling': 15,  # Changed 'cycling' to 'bicycling' to match documentation
+                'transit': 25
+            }
+            speed = speeds.get(transportation_mode, speeds['driving'])
+
+            # Calculate time in minutes
+            time_minutes = (distance / speed) * 60
+
+            return f"Approximately {int(time_minutes)} minutes (straight-line estimate)"
+        except Exception as e:
+            return f"Error calculating route: {str(e)}"