Add random dice roll

app.py

@@ -3,6 +3,7 @@ import datetime
 import requests
 import pytz
 import yaml
+import random
 from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
@@ -25,10 +26,13 @@ def toss_a_die(k: int) -> int:
     Args:
         k: An integer specifying the number of sides on the die. If not specified, a good default would be to assume k=6 sides.
     """
-    if k==None: 
-        k=6
-    value = 3
-    return f"The random roll of a dice gives a value of {value}"
+    if k==None: k=6
+
+    try: 
+        value = random.randint(1, k)
+        return f"The random roll of a dice gives a value of {value}"
+    except Exception as e:
+        return f"How many sides does the dice have?"
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: