Update app.py

app.py

@@ -6,27 +6,7 @@ import yaml
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
-@tool
-def my_custom_tool(arg1:str, arg2:int)-> str:
-    """A tool that does nothing yet
-     Args:
-        arg1: the first argument
-        arg2: the second argument
-    """
-    return "What magic will you build ?"
-
-@tool
-def get_current_time_in_timezone(timezone: str) -> str:
-    """A tool that fetches the current local time in a specified timezone.
-    Args:
-        timezone: A string representing a valid timezone (e.g., 'America/New_York').
-    """
-    try:
-        tz = pytz.timezone(timezone)
-        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
-        return f"The current local time in {timezone} is: {local_time}"
-    except Exception as e:
-        return f"Error fetching time for timezone '{timezone}': {str(e)}"
+# ... (Your existing my_custom_tool and get_current_time_in_timezone definitions)
 
 final_answer = FinalAnswerTool()
 model = HfApiModel(
@@ -39,22 +19,17 @@ model = HfApiModel(
 # Initialize the DuckDuckGo Search Tool
 duckduckgo_search_tool = DuckDuckGoSearchTool()
 
-# Attempt to load the Wikipedia tool from the smolagents tool hub
-try:
-    wikipedia_tool = load_tool("wikipedia", trust_remote_code=True)
-except Exception as e:
-    print(f"Could not load Wikipedia tool. You might need to find the correct tool ID in the smolagents tool hub. Error: {e}")
-    wikipedia_tool = None
+# Removed the attempt to load the Wikipedia tool
+# The previous code to load the Wikipedia tool was here.
 
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
     tools_list = [final_answer, duckduckgo_search_tool]
-    if wikipedia_tool:
-        tools_list.append(wikipedia_tool)
+    # The conditional block to add the Wikipedia tool has been removed.
 
     agent = CodeAgent(
         model=model,
-        tools=tools_list,  # Add the DuckDuckGo Search Tool and Wikipedia Tool here
+        tools=tools_list,  # Only includes final_answer and duckduckgo_search_tool now
         max_steps=6,
         verbosity_level=1,
         grammar=None,