Update app.py

app.py

@@ -19,20 +19,10 @@ from Gradio_UI import GradioUI
   #  """
   #  return "What magic will you build ?"
 
-@tool
-def my_custom_tool(arg1: str, arg2: int) -> str:
-    """
-    A tool that repeats the input string in uppercase a given number of times.
-    
-    Args:
-        arg1: the string to repeat
-        arg2: the number of repetitions
-    """
-    repeated = (arg1.upper() + " ") * arg2
-    return repeated.strip()
 
+#  Tool to create a personalized greeting repeated a specific number of times.
 @tool
-def my_custom_tool2(arg1: str, arg2: int) -> str:
+def my_custom_tool(arg1: str, arg2: int) -> str:
     """
     A tool that creates a personalized greeting repeated multiple times.
     
@@ -44,7 +34,7 @@ def my_custom_tool2(arg1: str, arg2: int) -> str:
     return " ".join([greeting] * arg2)
 
 
-
+# Tool to get the current local time in your chosen timezone.
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -90,7 +80,7 @@ with open("prompts.yaml", 'r') as stream:
 agent = CodeAgent(
     model=model,
     #tools=[final_answer], ## add your tools here (don't remove final answer)
-    tools=[get_current_time_in_timezone, final_answer], ## add your tools here (don't remove final answer)
+    tools=[my_custom_tool, get_current_time_in_timezone, final_answer], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,