Update app.py

app.py

@@ -5,6 +5,9 @@ import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
 
+# import random
+import random
+
 from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
@@ -18,6 +21,46 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     """
     return "What magic will you build ?"
 
+@tool
+def generate_cat_name(personality:str)-> str:
+    """
+    A tool that generates a cat name based on personality trait. 
+    Args:
+        personality: A personality and or a trait description
+    """
+
+    personality = personality.lower().strip()
+
+    cat_name_options = {
+        "playful": ["Whiskers", "Jinx"],
+        "lazy": ["Napper", "Dozer", "Snoozy", "Pillow", "Flop"],
+        "mysterious": ["Shadow", "Midnight"],
+        "elegant": ["Duchess", "Baron"],
+        "shy": ["Whisper", "Hush"],
+        "adventurous": ["Scout", "Rover"],
+        "grumpy": ["Grucho", "Cranky"],
+    }
+
+    
+    # If personality is not found, then this could be the default names
+    default_names = ["Leo", "Luna", "Simba"]
+
+    # Find the most similar category if exact match not found
+    matching_category = None
+
+    for category in cat_name_options:
+        if category in personality or personality in category:
+            matching_category = category
+            break 
+
+    if matching_category:
+        name = random.choice(cat_name_options[matching_category])
+        return f"Based on the '{matching_category}' personality, I recommend the cat name {name}"
+    else:
+        name = random.choice(default_names)
+        return f"Here's a classic name for your cat: {name}"
+    
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -55,7 +98,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=[generate_cat_name, final_answer], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,