Update selenium.py

format change
richardyc · Aug 3, 2023 · b6e7488 · b6e7488
1 parent 396aae5
commit b6e7488
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/chromegpt/tools/selenium.py b/chromegpt/tools/selenium.py
@@ -19,7 +19,6 @@
 
 from chromegpt.tools.utils import (
     find_parent_element_text,
-    get_all_text_elements,
     prettify_text,
     truncate_string_from_last_occurrence,
 )
@@ -317,7 +316,9 @@ def scroll(self, direction: str) -> str:
         return self.describe_website()
 
     def _get_website_main_content(self, soup: BeautifulSoup) -> str:
-        texts = [element.get_text(separator=' ') for element in soup.find_all(string=True)]
+        texts = [
+            element.get_text(separator=' ') for element in soup.find_all(string=True)
+        ]
         pretty_texts = [prettify_text(text) for text in texts]
         if not pretty_texts:
             return ""
@@ -337,7 +338,11 @@ def _get_interactable_elements(self, soup: BeautifulSoup) -> str:
                 tag.name == "button"
                 or (tag.name == "div" and tag.get("role") == "button")
                 or tag.name == "a"
-                or (tag.name == "input" and tag.get("type") in ["checkbox", "submit", "button", "radio", "reset"])
+                or (
+                    tag.name == "input" 
+                    and tag.get("type") 
+                    in ["checkbox", "submit", "button", "radio", "reset"]
+                )
                 or tag.has_attr("data-href")
             )
         )