Update app.py

app.py

@@ -3,6 +3,7 @@ import datetime
 import pytz
 import yaml
 import uuid
+import urllib.parse
 
 from smolagents.agent_types import AgentImage
 from tools.final_answer import FinalAnswerTool
@@ -10,13 +11,16 @@ from Gradio_UI import GradioUI
 
 
 @tool
-def my_custom_tool(arg1: str, arg2: int) -> str:
-    """A tool that does nothing yet
+def wikipedia_url(query: str) -> str:
+    """A tool that provides the URL to the wikipedia article for a query.
     Args:
-        arg1: the first argument
-        arg2: the second argument
+        query: Topic to search on Wikipedia.
     """
-    return "What magic will you build ?"
+    try:
+        q = query.strip()
+        if not q:
+            return "Please provide a non-empty query."
+        return "https://en.wikipedia.org/wiki/Special:Search?search=" + urllib.parse.quote(q)
 
 
 @tool
@@ -41,21 +45,6 @@ model = HfApiModel(
     custom_role_conversions=None,
 )
 
-# Load the low-level image tool (keep this private)
-_image_generator = load_tool("agents-course/text-to-image", trust_remote_code=True)
-
-# Expose a UI-friendly wrapper tool that returns AgentImage
-@tool
-def generate_image(prompt: str) -> AgentImage:
-    """Generate an image from a text prompt and return it as an AgentImage for Gradio rendering.
-    Args:
-        prompt: The text description of the image to generate.
-    """
-    pil_img = _image_generator(prompt=prompt)  # PIL image
-    return AgentImage(pil_img)  # ✅ wrap the PIL object, not a filepath
-
-
-
 final_answer = FinalAnswerTool()
 
 with open("prompts.yaml", "r") as stream:
@@ -65,9 +54,8 @@ agent = CodeAgent(
     model=model,
     tools=[
         final_answer,
-        generate_image,                 # <-- this is what you want the agent to call
+        wikipedia_url,
         get_current_time_in_timezone,
-        my_custom_tool,
     ],
     max_steps=6,
     verbosity_level=1,