Feat: Second tool, tribble counter

app.py

@@ -9,6 +9,7 @@ import random
 from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
+# In this case, the user should input the name of the character to get the quote :)
 @tool
 def get_star_trek_quote(character: str)-> 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
@@ -18,7 +19,7 @@ def get_star_trek_quote(character: str)-> str: #it's import to specify the retur
         character: The name of the character whose quote is requested (e.g., "Seven of Nine", "Spock", "Picard", "Data").
 
     Returns: 
-        A quote from the specified Star Trek character.
+        A quote from the specified Star Trek character, or a message if the character is not found.
     """
     quotes = {
         "Seven of Nine": [
@@ -38,7 +39,29 @@ def get_star_trek_quote(character: str)-> str: #it's import to specify the retur
             "One is my name. The other is not."
         ]
     }
-    return quotes.get(character, ["Sorry, no quotes available for this character"])[0]
+
+    # If the character exists in the dictionary, return a quote
+    if character in quote:
+        return quotes[character][0] # Fetch the first character
+    else:
+        return f"Sorry, no quotes available for {character}. Please provide a valid Star Trek character name."
+
+# To invoke this tool, just type something like "How many tribbles are there? or "Check the Tribble infestation!" 
+@tool
+def tribbles_counter() -> str:
+    """
+    Simulates the number of Tribbles aboard a starship.
+
+    Returns:
+        A string with the count of Tribbles aboard.
+    """
+    tribbles_counter = random.randint(1,1000)
+    if tribbles_count > 500:
+        return f"There are {tribbles_count} Tribbles aboard! It's a *serious* infestation!"
+    elif tribbles_count > 200:
+        return f"There are {tribbles_count} Tribbles aboard. It's getting out of hand!"
+    else:
+        return f"There are {tribbles_count} Tribbles aboard. It's under control...for now."
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -77,7 +100,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_star_trek_quote], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_star_trek_quote, tribbles_counter], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,