Update app.py

app.py

@@ -77,6 +77,19 @@ class CachedWikiTool(WikipediaSearchTool):
     def run(self, page: str):
         return super().run(page)
 
+class PreloadedPythonTool(PythonInterpreterTool):
+    """
+    A PythonInterpreterTool that automatically prepends the necessary imports
+    (bs4, BeautifulSoup, regex) so you never hit NameError inside your code blocks.
+    """
+    def run(self, code: str) -> str:
+        preamble = (
+            "import bs4\n"
+            "from bs4 import BeautifulSoup\n"
+            "import regex\n"
+        )
+        return super().run(preamble + code)
+
 # --- Basic Agent Definition ---
 # ----- THIS IS WHERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
@@ -122,13 +135,17 @@ class BasicAgent:
         # Introduce tools
         wiki_tool = CachedWikiTool()
         search_tool = CachedWebSearchTool()
-        python_tool = PythonInterpreterTool()
+        python_tool = PreloadedPythonTool()
         html_parse_tool = VisitWebpageTool()
         # Initialize the agent
         self.agent = CodeAgent(model=self, 
                                tools=[wiki_tool, search_tool, python_tool, html_parse_tool], 
                                add_base_tools=True,
-                               additional_authorized_imports=["dateparser", "bs4", "regex"])
+                               additional_authorized_imports=["dateparser", "bs4", "regex"],
+                               system_message=("When writing Python code in the PythonInterpreterTool, you must "
+                                 "always import bs4 and regex (already preloaded), and *return* your "
+                                 "result by calling final_answer(...). Do not assign to a variable named final_answer.")
+         )
         
     def _serialize_messages(self, messages):
         prompt = []