From dcbbea0ce057841e1ec2d0067c36e981323a4ccd Mon Sep 17 00:00:00 2001 From: Saifeddine ALOUI Date: Tue, 19 Mar 2024 15:18:15 +0100 Subject: [PATCH] Update internet.py --- lollms/internet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lollms/internet.py b/lollms/internet.py index f00c89a..c1d16d9 100644 --- a/lollms/internet.py +++ b/lollms/internet.py @@ -90,7 +90,7 @@ def press_buttons(driver, buttons_to_press=['accept']): soup = BeautifulSoup(driver.page_source, 'html.parser') # Find the button that contains the text "accept" (case-insensitive) - for button_to_press in buttons_to_press.split(","): + for button_to_press in buttons_to_press.split(",") if type(buttons_to_press)==str else buttons_to_press: try: button_to_press = button_to_press.strip() button = soup.find('button', text=lambda t: button_to_press in t.lower())