Update app.py

app.py

@@ -7,34 +7,26 @@ from tools.final_answer import FinalAnswerTool
 import random
 from Gradio_UI import GradioUI
 
-from pydantic import BaseModel, Field
-import random
-
-class RandomNumberInput(BaseModel):
-    start_range: int = Field(..., description="The start number of the range")
-    end_range: int = Field(..., description="The end number of the range")
-
-@tool(args_schema=RandomNumberInput)
-def get_random_number_between_range(start_range: int, end_range: int) -> int:
+@tool
+def get_random_number_between_range(start_range: int = None, end_range: int = None) -> int:
     """A tool that generates a random number given a range.
-    
+
     Args:
         start_range: The start number of the range.
         end_range: The end number of the range.
     """
+    # Ask the user for missing inputs
+    if start_range is None:
+        return "Please provide the start number of the range."
+    if end_range is None:
+        return "Please provide the end number of the range."
+    
+    # Ensure valid range
+    if start_range >= end_range:
+        return "Error: The start number must be smaller than the end number."
+    
     return random.randint(start_range, end_range)
 
-# Below is an example of a tool that does nothing. Amaze us with your creativity !
-@tool
-def get_random_number_between_range(start_range:int, end_range:int)-> int: #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 generate a random number given a range
-    Args:
-        start_range: the start number of the range
-        end_range: the end number of the range
-    """
-    return random.randrange(start_range, end_range)
-
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.