Update app.py

app.py

@@ -7,6 +7,8 @@ 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 !
 @tool
 def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
@@ -34,6 +36,44 @@ def get_current_time_in_timezone(timezone: str) -> str:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
 
+
+
+##########################
+#   Restaurant Tools     #
+##########################
+
+@tool
+
+def summarize_top_pizza_places(search_results: str) -> str:
+    """
+    A tool that takes raw search results from DuckDuckGoSearchTool (in text form)
+    and attempts to figure out the top pizza place based on reviews or links.
+    
+    Because the search results come in as text, we can parse or heuristically
+    find the best pizza place. Here we'll do a simplistic approach. 
+    In a more advanced setup, you'd do real scraping or structured parsing.
+    Args:
+        search_results: The text returned by a search tool
+    Returns:
+        A short summary of the best pizza place or a guessed name.
+    """
+
+    lines = search_results.splitlines()
+    best_place = None
+
+    for line in lines:
+        if "best" in line.lower() or "top" in line.lower():
+            best_place = line
+            break
+
+    if best_place:
+        return f"I found a recommended pizza place: '{best_place}'"
+    else:
+        return f"I couldn't parse the top pizza place from these results. Maybe check manually?"
+
+
+
+
 final_answer = FinalAnswerTool()
 
 # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
@@ -50,12 +90,15 @@ custom_role_conversions=None,
 # Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
+# DuckDuckGo Search tool for retrieving web search results
+search_tool = DuckDuckGoSearchTool()
+
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[search_tool, summarize_top_pizza_places, image_generation_tool, final_answer]
     max_steps=6,
     verbosity_level=1,
     grammar=None,