Update tools/scrape_text.py

tools/scrape_text.py

@@ -2,6 +2,9 @@ from smolagents.tools import Tool
 from helium import S
 from selenium.webdriver.common.by import By
 import json
+import logging
+
+logger = logging.getLogger(__name__)
 
 class ScrapeTextTool(Tool):
     name = "scrape_text"
@@ -13,9 +16,14 @@ class ScrapeTextTool(Tool):
     output_type = "string"
 
     def __init__(self, driver):
+        super().__init__()
         self.driver = driver
+        self.is_initialized = self.driver is not None
+        logger.debug(f"ScrapeTextTool initialized: is_initialized={self.is_initialized}")
 
     def forward(self, selector="p", extract_table=False):
+        if not self.is_initialized:
+            return "Error: ScrapeTextTool is not initialized"
         try:
             if extract_table:
                 tables = self.driver.find_elements(By.CSS_SELECTOR, selector)