Update app.py

app.py

@@ -8,21 +8,27 @@ from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
+python
+import requests
+from bs4 import BeautifulSoup
+
 @tool
-def fetch_top_restaurants(location: str): #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 fetches the top retraunts in a given location
-    Args:location (str): The location to search for restaurants.
-    """
-       Returns:
+def fetch_top_restaurants(location: str) -> list:
+    """A tool that fetches the top restaurants in a given location.
+
+    Args:
+        location (str): The location to search for restaurants.
+
+    Returns:
         list: A list of dictionaries containing restaurant names and links.
-    
-       # Construct the query for DuckDuckGo
+    """
+    # Construct the query for DuckDuckGo
     query = f"top 5 restaurants in {location}"
     url = f"https://duckduckgo.com/html/?q={query}"
     
     # Perform the search request
     response = requests.get(url)
+    
     # Parse the response if successful
     if response.status_code == 200:
         soup = BeautifulSoup(response.text, 'html.parser')
@@ -32,7 +38,7 @@ def fetch_top_restaurants(location: str): #it's import to specify the return typ
             link = result['href']
             results.append({'name': title, 'link': link})
         return results
-        else:
+    else:
         return f"Failed to retrieve search results for {location}"
 
 # Example usage - Fetching restaurants for a specific location
@@ -40,7 +46,7 @@ location = "New York"
 top_restaurants = fetch_top_restaurants(location)
 for i, restaurant in enumerate(top_restaurants, start=1):
     print(f"{i}. {restaurant['name']} - {restaurant['link']}")
-```
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.