Update app.py

app.py

@@ -1,12 +1,39 @@
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
+from smolagents import CodeAgent, HfApiModel, load_tool, tool  # Removed DuckDuckGoSearchTool
 import datetime
 import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
-from tools.recipe_tools import search_recipe, parse_recipe, generate_shopping_list, format_shopping_list
+from tools.web_search import DuckDuckGoSearchTool  # Added your custom web search
+from tools.visit_webpage import VisitWebPageTool  # Added visit webpage tool
+from tools.recipe_tools import (
+    search_recipe, 
+    parse_recipe, 
+    generate_shopping_list, 
+    format_shopping_list
+)
 from Gradio_UI import GradioUI
 
+# Initialize tools with dependencies
+def create_search_recipe_with_deps(web_search, visit_webpage):
+    def search_recipe_wrapped(query: str) -> List[Dict[str, str]]:
+        return search_recipe(query, web_search, visit_webpage)
+    return search_recipe_wrapped
+
+def create_parse_recipe_with_deps(visit_webpage):
+    def parse_recipe_wrapped(url: str) -> Dict:
+        return parse_recipe(url, visit_webpage)
+    return parse_recipe_wrapped
+
+# Create tool instances
+final_answer = FinalAnswerTool()
+web_search = DuckDuckGoSearchTool()
+visit_webpage = VisitWebPageTool()
+
+# Create wrapped recipe tools with dependencies
+search_recipe_tool = create_search_recipe_with_deps(web_search, visit_webpage)
+parse_recipe_tool = create_parse_recipe_with_deps(visit_webpage)
+
 # Initialize tools
 final_answer = FinalAnswerTool()
 
@@ -30,11 +57,13 @@ agent = CodeAgent(
     model=model,
     tools=[
         final_answer,
-        search_recipe,
-        parse_recipe,
+        final_answer,
+        web_search,
+        visit_webpage,
+        search_recipe_tool,
+        parse_recipe_tool,
         generate_shopping_list,
-        format_shopping_list,
-        image_generation_tool
+        format_shopping_list
     ],
     max_steps=6,
     verbosity_level=1,