leonid11l commited on
Commit
a4a9ac4
·
verified ·
1 Parent(s): 8c5c24b

Delete app.py

Browse files

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

Files changed (1) hide show
  1. app.py +0 -69
app.py DELETED
@@ -1,69 +0,0 @@
1
- from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
2
- import datetime
3
- import requests
4
- import pytz
5
- import yaml
6
- from tools.final_answer import FinalAnswerTool
7
-
8
- from Gradio_UI import GradioUI
9
-
10
- # Below is an example of a tool that does nothing. Amaze us with your creativity !
11
- @tool
12
- def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
13
- #Keep this format for the description / args / args description but feel free to modify the tool
14
- """A tool that does nothing yet
15
- Args:
16
- arg1: the first argument
17
- arg2: the second argument
18
- """
19
- return "What magic will you build ?"
20
-
21
- @tool
22
- def get_current_time_in_timezone(timezone: str) -> str:
23
- """A tool that fetches the current local time in a specified timezone.
24
- Args:
25
- timezone: A string representing a valid timezone (e.g., 'America/New_York').
26
- """
27
- try:
28
- # Create timezone object
29
- tz = pytz.timezone(timezone)
30
- # Get current time in that timezone
31
- local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
32
- return f"The current local time in {timezone} is: {local_time}"
33
- except Exception as e:
34
- return f"Error fetching time for timezone '{timezone}': {str(e)}"
35
-
36
-
37
- final_answer = FinalAnswerTool()
38
-
39
- # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
40
- # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
41
-
42
- model = HfApiModel(
43
- max_tokens=2096,
44
- temperature=0.5,
45
- model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
46
- custom_role_conversions=None,
47
- )
48
-
49
-
50
- # Import tool from Hub
51
- image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
52
-
53
- with open("prompts.yaml", 'r') as stream:
54
- prompt_templates = yaml.safe_load(stream)
55
-
56
- agent = CodeAgent(
57
- model=model,
58
- tools=[final_answer], ## add your tools here (don't remove final answer)
59
- max_steps=6,
60
- verbosity_level=1,
61
- grammar=None,
62
- planning_interval=None,
63
- name=None,
64
- description=None,
65
- prompt_templates=prompt_templates
66
- )
67
-
68
-
69
- GradioUI(agent).launch()