Update app.py

app.py

@@ -1,33 +1,27 @@
-from smolagents import CodeAgent, DuckDuckGoSearchTool, tool
+from smolagents import CodeAgent, HfApiModel, load_tool, tool, DuckDuckGoSearchTool
 import datetime
 import requests
 import pytz
 import yaml
+import base64
+import os
+import io
+from PIL import Image
+from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
-# Создаем свой собственный инструмент для финального ответа
+# 1. функции-инструменты
+
 @tool
-def final_answer(answer: str) -> str:
-    """Provide the final answer to the user. Use this when you have the answer to the user's question.
+def web_search(query: str) -> str:
+    """Search the web for information using DuckDuckGo.
     
     Args:
-        answer: The final answer to present to the user
+        query: The search query to look up
+        
+    Returns:
+        Search results from DuckDuckGo
     """
-    return answer
-
-# Инструменты (tools), используемые агентом:
-# - web_search: поиск в интернете через DuckDuckGo
-# - get_weather: получение текущей погоды для города
-# - get_weather_detailed: подробный прогноз погоды
-# - convert_fahrenheit_to_celsius: конвертация температуры
-# - get_current_time_in_timezone: время в указанном часовом поясе
-# - calculate_math: вычисление математических выражений
-# - suggest_weather_sources: рекомендации источников погоды
-# - final_answer: инструмент для финального ответа
-
-@tool
-def web_search(query: str) -> str:
-    """Search the web for information using DuckDuckGo."""
     try:
         search_tool = DuckDuckGoSearchTool()
         results = search_tool(query)
@@ -39,17 +33,26 @@ def web_search(query: str) -> str:
 
 @tool
 def get_weather(city: str) -> str:
-    """Get current weather for a city using reliable weather services."""
+    """Get current weather for a city using reliable weather services.
+    
+    Args:
+        city: The name of the city to get weather for (e.g., 'Moscow', 'London')
+        
+    Returns:
+        Current weather information in Celsius
+    """
     try:
+       
         url = f"https://wttr.in/{city}?format=%C+%t+%h+%w&m&lang=ru"
         response = requests.get(url, timeout=10)
         
         if response.status_code == 200 and response.text.strip():
             data = response.text.strip()
+           
             if '°F' in data:
                 data = data.replace('°F', '°C')
             return f"Погода в {city}:\n{data}"
-        
+       
         url2 = f"https://wttr.in/{city}?format=%C+%t+%h+%w+%P&m&lang=ru"
         response2 = requests.get(url2, timeout=10)
         
@@ -66,14 +69,24 @@ def get_weather(city: str) -> str:
 
 @tool
 def get_weather_detailed(city: str) -> str:
-    """Get detailed weather forecast for a city."""
+    """Get detailed weather forecast for a city.
+    
+    Args:
+        city: The name of the city
+        
+    Returns:
+        Detailed weather information in Celsius
+    """
     try:
+       
         url = f"https://wttr.in/{city}?m&lang=ru"
         response = requests.get(url, timeout=10)
         
         if response.status_code == 200:
+           
             lines = response.text.split('\n')
-            short_forecast = '\n'.join(lines[:8])
+            short_forecast = '\n'.join(lines[:8])  # Первые 8 строк
+            
             short_forecast = short_forecast.replace('°F', '°C')
             return f"Подробный прогноз для {city}:\n{short_forecast}"
         else:
@@ -84,7 +97,14 @@ def get_weather_detailed(city: str) -> str:
 
 @tool
 def convert_fahrenheit_to_celsius(f_temp: float) -> str:
-    """Convert Fahrenheit temperature to Celsius."""
+    """Convert Fahrenheit temperature to Celsius.
+    
+    Args:
+        f_temp: Temperature in Fahrenheit
+        
+    Returns:
+        Temperature in Celsius
+    """
     try:
         c_temp = (f_temp - 32) * 5/9
         return f"{f_temp}°F = {c_temp:.1f}°C"
@@ -93,18 +113,32 @@ def convert_fahrenheit_to_celsius(f_temp: float) -> str:
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
-    """Get current local time in specified timezone."""
+    """Get current local time in specified timezone.
+    
+    Args:
+        timezone: A valid timezone (e.g., 'America/New_York', 'Europe/Moscow')
+        
+    Returns:
+        Current time in the specified timezone
+    """
     try:
         tz = pytz.timezone(timezone)
         local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
         return f"Текущее время в {timezone}: {local_time}"
     except Exception as e:
         return f"Ошибка получения времени для часового пояса '{timezone}': {str(e)}"
-
 @tool
 def calculate_math(expression: str) -> str:
-    """Calculate mathematical expressions safely."""
+    """Calculate mathematical expressions safely.
+    
+    Args:
+        expression: A mathematical expression (e.g., '2+2', '5*3/2')
+        
+    Returns:
+        Result of the calculation
+    """
     try:
+       
         allowed_chars = set('0123456789+-*/.() ')
         if all(c in allowed_chars for c in expression):
             result = eval(expression)
@@ -116,7 +150,14 @@ def calculate_math(expression: str) -> str:
 
 @tool
 def suggest_weather_sources(city: str) -> str:
-    """Suggest reliable weather sources for a city."""
+    """Suggest reliable weather sources for a city.
+    
+    Args:
+        city: The name of the city
+        
+    Returns:
+        List of reliable weather sources
+    """
     sources = [
         f"Gismeteo: https://www.gismeteo.ru/weather-{city.lower()}-4368/",
         f"Yandex.Погода: https://yandex.ru/pogoda/{city.lower()}",
@@ -126,36 +167,21 @@ def suggest_weather_sources(city: str) -> str:
     
     return f"Надежные источники погоды для {city}:\n" + "\n".join(sources)
 
-# Используем локальную модель вместо HfApiModel
-try:
-    from transformers import AutoModelForCausalLM, AutoTokenizer
-    from smolagents import TransformersModel
-    
-    model = TransformersModel(
-        model_id='Qwen/Qwen2.5-Coder-1.5B',  # Используем меньшую модель для совместимости
-        max_tokens=2096,
-        temperature=0.5
-    )
-except ImportError:
-    # Если transformers не установлен, используем простую модель
-    from smolagents import LiteLLMModel
-    model = LiteLLMModel(
-        model_id='gpt-3.5-turbo',
-        max_tokens=2096,
-        temperature=0.5
-    )
+# 2. остальные объекты
 
-# Загружаем промпты, если файл существует, иначе создаем базовый шаблон
-try:
-    with open("prompts.yaml", 'r') as stream:
-        prompt_templates = yaml.safe_load(stream)
-except FileNotFoundError:
-    prompt_templates = {
-        "final_answer": {
-            "pre_messages": "Based on my research: ",
-            "post_messages": ""
-        }
-    }
+final_answer = FinalAnswerTool()
+
+model = HfApiModel(
+    max_tokens=2096,
+    temperature=0.5,
+    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
+    custom_role_conversions=None,
+)
+
+image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+
+with open("prompts.yaml", 'r') as stream:
+    prompt_templates = yaml.safe_load(stream)
 
 if "final_answer" not in prompt_templates:
     prompt_templates["final_answer"] = {
@@ -167,7 +193,7 @@ agent = CodeAgent(
     model=model,
     tools=[final_answer, web_search, get_weather, get_weather_detailed, convert_fahrenheit_to_celsius, get_current_time_in_timezone, calculate_math, suggest_weather_sources],
     max_steps=8,
-    verbosity_level=0,
+    verbosity_level=1,
     grammar=None,
     planning_interval=None,
     name=None,
@@ -175,4 +201,4 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-GradioUI(agent).launch()
+GradioUI(agent).launch()