Update app.py

app.py

@@ -1,27 +1,43 @@
 import os
 import datetime
+import requests
 import pytz
 import yaml
-
-from smolagents import CodeAgent, load_tool, tool, HfApiModel
+from smolagents import CodeAgent, HfApiModel, load_tool, tool
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
-# Загружаем секрет из переменных окружения
-HF_TOKEN = os.environ.get("HF_API_TOKEN")
-if HF_TOKEN is None:
-    raise RuntimeError("HF_API_TOKEN is not set. Please set it in Space secrets.")
-
-# Определяем инструменты
 
+# -------------------------
+# Пример кастомного инструмента
+# -------------------------
 @tool
 def my_custom_tool(arg1: str, arg2: int) -> str:
-    """Инструмент-пример, пока ничего не делает"""
-    return f"Custom tool executed with arg1={arg1}, arg2={arg2}"
+    """A tool that does nothing yet.
+
+    Args:
+        arg1: the first argument
+        arg2: the second argument
+
+    Returns:
+        A placeholder string message.
+    """
+    return "What magic will you build ?"
 
+
+# -------------------------
+# Текущее время в таймзоне
+# -------------------------
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
-    """Получает текущее локальное время в указанном часовом поясе."""
+    """Fetches the current local time in a specified timezone.
+
+    Args:
+        timezone: A string representing a valid timezone (e.g., 'America/New_York').
+
+    Returns:
+        A string with the current local time in the given timezone.
+    """
     try:
         tz = pytz.timezone(timezone)
         local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
@@ -29,37 +45,114 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
-# Финальный инструмент
-final_answer = FinalAnswerTool()
 
-# Загружаем инструмент генерации изображений из Hub
-image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+# -------------------------
+# Курс EUR
+# -------------------------
+@tool
+def get_eur_exchange_rate(currency_code: str) -> str:
+    """Retrieves the current exchange rate from EUR to a given currency.
+
+    Args:
+        currency_code: The 3-letter currency code in lowercase (e.g., "usd", "gbp", "jpy").
+
+    Returns:
+        The exchange rate as a string.
+    """
+    try:
+        api_url = "https://cdn.jsdelivr.net/npm/@fawazahmed0/currency-api@latest/v1/currencies/eur.json"
+        response = requests.get(api_url, timeout=10)
+        response.raise_for_status()
+        data = response.json()
+
+        date = data.get("date", "Unknown date")
+        exchange_rates = data.get("eur", {})
+
+        if currency_code in exchange_rates:
+            rate = exchange_rates[currency_code]
+            return f"1 EUR = {rate} {currency_code.upper()} (as of {date})"
+        else:
+            return f"Currency code '{currency_code}' not found in the exchange rate data."
+    except Exception as e:
+        return f"Error fetching exchange rate: {str(e)}"
 
-# Загружаем шаблоны промптов, если есть
-prompt_templates = {}
-try:
-    with open("prompts.yaml", "r") as stream:
-        prompt_templates = yaml.safe_load(stream)
-except FileNotFoundError:
-    pass
 
-# Конфигурация модели с использованием секретного токена
+# -------------------------
+# Шутка
+# -------------------------
+@tool
+def get_simple_joke() -> str:
+    """Fetches a random joke from JokeAPI with minimal filtering.
+
+    Returns:
+        A string with the joke or an error message.
+    """
+    try:
+        api_url = "https://v2.jokeapi.dev/joke/any"
+        response = requests.get(api_url, timeout=10)
+        response.raise_for_status()
+        joke_data = response.json()
+
+        if joke_data.get("error", False):
+            return f"Error: {joke_data.get('message', 'Could not get joke')}"
+
+        if joke_data.get("type") == "single":
+            return f"Here's a joke:\n\n{joke_data.get('joke', '')}"
+        elif joke_data.get("type") == "twopart":
+            setup = joke_data.get("setup", "")
+            delivery = joke_data.get("delivery", "")
+            return f"Here's a joke:\n\n{setup}\n{delivery}"
+
+        return "Could not parse joke from API response"
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+
+# -------------------------
+# Настройка агента
+# -------------------------
+final_answer = FinalAnswerTool()
+
+# ✅ Токен теперь берём из Hugging Face Secrets
+hf_token = os.environ.get("HF_API_TOKEN")
+if not hf_token:
+    raise ValueError("HF_API_TOKEN is not set. Please add it in Hugging Face Secrets.")
+
 model = HfApiModel(
     model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
-    api_key=HF_TOKEN,
     max_tokens=2096,
     temperature=0.5,
-    custom_role_conversions=None,
+    api_key=hf_token,  # <--- ключ теперь из секрета
 )
 
-# Создаём агент с инструментами
+# Инструмент для генерации картинок
+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)
+
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, my_custom_tool, get_current_time_in_timezone, image_generation_tool],
+    tools=[
+        final_answer,
+        my_custom_tool,
+        get_current_time_in_timezone,
+        get_eur_exchange_rate,
+        get_simple_joke,
+        image_generation_tool,
+    ],
     max_steps=6,
     verbosity_level=1,
+    grammar=None,
+    planning_interval=None,
+    name="MyAgent",
+    description="An agent with custom tools (time, jokes, exchange rate, images).",
     prompt_templates=prompt_templates,
 )
 
-# Запуск интерфейса
-GradioUI(agent).launch()
+# -------------------------
+# Запуск Gradio UI
+# -------------------------
+if __name__ == "__main__":
+    GradioUI(agent).launch()