Spaces:
Running
Running
Update tools/interact_element.py
Browse files
tools/interact_element.py
CHANGED
@@ -1,6 +1,9 @@
|
|
1 |
from smolagents.tools import Tool
|
2 |
from helium import write, press, click, Text, Link, S
|
3 |
from selenium.webdriver.common.keys import Keys
|
|
|
|
|
|
|
4 |
|
5 |
class InteractElementTool(Tool):
|
6 |
name = "interact_element"
|
@@ -15,9 +18,14 @@ class InteractElementTool(Tool):
|
|
15 |
output_type = "string"
|
16 |
|
17 |
def __init__(self, driver):
|
|
|
18 |
self.driver = driver
|
|
|
|
|
19 |
|
20 |
def forward(self, selector=None, text=None, action="click", input_text=None, key=None):
|
|
|
|
|
21 |
try:
|
22 |
if selector:
|
23 |
element = self.driver.find_element(By.CSS_SELECTOR, selector)
|
|
|
1 |
from smolagents.tools import Tool
|
2 |
from helium import write, press, click, Text, Link, S
|
3 |
from selenium.webdriver.common.keys import Keys
|
4 |
+
import logging
|
5 |
+
|
6 |
+
logger = logging.getLogger(__name__)
|
7 |
|
8 |
class InteractElementTool(Tool):
|
9 |
name = "interact_element"
|
|
|
18 |
output_type = "string"
|
19 |
|
20 |
def __init__(self, driver):
|
21 |
+
super().__init__()
|
22 |
self.driver = driver
|
23 |
+
self.is_initialized = self.driver is not None # Check if driver is valid
|
24 |
+
logger.debug(f"InteractElementTool initialized: is_initialized={self.is_initialized}")
|
25 |
|
26 |
def forward(self, selector=None, text=None, action="click", input_text=None, key=None):
|
27 |
+
if not self.is_initialized:
|
28 |
+
return "Error: InteractElementTool is not initialized"
|
29 |
try:
|
30 |
if selector:
|
31 |
element = self.driver.find_element(By.CSS_SELECTOR, selector)
|