Update app.py

app.py

@@ -14,17 +14,16 @@ with open('movies.json') as f:
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def movie_info(arg1:str)-> str: #it's import to specify the return type
+def movie_info(movie_name:str)-> str: #it's import to specify the return type
     #Keep this format for the description / args / args description but feel free to modify the tool
     """A tool that does nothing yet 
     Args:
-        arg1: name of the movie
+        movie_name: name of the movie
     """
-    print ("Extracting movie info for --> ", arg1)
-    
+
     try:
         # read and create an object of information about a movie requested. 
-        filtered_movies = [movie for movie in movies if movie["Title"].lower() == arg1.lower()]
+        filtered_movies = [movie for movie in movies if movie["Title"].lower() == movie_name.lower()]
         if len(filtered_movies) > 0:
             output = ""
             for movie in filtered_movies:
@@ -37,7 +36,7 @@ def movie_info(arg1:str)-> str: #it's import to specify the return type
             output += "\n"
         else:
             output = "No information on that movie found!"
-        print ("output = ", output)
+        print (output)
         return output
     except Exception as e:
         return f"Error movies json and formatting."
@@ -75,7 +74,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, movie_info], ## add your tools here (don't remove final answer)
+    tools=[final_answer, ], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,