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 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)}"

@tool
def convert_fahrenheit_to_celsius(f_temp: float) -> str:
"""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"
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.

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)}"