Update app.py

app.py

@@ -1,95 +1,11 @@
-from smolagents import CodeAgent, HfApiModel, load_tool, tool, DuckDuckGoSearchTool
+from smolagents import CodeAgent, HfApiModel, tool
 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 web_search(query: str) -> str:
-    """Search the web for information using DuckDuckGo.
-    
-    Args:
-        query: The search query to look up
-        
-    Returns:
-        Search results from DuckDuckGo
-    """
-    try:
-        search_tool = DuckDuckGoSearchTool()
-        results = search_tool(query)
-        if not results or "No results found" in results:
-            return f"Поиск не дал результатов для запроса: '{query}'. Попробуйте другой запрос."
-        return f"**Результаты поиска по '{query}':**\n\n{results}"
-    except Exception as e:
-        return f"Ошибка поиска: {str(e)}"
-
-@tool
-def get_weather(city: str) -> str:
-    """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)
-        
-        if response2.status_code == 200 and response2.text.strip():
-            data = response2.text.strip()
-            if '°F' in data:
-                data = data.replace('°F', '°C')
-            return f"Погода в {city}:\n{data}"
-            
-        return f"Данные о погоде в {city} временно недоступны. Попробуйте позже или проверьте на сайте gismeteo.ru"
-            
-    except Exception as e:
-        return f"Ошибка при получении погоды: {str(e)}"
-
-@tool
-def get_weather_detailed(city: str) -> str:
-    """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])  # Первые 8 строк
-            
-            short_forecast = short_forecast.replace('°F', '°C')
-            return f"Подробный прогноз для {city}:\n{short_forecast}"
-        else:
-            return f"Не удалось получить подробный прогноз для {city}"
-            
-    except Exception as e:
-        return f"Ошибка: {str(e)}"
+# 1. Локальные функции-инструменты (без интернета)
 
 @tool
 def get_current_time() -> str:
@@ -146,9 +62,9 @@ def what_are_you_doing() -> str:
         Description of current activities
     """
     responses = [
-        "Я общаюсь с вами и готов помочь с поиском информации!",
-        "В данный момент я помогаю пользователям находить ответы на их вопросы.",
-        "Я работаю ассистентом - ищу информацию в интернете, сообщаю о погоде и текущем времени.",
+        "Я общаюсь с вами и готов помочь с любыми вопросами!",
+        "В данный момент я помогаю пользователям как ассистент.",
+        "Я работаю здесь, чтобы отвечать на ваши вопросы!",
         "Сейчас я здесь, чтобы помочь вам! Чем могу быть полезен?",
         "Я анализирую ваш запрос и готовлю полезный ответ."
     ]
@@ -176,25 +92,57 @@ def calculate_math(expression: str) -> str:
         return "Ошибка в выражении"
 
 @tool
-def suggest_weather_sources(city: str) -> str:
-    """Suggest reliable weather sources for a city.
+def get_day_of_week() -> str:
+    """Get current day of the week.
     
-    Args:
-        city: The name of the city
-        
     Returns:
-        List of reliable weather sources
+        Current day name in Russian
     """
-    sources = [
-        f"Gismeteo: https://www.gismeteo.ru/weather-{city.lower()}-4368/",
-        f"Yandex.Погода: https://yandex.ru/pogoda/{city.lower()}",
-        f"Weather.com: https://weather.com/weather/today/l/{city}",
-        f"AccuWeather: https://www.accuweather.com/ru/ru/{city.lower()}/weather-forecast"
-    ]
+    days = {
+        0: "понедельник",
+        1: "вторник", 
+        2: "среда",
+        3: "четверг",
+        4: "пятница",
+        5: "суббота",
+        6: "воскресенье"
+    }
+    day_num = datetime.datetime.now().weekday()
+    return f"Сегодня {days[day_num]}"
+
+@tool
+def get_date_info() -> str:
+    """Get current date information.
     
-    return f"Надежные источники погоды для {city}:\n" + "\n".join(sources)
+    Returns:
+        Detailed date information in Russian
+    """
+    now = datetime.datetime.now()
+    months = {
+        1: "января", 2: "февраля", 3: "марта", 4: "апреля",
+        5: "мая", 6: "июня", 7: "июля", 8: "августа", 
+        9: "сентября", 10: "октября", 11: "ноября", 12: "декабря"
+    }
+    return f"Сегодня {now.day} {months[now.month]} {now.year} года"
 
-# 2. остальные объекты
+@tool
+def simple_weather_advice() -> str:
+    """Provide general weather advice based on season.
+    
+    Returns:
+        Seasonal weather advice
+    """
+    month = datetime.datetime.now().month
+    if month in [12, 1, 2]:
+        return "Сейчас зима - одевайтесь тепло!"
+    elif month in [3, 4, 5]:
+        return "Сейчас весна - погода переменчива, берите зонт!"
+    elif month in [6, 7, 8]:
+        return "Сейчас лето - отличная погода для прогулок!"
+    else:
+        return "Сейчас осень - не забудьте куртку!"
+
+# 2. Остальные объекты
 
 final_answer = FinalAnswerTool()
 
@@ -214,27 +162,26 @@ if "final_answer" not in prompt_templates:
         "post_messages": ""
     }
     
-# Создаем агента с новым именем и обновленными инструментами
+# Создаем агента только с локальными инструментами
 agent = CodeAgent(
     model=model,
     tools=[
         final_answer, 
-        web_search, 
-        get_weather, 
-        get_weather_detailed, 
         get_current_time,
         get_current_time_in_timezone, 
         how_are_you,
         what_are_you_doing,
-        calculate_math, 
-        suggest_weather_sources
+        calculate_math,
+        get_day_of_week,
+        get_date_info,
+        simple_weather_advice
     ],
-    max_steps=8,
+    max_steps=6,
     verbosity_level=1,
     grammar=None,
     planning_interval=None,
-    name="WeatherTimeBot",  # Новое имя агента
-    description="Ассистент для поиска информации, прогноза погоды и определения времени",
+    name="LocalAssistant",  # Новое имя для локального агента
+    description="Локальный ассистент для времени, дат и простых вычислений",
     prompt_templates=prompt_templates
 )