From 237c6879ddb33a4a7bce3ea1cd67322cbed064a7 Mon Sep 17 00:00:00 2001 From: Javier Luraschi Date: Mon, 16 Sep 2024 13:55:14 -0700 Subject: [PATCH] [apps/browser] desktop size and improved error handling --- apps/browser/app.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/apps/browser/app.py b/apps/browser/app.py index 91febedc..91b4f1ac 100644 --- a/apps/browser/app.py +++ b/apps/browser/app.py @@ -10,7 +10,7 @@ from siteuse import site_use async def take_screenshot(page): - await asyncio.sleep(3) + await asyncio.sleep(5) await page.screenshot({'path': "screenshot.png"}) shutil.copy("screenshot.png", f"storage/screenshot-{int(time.time())}.png") @@ -29,10 +29,11 @@ async def main(): custom_user_agent = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/117.0.0.0 Safari/537.36" browser = await pyppeteer.launch( headless=True, - args=['--no-sandbox', '--disable-setuid-sandbox', f"--window-size=1470,832"] + args=['--no-sandbox', '--disable-setuid-sandbox'] ) page = await browser.newPage() + await page.setViewport(viewport={'width': 1280, 'height': 800}) await page.setUserAgent(custom_user_agent) @@ -68,6 +69,13 @@ async def main(): prompt = h9.input() except Exception as e: print(f"Failed to use browser:\n```\n{e}\n```\n") + + try: + await take_screenshot(page) + elements = await extract_elements(page) + except Exception as e: + print("Failed to get elements from page") + prompt = h9.input(f"Last request failed, should I retry?") prompt = f"Failed to run the following code:\n\n{code}\n\nCode triggered the following error:\n\n{e}.\n\nAsked users to retry, user replied: " + prompt