Update app.py

app.py

@@ -400,12 +400,16 @@ class SeleniumVisionAgent(CodeAgent):
         # Initialize
         print(f"Browser size: {self.sandbox.width}x{self.sandbox.height}")
         os.makedirs(self.data_dir, exist_ok=True)
+        
+        # Build tools list
+        tools_list = self.build_tools()
 
         super().__init__(
-            tools=[],
+            tools=tools_list,
             model=model,
             max_steps=max_steps,
             verbosity_level=verbosity_level,
+            step_callbacks=[self.take_screenshot_callback],
             **kwargs,
         )
         
@@ -413,11 +417,9 @@ class SeleniumVisionAgent(CodeAgent):
             "<<resolution_x>>", str(self.sandbox.width)
         ).replace("<<resolution_y>>", str(self.sandbox.height))
 
-        self.register_tools()
-        self.step_callbacks.append(self.take_screenshot_callback)
-
-    def register_tools(self):
-
+    def build_tools(self):
+        """Define and return the list of tools for this agent"""
+        
         @tool
         def click(x: int, y: int) -> str:
             """
@@ -534,16 +536,7 @@ class SeleniumVisionAgent(CodeAgent):
             self.sandbox.driver.back()
             return "Went back one page"
 
-        self.tools["click"] = click
-        self.tools["right_click"] = right_click
-        self.tools["double_click"] = double_click
-        self.tools["type_text"] = type_text
-        self.tools["press_key"] = press_key
-        self.tools["drag_and_drop"] = drag_and_drop
-        self.tools["scroll"] = scroll
-        self.tools["wait"] = wait
-        self.tools["open_url"] = open_url
-        self.tools["go_back"] = go_back
+        return [click, right_click, double_click, type_text, press_key, drag_and_drop, scroll, wait, open_url, go_back]
 
 
     def take_screenshot_callback(self, memory_step: ActionStep, agent=None) -> None: