Update app.py

app.py

@@ -44,22 +44,44 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
 
 
 # Debugging: Print statements to track execution
+# async def capture_screenshot():
+#     print("Launching Playwright...")
+#     async with async_playwright() as p:
+#         browser = await p.chromium.launch(headless=True)  
+#         page = await browser.new_page()
+#         print("Opening the p5.js sketch...")
+#         await page.goto("https://editor.p5js.org/kfahn/full/2XD5Y8MiV", timeout=6000)
+
+#         print("Waiting for canvas element...")
+#         await page.wait_for_selector("canvas")  # Ensure p5.js has loaded
+#         await page.wait_for_timeout(5000)  # Additional wait
+
+#         print("Capturing screenshot...")
+#         await page.screenshot(path="img.png")
+#         await browser.close()
+#         print("Screenshot saved!")
+
 async def capture_screenshot():
+    """Launches Playwright and captures a screenshot of a p5.js sketch."""
     print("Launching Playwright...")
     async with async_playwright() as p:
-        browser = await p.chromium.launch(headless=True)  
+        browser = await p.chromium.launch(headless=True)
         page = await browser.new_page()
+        
         print("Opening the p5.js sketch...")
-        await page.goto("https://editor.p5js.org/kfahn/full/2XD5Y8MiV", timeout=6000)
+        await page.goto("https://editor.p5js.org/kfahn/full/2XD5Y8MiV", timeout=60000)  # Increased timeout
+
+        print("Waiting for page to load fully...")
+        await page.wait_for_load_state("networkidle")  # Ensures all requests finish
 
         print("Waiting for canvas element...")
-        await page.wait_for_selector("canvas")  # Ensure p5.js has loaded
-        await page.wait_for_timeout(5000)  # Additional wait
+        await page.wait_for_selector("canvas", timeout=60000)  # Increased timeout
 
         print("Capturing screenshot...")
         await page.screenshot(path="img.png")
         await browser.close()
         print("Screenshot saved!")
+
         
 @tool
 def grab_image() -> Image: