Update app.py

app.py

@@ -10,8 +10,12 @@ from Gradio_UI import GradioUI
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 
 
+import random
+from smolagents import tool
+
 @tool
 def generate_password(arg1: str, arg2: int) -> str:
+    # Keep this format for the tool description / args description but feel free to modify the tool
     """A tool that generates a random password based on the specified complexity.
 
     Args:
@@ -21,10 +25,15 @@ def generate_password(arg1: str, arg2: int) -> str:
     if arg2 < 4:
         return "Error: Password length must be at least 4 characters."
 
+    # Manually define character sets (since `string` is blocked)
+    letters = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
+    digits = "0123456789"
+    symbols = "!@#$%^&*()-_=+[]{}|;:,.<>?/"
+
     if arg1.lower() == "simple":
-        chars = string.ascii_letters + string.digits  # Letters and numbers
+        chars = letters + digits  # Simple passwords: letters + numbers
     elif arg1.lower() == "complex":
-        chars = string.ascii_letters + string.digits + string.punctuation  # Includes symbols
+        chars = letters + digits + symbols  # Complex passwords: add symbols
     else:
         return "Error: Use 'simple' or 'complex' for the complexity type."