Update app.py

app.py

@@ -19,7 +19,7 @@ def construct_course_search_query(interest: str, expertise: str, budget: str) ->
     return query
 
 # Existing tools from the template
-search_tool = DuckDuckGoSearchTool()
+search_tool = DuckDuckGoSearchTool(query)
 final_answer = FinalAnswerTool()
 
 # Model configuration (unchanged from template)
@@ -30,8 +30,6 @@ model = HfApiModel(
     custom_role_conversions=None,
 )
 
-# Import tool from Hub (unchanged from template)
-image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
 # Load prompt templates (unchanged from template, but updated in prompt.yaml)
 with open("prompts.yaml", 'r') as stream:
@@ -41,7 +39,7 @@ with open("prompts.yaml", 'r') as stream:
 agent = CodeAgent(
     model=model,
     tools=[construct_course_search_query, search_tool, final_answer],  # Added custom tool
-    max_steps=6,
+    max_steps=9,
     verbosity_level=1,
     grammar=None,
     planning_interval=None,