Update app.py

app.py

@@ -11,37 +11,26 @@ from Gradio_UI import GradioUI
 
 
 @tool
-def convert_currency(amount: float, from_currency: str, to_currency: str) -> str:
-    """Converts an amount from one currency to another using real-time exchange rates.
-    
+def generate_password(arg1: str, arg2: int) -> str:
+    """A tool that generates a random password based on the specified complexity.
+
     Args:
-        amount (float): The amount to convert.
-        from_currency (str): The 3-letter currency code to convert from (e.g., 'USD').
-        to_currency (str): The 3-letter currency code to convert to (e.g., 'EUR').
-    
-    Returns:
-        str: A formatted string with the converted amount and exchange rate.
+        arg1: The complexity type ('simple' or 'complex').
+        arg2: The length of the password.
     """
-    try:
-        # Fetch exchange rate data
-        api_url = f"https://api.exchangerate-api.com/v4/latest/{from_currency.upper()}"
-        response = requests.get(api_url)
-        data = response.json()
-
-        if "rates" not in data:
-            return "Error: Could not retrieve exchange rates."
-
-        # Convert the amount
-        rate = data["rates"].get(to_currency.upper())
-        if not rate:
-            return f"Error: Unsupported currency '{to_currency}'."
-
-        converted_amount = round(amount * rate, 2)
-        return f"{amount} {from_currency.upper()} is approximately {converted_amount} {to_currency.upper()} (Rate: {rate})."
+    if arg2 < 4:
+        return "Error: Password length must be at least 4 characters."
+
+    if arg1.lower() == "simple":
+        chars = string.ascii_letters + string.digits  # Letters and numbers
+    elif arg1.lower() == "complex":
+        chars = string.ascii_letters + string.digits + string.punctuation  # Includes symbols
+    else:
+        return "Error: Use 'simple' or 'complex' for the complexity type."
+
+    password = "".join(random.choice(chars) for _ in range(arg2))
+    return f"Generated {arg1.lower()} password: {password}"
     
-    except Exception as e:
-        return f"Error fetching exchange rates: {str(e)}"
-
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -79,7 +68,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, convert_currency], ## add your tools here (don't remove final answer)
+    tools=[final_answer, generate_password], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,