Update app.py

app.py

@@ -7,17 +7,15 @@ 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 get_wikipedia_movie_summary(movie_title:str) -> str:
-    """A tool that queries wikipedia for a movie title in English and returns a page's introduction.
+    """A tool that queries Wikipedia for a movie title in English and returns a page's introduction.
       Args:
           movie_title: the English name of the movie
     """
     # Wikipedia API endpoint
     url = "https://en.wikipedia.org/w/api.php"
     
-    # Parameters for the API request
     params = {
         "action": "query",
         "format": "json",
@@ -54,15 +52,11 @@ model = HfApiModel(
     custom_role_conversions=None,
 )
 
-
-# Import tool from Hub
-image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
-
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
     
 name = "Bootyful Bot"
-task = """
+task_description = """
 - Bootyful Bot will search wikipedia for a movie name
 - Bootyful Bot will read the page and tell whether the movie was inspired by factual events or not
 - In case it is, tell user about the events
@@ -73,14 +67,13 @@ task = """
 
 agent = CodeAgent(
     model=model,
-    tools=[get_wikipedia_movie_summary, final_answer], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_wikipedia_movie_summary], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,
     planning_interval=None,
     name=name,
-    description=None,
-    task=task,
+    description=task_description,
     prompt_templates=prompt_templates
 )