Spaces:
Runtime error
Runtime error
File size: 2,055 Bytes
670dd87 |
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 |
from typing import Literal
from pydantic import Field
from agency_swarm.tools import BaseTool
from .util.selenium import get_web_driver, set_web_driver
class Scroll(BaseTool):
"""
This tool allows you to scroll the current web page up or down by 1 screen height.
"""
direction: Literal["up", "down"] = Field(..., description="Direction to scroll.")
def run(self):
wd = get_web_driver()
height = wd.get_window_size()["height"]
# Get the zoom level
zoom_level = wd.execute_script("return document.body.style.zoom || '1';")
zoom_level = (
float(zoom_level.strip("%")) / 100
if "%" in zoom_level
else float(zoom_level)
)
# Adjust height by zoom level
adjusted_height = height / zoom_level
current_scroll_position = wd.execute_script("return window.pageYOffset;")
total_scroll_height = wd.execute_script("return document.body.scrollHeight;")
result = ""
if self.direction == "up":
if current_scroll_position == 0:
# Reached the top of the page
result = "Reached the top of the page. Cannot scroll up any further.\n"
else:
wd.execute_script(f"window.scrollBy(0, -{adjusted_height});")
result = "Scrolled up by 1 screen height. Make sure to output '[send screenshot]' command to analyze the page after scrolling."
elif self.direction == "down":
if current_scroll_position + adjusted_height >= total_scroll_height:
# Reached the bottom of the page
result = (
"Reached the bottom of the page. Cannot scroll down any further.\n"
)
else:
wd.execute_script(f"window.scrollBy(0, {adjusted_height});")
result = "Scrolled down by 1 screen height. Make sure to output '[send screenshot]' command to analyze the page after scrolling."
set_web_driver(wd)
return result
|