change custom tool to nap time

app.py

@@ -9,21 +9,30 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def check_hours_minutes_sum(date_str: str) -> bool: #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 checks if the sum of hours and minutes in a datetime string equals 67. 
+def is_good_nap_time(datetime_str: str) -> bool:
+    """Checks whether the given time falls within a recommended nap window.
+    
+    Good nap windows are 12:00–14:30 (post-lunch) and 16:30–18:30 (afternoon).
+    
     Args:
-        date_str: DateTime string in format "%Y-%m-%d %H:%M:%S"
+        datetime_str: DateTime string in format "%Y-%m-%d %H:%M:%S"
+    
+    Returns:
+        True if the time is within a nap window, False otherwise.
     """
-    # Split the datetime string to extract time part
-    time_part = date_str.split(' ')[1]
+    try:
+        dt = datetime.strptime(datetime_str, "%Y-%m-%d %H:%M:%S")
+    except ValueError:
+        raise ValueError(f"Invalid datetime format: '{datetime_str}'. Expected '%Y-%m-%d %H:%M:%S'.")
+
+    t = dt.time()
     
-    # Split hours and minutes
-    hours, minutes, seconds = time_part.split(':')
+    window_1 = (datetime.strptime("12:00:00", "%H:%M:%S").time(),
+                datetime.strptime("14:30:00", "%H:%M:%S").time())
+    window_2 = (datetime.strptime("16:30:00", "%H:%M:%S").time(),
+                datetime.strptime("18:30:00", "%H:%M:%S").time())
     
-    # Convert to integers and check if sum equals 67
-    total = int(hours) + int(minutes)
-    return total == 67
+    return window_1[0] <= t <= window_1[1] or window_2[0] <= t <= window_2[1]
     
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: