Update app.py

Added password gen tooling

app.py

@@ -1,6 +1,8 @@
 from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
 import datetime
 import requests
+import random
+import string
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
@@ -18,6 +20,32 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     """
     return "What magic will you build ?"
 
+# my test tool
+@tool
+def generate_password(length: int = 12, include_special: bool = True) -> str:
+    """A tool that generates a random password of specified length.
+    
+    Args:
+        length: The length of the password (default: 12)
+        include_special: Whether to include special characters (default: True)
+    """
+    # Define character sets
+    lowercase = string.ascii_lowercase
+    uppercase = string.ascii_uppercase
+    digits = string.digits
+    special = string.punctuation if include_special else ""
+    
+    # Create character pool
+    all_chars = lowercase + uppercase + digits + special
+
+    try:
+    # Generate password
+        password = ''.join(random.choice(all_chars) for _ in range(length))
+    
+        return f"Here is a randomly generated password of length {length}: {password}"
+    except Exception as e:
+        return f"Error generating password of length {length} and special characters {include_special}"
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -55,7 +83,9 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[final_answer,
+          get_current_time_in_timezone,
+          generate_password], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,