Update app.py

app.py

@@ -11,12 +11,25 @@ from Gradio_UI import GradioUI
 @tool
 def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
     #Keep this format for the description / args / args description but feel free to modify the tool
-    """A tool that does nothing yet 
+    """A tool that generates a personalized message based on the input arguments.
     Args:
-        arg1: the first argument
-        arg2: the second argument
+        arg1: The first argument, which is used as a name or identifier.
+        arg2: The second argument, which is used to determine the type of message.
     """
-    return "What magic will you build ?"
+    creative_responses = {
+        1: f"Hello, {arg1}! Today is a great day to chase your dreams!",
+        2: f"{arg1}, you are on the brink of something amazing! Keep pushing forward.",
+        3: f"Good news, {arg1}! Your hard work is about to pay off.",
+        4: f"{arg1}, you are one step closer to achieving your goals.",
+        5: f"Hi, {arg1}! Remember, every small step counts."
+    }
+    # Check if arg2 is within the range of our responses
+    if arg2 in creative_responses:
+        return creative_responses[arg2]
+    else:
+        # If arg2 is not in the range, generate a random creative message
+        random_message = random.choice(list(creative_responses.values()))
+        return f"{arg1}, here's a surprise message for you: {random_message}"
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: