added hubStats tool

tools.py

@@ -1,8 +1,8 @@
-from smolagents import DuckDuckGoSearchTool
-
-from smolagents import Tool
+from smolagents import DuckDuckGoSearchTool, Tool
 import random
+from huggingface_hub import list_models
 
+# --- Weather Info Tool ---
 class WeatherInfoTool(Tool):
     name = "weather_info"
     description = "Fetches dummy weather information for a given location."
@@ -15,18 +15,41 @@ class WeatherInfoTool(Tool):
     output_type = "string"
 
     def forward(self, location: str):
-        # Dummy weather data
         weather_conditions = [
             {"condition": "Rainy", "temp_c": 15},
             {"condition": "Clear", "temp_c": 25},
             {"condition": "Windy", "temp_c": 20}
         ]
-        # Randomly select a weather condition
         data = random.choice(weather_conditions)
         return f"Weather in {location}: {data['condition']}, {data['temp_c']}°C"
 
-# Initialize the tool
 weather_info_tool = WeatherInfoTool()
 
-# Initialize the DuckDuckGo search tool
+# --- DuckDuckGo Search Tool ---
 search_tool = DuckDuckGoSearchTool()
+
+# --- Hub Stats Tool ---
+class HubStatsTool(Tool):
+    name = "hub_stats"
+    description = "Fetches the most downloaded model from a specific author on the Hugging Face Hub."
+    inputs = {
+        "author": {
+            "type": "string",
+            "description": "The username of the model author/organization to find models from."
+        }
+    }
+    output_type = "string"
+
+    def forward(self, author: str):
+        try:
+            models = list(list_models(author=author, sort="downloads", direction=-1, limit=1))
+            if models:
+                model = models[0]
+                return f"The most downloaded model by {author} is {model.id} with {model.downloads:,} downloads."
+            else:
+                return f"No models found for author {author}."
+        except Exception as e:
+            return f"Error fetching models for {author}: {str(e)}"
+
+hub_stats_tool = HubStatsTool()
+