Generalise the function and make some args optional

app.py

@@ -11,28 +11,40 @@ from Gradio_UI import GradioUI
 search_tool = DuckDuckGoSearchTool()
 
 @tool
-def find_open_restaurants(location: str, cuisine: str, rating: float, timezone: str)-> str: # it's important to specify the return type
-    # Keep this format for the tool description / args description but feel free to modify the tool
-    """A tool that does nothing yet 
+def find_restaurants_by_location(location: str, 
+                                cuisine: str = None, 
+                                rating: float = None, 
+                                timezone: str = None
+                               )-> str: 
+    """A tool that searches for restaurants according to location, cuisine, rating and local time. 
     Args:
         location: A string representing the location area to search for open restaurants.
-        cuisine: A string representing the cuisine type (e.g. Asian, Indian, Italian, breakfast, lunch, etc.)
-        rating: A float representing the average customer rating of the restaurant from Google, TripAdvisor, or Yelp.
-        timezone: A string representing a valid timezone (e.g., 'America/New_York').
+        cuisine: An optional string representing the cuisine type (e.g. Asian, Indian, Italian, breakfast, lunch, etc.)
+        rating: An optional float representing the average customer rating of the restaurant from Google, TripAdvisor, or Yelp.
+        timezone: An optional string representing a valid timezone (e.g., 'America/New_York').
     """
     try:
-        # Get the current local time
-        tz = pytz.timezone(timezone)
+       # If no timezone is provided, default to UTC
+        if timezone:
+            tz = pytz.timezone(timezone)
+        else:
+            tz = pytz.utc  # Default timezone
         local_time = datetime.datetime.now(tz).strftime("%H:%M")
 
-        # Search for restaurants of the specified cuisine that are open now
-        query = f"{cuisine} restaurants open now in {location} with rating equal to or greater than {rating}."
-        search_results = search_tool(query)  # Properly call the search tool
+        # Build query dynamically based on provided inputs
+        query_parts = [f"restaurants open now in {location}"]
+        if cuisine:
+            query_parts.append(f"serving {cuisine} cuisine")
+        if rating:
+            query_parts.append(f"with a rating of {rating} or higher")
+        
+        query = " ".join(query_parts)  # Construct final query
+        search_results = search_tool(query)  
 
         if not search_results:
-            return f"❌ No open {cuisine} restaurants found in {location} at {local_time}."
+            return f"❌ No matching restaurants found in {location} at {local_time}."
 
-        return f"🍽️ Open {cuisine} restaurants in {location} at {local_time}:\n\n" + "\n".join(
+        return f"🍽️ Open restaurants in {location} at {local_time}:\n\n" + "\n".join(
             [result["title"] + " - " + result["url"] for result in search_results[:5]]
         )