Update app.py

app.py

@@ -4,30 +4,26 @@ import requests
 import pytz
 import yaml
 
-from smolagents import CodeAgent, HfApiModel, load_tool, tool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
-# ---------------------------
-# Custom tools
-# ---------------------------
+# ---------------------- Custom Tools ----------------------
 
 @tool
 def my_custom_tool(arg1: str, arg2: int) -> str:
-    """
-    A tool that demonstrates how to define custom functions.
+    """A simple example tool that just echoes its inputs.
     Args:
-        arg1: The first argument as a string.
-        arg2: The second argument as an integer.
+        arg1: the first argument
+        arg2: the second argument
     """
-    return f"You passed arg1={arg1} and arg2={arg2}. Imagine some real magic here!"
+    return f"Tool received arg1={arg1}, arg2={arg2}"
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
-    """
-    A tool that fetches the current local time in a specified timezone.
+    """Fetches the current local time in a specified timezone.
     Args:
-        timezone: A string representing a valid timezone (e.g., 'America/New_York').
+        timezone: e.g., 'America/New_York'
     """
     try:
         tz = pytz.timezone(timezone)
@@ -38,10 +34,9 @@ def get_current_time_in_timezone(timezone: str) -> str:
 
 @tool
 def get_eur_exchange_rate(currency_code: str) -> str:
-    """
-    A tool that retrieves the current exchange rate from EUR to a given currency.
+    """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").
+        currency_code: 3-letter code, e.g. "usd", "gbp"
     """
     try:
         api_url = "https://cdn.jsdelivr.net/npm/@fawazahmed0/currency-api@latest/v1/currencies/eur.json"
@@ -57,13 +52,11 @@ def get_eur_exchange_rate(currency_code: str) -> str:
         else:
             return f"Currency code '{currency_code}' not found."
     except Exception as e:
-        return f"Error fetching exchange rate data: {str(e)}"
+        return f"Error fetching exchange rate: {str(e)}"
 
 @tool
 def get_simple_joke() -> str:
-    """
-    A tool that fetches a random joke from JokeAPI with minimal filtering.
-    """
+    """Fetches a random joke from JokeAPI."""
     try:
         api_url = "https://v2.jokeapi.dev/joke/any"
         response = requests.get(api_url, timeout=10)
@@ -78,51 +71,45 @@ def get_simple_joke() -> str:
         elif joke_data.get("type") == "twopart":
             return f"Here's a joke:\n\n{joke_data.get('setup', '')}\n{joke_data.get('delivery', '')}"
 
-        return "Could not parse joke from API response"
+        return "Could not parse joke from API response."
     except Exception as e:
-        return f"Error: {str(e)}"
-
+        return f"Error fetching joke: {str(e)}"
 
-# ---------------------------
-# Setup model and agent
-# ---------------------------
+# ---------------------- Model ----------------------
 
-# ✅ Токен берется автоматически из Secrets (HF_TOKEN), его не надо писать в коде!
-final_answer = FinalAnswerTool()
+# ВАЖНО: токен нужно добавить в Secrets как HF_TOKEN
+hf_token = os.environ.get("HF_TOKEN")
+if not hf_token:
+    raise RuntimeError("❌ HF_TOKEN is not set in repository secrets!")
 
 model = HfApiModel(
     model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
-    max_tokens=2096,
+    max_tokens=2048,
     temperature=0.5,
-    custom_role_conversions=None,
+    api_key=hf_token,  # <-- токен берется из Secrets
 )
 
-# Load external tool from Hub
-image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+# ---------------------- Agent ----------------------
+
+final_answer = FinalAnswerTool()
 
-# Load prompts
+# Подгружаем промпты
 with open("prompts.yaml", "r") as stream:
     prompt_templates = yaml.safe_load(stream)
 
-# Build agent
+# Импорт примера инструмента с Hugging Face Hub
+image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+
 agent = CodeAgent(
     model=model,
-    tools=[
-        final_answer,
-        my_custom_tool,
-        get_current_time_in_timezone,
-        get_eur_exchange_rate,
-        get_simple_joke,
-        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,
     prompt_templates=prompt_templates,
 )
 
-# ---------------------------
-# Launch Gradio UI
-# ---------------------------
-GradioUI(agent).launch()
+# ---------------------- UI ----------------------
+
+# Исправлено: убран подсчет last_input_token_count (он может быть None)
+GradioUI(agent).launch(share=True)