Update app.py

app.py

@@ -1,35 +1,94 @@
-from smolagents import CodeAgent, tool
-from smolagents.models import TransformersModel
+from smolagents import CodeAgent, HfApiModel, load_tool, tool, DuckDuckGoSearchTool
 import datetime
+import requests
 import pytz
 import yaml
-import random
+import base64
+import os
+import io
+from PIL import Image
+from tools.final_answer import FinalAnswerTool
+from Gradio_UI import GradioUI
+
+# Инструменты (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: инструмент для финального ответа
 
-# 1. Локальные функции-инструменты (без интернета)
+@tool
+def web_search(query: str) -> str:
+    """Search the web for information using 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_current_time() -> str:
-    """Get current local time.
-    
-    Returns:
-        Current date and time
-    """
+def get_weather(city: str) -> str:
+    """Get current weather for a city using reliable weather services."""
     try:
-        current_time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-        return f"Текущее время: {current_time}"
+        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)}"
+        return f"Ошибка при получении погоды: {str(e)}"
 
 @tool
-def get_current_time_in_timezone(timezone: str) -> str:
-    """Get current local time in specified timezone.
-    
-    Args:
-        timezone: A valid timezone (e.g., 'America/New_York', 'Europe/Moscow')
+def get_weather_detailed(city: str) -> str:
+    """Get detailed weather forecast for a city."""
+    try:
+        url = f"https://wttr.in/{city}?m&lang=ru"
+        response = requests.get(url, timeout=10)
         
-    Returns:
-        Current time in the specified timezone
-    """
+        if response.status_code == 200:
+            lines = response.text.split('\n')
+            short_forecast = '\n'.join(lines[: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)}"
+
+@tool
+def convert_fahrenheit_to_celsius(f_temp: float) -> str:
+    """Convert Fahrenheit temperature to Celsius."""
+    try:
+        c_temp = (f_temp - 32) * 5/9
+        return f"{f_temp}°F = {c_temp:.1f}°C"
+    except Exception as e:
+        return f"Ошибка конвертации: {str(e)}"
+
+@tool
+def get_current_time_in_timezone(timezone: str) -> str:
+    """Get current local time in specified timezone."""
     try:
         tz = pytz.timezone(timezone)
         local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
@@ -37,48 +96,9 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Ошибка получения времени для часового пояса '{timezone}': {str(e)}"
 
-@tool
-def how_are_you() -> str:
-    """Answer questions about how you are doing.
-    
-    Returns:
-        Friendly response about current status
-    """
-    responses = [
-        "У меня всё отлично! Готов помочь вам с любыми вопросами.",
-        "Прекрасно себя чувствую! Что хотели бы узнать?",
-        "Всё хорошо, спасибо! Чем могу быть полезен?",
-        "Отлично! Готов к работе и жду ваших вопросов.",
-        "Замечательно! Как ваши дела?"
-    ]
-    return random.choice(responses)
-
-@tool
-def what_are_you_doing() -> str:
-    """Answer questions about what you are currently doing.
-    
-    Returns:
-        Description of current activities
-    """
-    responses = [
-        "Я общаюсь с вами и готов помочь с любыми вопросами!",
-        "В данный момент я помогаю пользователям как ассистент.",
-        "Я работаю здесь, чтобы отвечать на ваши вопросы!",
-        "Сейчас я здесь, чтобы помочь вам! Чем могу быть полезен?",
-        "Я анализирую ваш запрос и готовлю полезный ответ."
-    ]
-    return random.choice(responses)
-
 @tool
 def calculate_math(expression: str) -> str:
-    """Calculate mathematical expressions safely.
-    
-    Args:
-        expression: A mathematical expression (e.g., '2+2', '5*3/2')
-        
-    Returns:
-        Result of the calculation
-    """
+    """Calculate mathematical expressions safely."""
     try:
         allowed_chars = set('0123456789+-*/.() ')
         if all(c in allowed_chars for c in expression):
@@ -90,144 +110,45 @@ def calculate_math(expression: str) -> str:
         return "Ошибка в выражении"
 
 @tool
-def get_day_of_week() -> str:
-    """Get current day of the week.
+def suggest_weather_sources(city: str) -> str:
+    """Suggest reliable weather sources for a city."""
+    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"
+    ]
     
-    Returns:
-        Current day name in Russian
-    """
-    days = {
-        0: "понедельник",
-        1: "вторник", 
-        2: "среда",
-        3: "четверг",
-        4: "пятница",
-        5: "суббота",
-        6: "воскресенье"
-    }
-    day_num = datetime.datetime.now().weekday()
-    return f"Сегодня {days[day_num]}"
+    return f"Надежные источники погоды для {city}:\n" + "\n".join(sources)
 
-@tool
-def get_date_info() -> str:
-    """Get current date information.
-    
-    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} года"
+final_answer = FinalAnswerTool()
 
-@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 "Сейчас осень - не забудьте куртку!"
+model = HfApiModel(
+    max_tokens=2096,
+    temperature=0.5,
+    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
+    custom_role_conversions=None,
+)
 
-@tool
-def count_characters(text: str) -> str:
-    """Count the number of characters in a text.
-    
-    Args:
-        text: The text to count characters in
-        
-    Returns:
-        Number of characters
-    """
-    return f"В тексте {len(text)} символов"
+with open("prompts.yaml", 'r') as stream:
+    prompt_templates = yaml.safe_load(stream)
 
-@tool
-def repeat_text(text: str, times: int) -> str:
-    """Repeat text specified number of times.
+if "final_answer" not in prompt_templates:
+    prompt_templates["final_answer"] = {
+        "pre_messages": "Based on my research: ",
+        "post_messages": ""
+    }
     
-    Args:
-        text: Text to repeat
-        times: Number of times to repeat
-        
-    Returns:
-        Repeated text
-    """
-    if times <= 10:
-        return (text + " ") * times
-    else:
-        return "Слишком много повторений! Максимум 10 раз."
-
-# 2. Создаем модель и агента
-
-# Используем легкую локальную модель
-model = TransformersModel("microsoft/DialoGPT-small")
-
-# Создаем агента только с локальными инструментами
 agent = CodeAgent(
     model=model,
-    tools=[
-        get_current_time,
-        get_current_time_in_timezone, 
-        how_are_you,
-        what_are_you_doing,
-        calculate_math,
-        get_day_of_week,
-        get_date_info,
-        simple_weather_advice,
-        count_characters,
-        repeat_text
-    ],
-    max_steps=6,
-    verbosity_level=1,
+    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,
     grammar=None,
     planning_interval=None,
-    name="LocalAssistant",
-    description="Локальный ассистент для времени, дат и простых вычислений"
+    name=None,
+    description=None,
+    prompt_templates=prompt_templates
 )
 
-# 3. Простой интерфейс Gradio (если нет Gradio_UI)
-import gradio as gr
-
-def chat_with_agent(message, history):
-    """Функция для общения с агентом"""
-    try:
-        response = agent.run(message)
-        return response
-    except Exception as e:
-        return f"Ошибка: {str(e)}"
-
-# Создаем простой интерфейс
-with gr.Blocks() as demo:
-    gr.Markdown("# 🎯 Local Assistant")
-    gr.Markdown("Локальный ассистент для времени, дат и вычислений")
-    
-    chatbot = gr.Chatbot(height=400)
-    msg = gr.Textbox(
-        label="Ваш вопрос", 
-        placeholder="Спросите о времени, дате или сделайте вычисление...",
-        lines=2
-    )
-    clear_btn = gr.Button("Очистить чат")
-    
-    def respond(message, chat_history):
-        bot_response = chat_with_agent(message, chat_history)
-        chat_history.append((message, bot_response))
-        return "", chat_history
-    
-    msg.submit(respond, [msg, chatbot], [msg, chatbot])
-    clear_btn.click(lambda: None, None, chatbot, queue=False)
-
-if __name__ == "__main__":
-    print("🚀 Starting Local Assistant...")
-    demo.launch(share=True)
+GradioUI(agent).launch()