Update app.py

app.py

@@ -15,7 +15,7 @@ def construct_course_search_query(interest: str, expertise: str, budget: str) ->
         expertise: The user's current level of expertise (e.g., 'beginner', 'intermediate').
         budget: The budget available for the course (e.g., '$100', 'free').
     """
-    query = f"best {interest} courses for {expertise} under {budget}"
+    query = f"top {interest} courses for {expertise} under {budget}"
     return query
 
 # Existing tools from the template
@@ -33,11 +33,11 @@ model = HfApiModel(
 # 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)
+# Load prompt templates (unchanged from template, but updated in prompt.yaml)
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
 
-# Initialize the agent with the updated tools list
+# Initialize the agent with the updated tools list and custom logic
 agent = CodeAgent(
     model=model,
     tools=[construct_course_search_query, search_tool, final_answer],  # Added custom tool
@@ -47,7 +47,8 @@ agent = CodeAgent(
     planning_interval=None,
     name=None,
     description=None,
-    prompt_templates=prompt_templates
+    prompt_templates=prompt_templates,
+    custom_logic=lambda state: all(key in state for key in ['user_interest', 'user_expertise', 'user_budget'])
 )
 
 GradioUI(agent).launch()