Update app.py

app.py

@@ -1,28 +1,22 @@
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
+# app.py
+
 import datetime
 import requests
 import pytz
 import yaml
+
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
-# Инструмент-заглушка, который пока ничего не делает
 @tool
-def my_cutom_tool(arg1: str, arg2: int) -> str:
-    """A tool that does nothing yet.
-    Args:
-        arg1: первый аргумент
-        arg2: второй аргумент (проверка)
-    """
+def my_custom_tool(arg1: str, arg2: int) -> str:
+    """A tool that does nothing yet."""
     return "What magic will you build?"
 
-# Инструмент для получения текущего времени в заданном часовом поясе
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
-    """A tool that fetches the current local time in a specified timezone.
-    Args:
-        timezone: строка с названием часового пояса (например, 'America/New_York').
-    """
+    """Get the current local time in a specified timezone."""
     try:
         tz = pytz.timezone(timezone)
         local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
@@ -30,13 +24,9 @@ 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 get_weather(location: str) -> str:
-    """A tool that fetches the current weather for a given location.
-    Args:
-        location: название местоположения (например, 'London').
-    """
+    """Fetch the current weather for a given location."""
     try:
         response = requests.get(f"http://wttr.in/{location}?format=3")
         if response.status_code == 200:
@@ -46,10 +36,9 @@ def get_weather(location: str) -> str:
     except Exception as e:
         return f"Error fetching weather: {str(e)}"
 
-# Новый инструмент: получение случайной шутки
 @tool
 def get_random_joke() -> str:
-    """A tool that fetches a random joke."""
+    """Fetch a random joke."""
     try:
         response = requests.get("https://official-joke-api.appspot.com/random_joke")
         if response.status_code == 200:
@@ -62,10 +51,9 @@ def get_random_joke() -> str:
     except Exception as e:
         return f"Error fetching joke: {str(e)}"
 
-# Инструмент для вывода финального ответа
 final_answer = FinalAnswerTool()
 
-# Инициализация модели для агента через API Hugging Face
+# Настройка модели (убедитесь, что модель доступна)
 model = HfApiModel(
     max_tokens=2096,
     temperature=0.5,
@@ -73,19 +61,16 @@ model = HfApiModel(
     custom_role_conversions=None,
 )
 
-# Загрузка инструмента генерации изображений с удалённого репозитория
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
-# Загрузка шаблонов промтов из YAML-файла
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
-    
-# Создание агента с расширенным списком инструментов
+
 agent = CodeAgent(
     model=model,
     tools=[
         final_answer,
-        my_cutom_tool,
+        my_custom_tool,
         get_current_time_in_timezone,
         get_weather,
         get_random_joke,
@@ -101,5 +86,4 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-# Запуск пользовательского интерфейса через Gradio
 GradioUI(agent).launch()