ai_browser/test_tools.py at main · Novaturiente/ai_browser · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
import os
from selenium import webdriver
from selenium.webdriver.chrome.service import Service
from selenium.webdriver.chrome.options import Options
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from bs4 import BeautifulSoup

script_dir = os.path.dirname(os.path.abspath(__file__))
chromedriver_path = os.path.join(script_dir, "chromedriver")
chrome_options = Options()
chrome_options.binary_location = "/nix/var/nix/profiles/default/bin/chromium"
service = Service(executable_path=chromedriver_path)
driver = webdriver.Chrome(service=service, options=chrome_options)

global latest_source
latest_source = ""

def browser_webpage(url: str) -> str:
    """Open a url and get contents of the page"""
    print(f"Opening {url}")
    global latest_source
    driver.get(url)
    link = driver.current_url
    html_data = driver.page_source
    soup = BeautifulSoup(html_data, "lxml")
    latest_source = f"url: {link}\n\nSource: {soup}"

    return f"opened {url} and updated Latest source"


def input_text(locator: str, element: str, text: str) -> str:
    """Input text to a field in the webpage"""
    print("Input")
    wait = WebDriverWait(driver, 10)
    input_field = wait.until(EC.presence_of_element_located((By.XPATH, element)))
    input_field.send_keys(text)

    return f"filled {element} with {text}"


def click_on_element(locator: str, element: str) -> str:
    """Click on the given element in the webpage"""
    print("click")
    wait = WebDriverWait(driver, 10)

    element_to_click = wait.until(EC.presence_of_element_located((By.XPATH, element)))
    element_to_click.click()

    return f"clicked on {element}"


def get_current_status() -> str:
    """Get current source of the webpage"""
    print("update")
    global latest_source
    link = driver.current_url
    html_data = driver.page_source
    soup = BeautifulSoup(html_data, "lxml")
    latest_source = f"url: {link}\n\nSource: {soup}"

    return "Updated Latest source"


tool_list = [
    {
        "name": "browser_webpage",
        "description": "Open a webpage and get contents of the page as source data",
        "parameters": {
            "type": "object",
            "properties": {
                "url": {
                    "type": "string",
                    "description": "url for the webpage to open",
                },
            },
            "required": ["url"],
        },
    },
    {
        "name": "input_text",
        "description": "Enter text an input field identified by the XPATH on the webpage",
        "parameters": {
            "type": "object",
            "properties": {
                "locator": {
                    "type": "string",
                    "description": "locator to select the element by",
                },
                "element": {
                    "type": "string",
                    "description": "Selector for the input field element",
                },
                "text": {
                    "type": "string",
                    "description": "text to enter in the field",
                },
            },
            "required": ["locator","element", "text"],
        },
    },
    {
        "name": "click_on_element",
        "description": "Click on an element identified by XPATH in the webpage",
        "parameters": {
            "type": "object",
            "properties": {
                "locator": {
                    "type": "string",
                    "description": "locator to select the element by",
                },
                "element": {
                    "type": "string",
                    "description": "XPATH for the element to click",
                },
            },
            "required": ["locator","element"],
        },
    },
    {
        "name": "get_current_status",
        "description": "Get source of the webpage to see the updated status of the page",
    },
]


tool_map = {
    "browser_webpage": browser_webpage,
    "input_text": input_text,
    "click_on_element": click_on_element,
    "get_current_status": get_current_status,
}