Update app.py

app.py

@@ -1,37 +1,21 @@
-from smolagents import Tool
-from typing import Optional
+import os
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel
 
-class SimpleTool(Tool):
-    name = "get_travel_duration"
-    description = "Gets the travel time between two places."
-    inputs = {"start_location":{"type":"string","description":"the place from which you start your ride"},"destination_location":{"type":"string","description":"the place of arrival"},"transportation_mode":{"type":"string","nullable":True,"description":"The transportation mode, in 'driving', 'walking', 'bicycling', or 'transit'. Defaults to 'driving'."}}
-    output_type = "string"
+hf_token = os.getenv("HF_TOKEN")
 
-    def forward(self, start_location: str, destination_location: str, transportation_mode: Optional[str] = None) -> str:
-        """Gets the travel time between two places.
-        Args:
-            start_location: the place from which you start your ride
-            destination_location: the place of arrival
-            transportation_mode: The transportation mode, in 'driving', 'walking', 'bicycling', or 'transit'. Defaults to 'driving'.
-        """
-        import os   # All imports are placed within the function, to allow for sharing to Hub.
-        import googlemaps
-        from datetime import datetime
+def who_is_bot(person_name: str) -> str:
+    agent = CodeAgent(
+        tools=[DuckDuckGoSearchTool()],
+        model=HfApiModel(hf_token=hf_token)
+    )
+    prompt = (
+        f"Who is {person_name}? "
+        "Give a short 1-2 sentence summary."
+    )
+    result = agent.run(prompt)
+    return result.strip()
 
-        gmaps = googlemaps.Client(os.getenv("GMAPS_API_KEY"))
-
-        if transportation_mode is None:
-            transportation_mode = "driving"
-        try:
-            directions_result = gmaps.directions(
-                start_location,
-                destination_location,
-                mode=transportation_mode,
-                departure_time=datetime(2025, 12, 6, 11, 0), # At 11, date far in the future
-            )
-            if len(directions_result) == 0:
-                return "No way found between these places with the required transportation mode."
-            return directions_result[0]["legs"][0]["duration"]["text"]
-        except Exception as e:
-            print(e)
-            return e
+if __name__ == "__main__":
+    person = "Sundar Pichai"
+    summary = who_is_bot(person)
+    print(f"About {person}: {summary}")