Update app.py

app.py

@@ -4,21 +4,24 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
+import random
 
 from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def roll_dice() -> str:
+def roll_dice(num_dice: int) -> str:
     """
-    Simulates rolling two dice and returns their values as a string.
+    Simulates rolling a specified number of dice and returns their values as a string.
+
+    Args:
+        num_dice (int): The number of dice to roll.
 
     Returns:
-        str: A message indicating the values of the two dice.
+        str: A message indicating the values of the rolled dice.
     """
-    die1 = random.randint(1, 6)
-    die2 = random.randint(1, 6)
-    return f"The values of the dice are {die1} and {die2}."
+    dice_values = [random.randint(1, 6) for _ in range(num_dice)]
+    return f"The values of the {num_dice} dice are: {', '.join(map(str, dice_values))}."
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: