Update app.py

app.py

@@ -1,4 +1,7 @@
 import os
+from dotenv import load_dotenv
+load_dotenv()
+
 from smolagents import CodeAgent, HfApiModel, tool
 import datetime
 import pytz
@@ -7,9 +10,11 @@ from geopy.geocoders import Nominatim
 from timezonefinder import TimezoneFinder
 import requests
 
-# Инструмент: Получение таймзоны по названию города
 @tool
 def get_timezone_by_city(city: str) -> str:
+    """
+    Get the timezone string for a given city name.
+    """
     if not isinstance(city, str):
         return f"Error: Expected a string for city, got: {type(city)}"
     try:
@@ -26,9 +31,11 @@ def get_timezone_by_city(city: str) -> str:
     except Exception as e:
         return f"Error fetching timezone: {str(e)}"
 
-# Инструмент: Время по таймзоне
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
+    """
+    Return the current local time in the specified timezone.
+    """
     try:
         tz = pytz.timezone(timezone)
         local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
@@ -36,9 +43,11 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time: {str(e)}"
 
-# Инструмент: Качество воздуха
 @tool
 def get_air_quality(city: str, lang: str = "en") -> str:
+    """
+    Return formatted air quality information for a given city.
+    """
     try:
         geolocator = Nominatim(user_agent="smolagents_bot")
         location = geolocator.geocode(city, language="en")
@@ -66,9 +75,11 @@ def get_air_quality(city: str, lang: str = "en") -> str:
     except Exception as e:
         return f"Error fetching air quality: {str(e)}"
 
-# Инструмент: Погода
 @tool
 def get_weather(city: str, lang: str = "en") -> str:
+    """
+    Return formatted weather information for a given city.
+    """
     try:
         geolocator = Nominatim(user_agent="smolagents_bot")
         location = geolocator.geocode(city, language="en")
@@ -89,19 +100,19 @@ def get_weather(city: str, lang: str = "en") -> str:
     except Exception as e:
         return f"Error fetching weather: {str(e)}"
 
-# Инструмент: Финальный ответ
 @tool
 def final_answer(answer: str) -> str:
+    """
+    Return the final answer string to be displayed to the user.
+    """
     return answer
 
-# Модель
 model = HfApiModel(
     max_tokens=1000,
     temperature=0.5,
     model_id='mistralai/Mixtral-8x7B-Instruct-v0.1',
 )
 
-# System prompt — что может делать агент
 system_prompt = (
     "You are a helpful assistant. Given a user question, use ONLY the available tools: "
     "`get_timezone_by_city`, `get_current_time_in_timezone`, `get_air_quality`, `get_weather`, `final_answer`. "
@@ -111,7 +122,6 @@ system_prompt = (
     "```py\nfinal_answer('...')\n```"
 )
 
-# Агент
 agent = CodeAgent(
     model=model,
     tools=[get_timezone_by_city, get_current_time_in_timezone, get_air_quality, get_weather, final_answer],
@@ -123,7 +133,6 @@ agent = CodeAgent(
     verbosity_level=2,
 )
 
-# Gradio интерфейс
 def process_input(user_input):
     try:
         response = agent.run(user_input)