Spaces:
Sleeping
Sleeping
| from smolagents import DuckDuckGoSearchTool | |
| from smolagents import Tool | |
| import random | |
| import requests | |
| from huggingface_hub import list_models | |
| # Initialize the DuckDuckGo search tool | |
| #search_tool = DuckDuckGoSearchTool() | |
| class WeatherInfoTool(Tool): | |
| name = "weather_info" | |
| description = "Fetches dummy weather information for a given location." | |
| inputs = { | |
| "location": { | |
| "type": "string", | |
| "description": "The location to get weather information for." | |
| } | |
| } | |
| 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" | |
| 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: | |
| # List models from the specified author, sorted by downloads | |
| 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)}" | |
| class CountryLanguageTool(Tool): | |
| name = "country_language" | |
| description = "Returns the main language(s) spoken in a given country (offline database)." | |
| inputs = { | |
| "country": { | |
| "type": "string", | |
| "description": "The name of the country (e.g., France, Japan, Brazil)." | |
| } | |
| } | |
| output_type = "string" | |
| def forward(self, country: str): | |
| try: | |
| country_languages = { | |
| "united states": "English", | |
| "usa": "English", | |
| "canada": "English, French", | |
| "france": "French", | |
| "germany": "German", | |
| "spain": "Spanish", | |
| "mexico": "Spanish", | |
| "brazil": "Portuguese", | |
| "japan": "Japanese", | |
| "china": "Mandarin Chinese", | |
| "india": "Hindi, English", | |
| "russia": "Russian", | |
| "italy": "Italian", | |
| "south korea": "Korean", | |
| "korea": "Korean", | |
| "united kingdom": "English", | |
| "uk": "English", | |
| "australia": "English", | |
| "nigeria": "English", | |
| "egypt": "Arabic", | |
| "saudi arabia": "Arabic", | |
| "argentina": "Spanish", | |
| "switzerland": "German, French, Italian, Romansh", | |
| "south africa": "Zulu, Xhosa, Afrikaans, English", | |
| } | |
| key = country.lower().strip() | |
| if key in country_languages: | |
| return ( | |
| f"The main language(s) spoken in {country} " | |
| f"is: {country_languages[key]}." | |
| ) | |
| return f"Sorry, I don't have language data for {country}." | |
| except Exception as e: | |
| return f"Error retrieving language info: {str(e)}" | |
| # class CountryLanguageTool(Tool): | |
| # name = "country_language" | |
| # description = "Gives the language(s) spoken in a requested country." | |
| # inputs = { | |
| # "country":{ | |
| # "type": "string", | |
| # "description": "The country to get the language from." | |
| # } | |
| # } | |
| # output_type = "string" | |
| # def forward(self, country: str): | |
| # try: | |
| # url = f"https://restcountries.com/v3.1/name/{country}" | |
| # response = requests.get(url, timeout=10) | |
| # response.raise_for_status() | |
| # data = response.json() | |
| # if not data: | |
| # return f"No data found for {country}." | |
| # country_data = data[0] | |
| # languages = country_data.get("languages", {}) | |
| # if not languages: | |
| # return f"No language information available for {country}." | |
| # language_list = ", ".join(languages.values()) | |
| # return ( | |
| # f"The main language(s) spoken in {country} " | |
| # f"is: {language_list}." | |
| # ) | |
| # except requests.exceptions.HTTPError: | |
| # return f"Country '{country}' not found." | |
| # except requests.exceptions.RequestException as e: | |
| # return f"Network error while fetching data: {str(e)}" | |
| # except Exception as e: | |
| # return f"Unexpected error: {str(e)}" |