Update app.py

app.py

@@ -88,8 +88,8 @@ async def capture_screenshot():
         await page.goto(image_url, timeout=120000)  # Wait for the image page to load
 
         print("Waiting for image element...")
-        await page.wait_for_selector("img", timeout=120000)  # Wait for the <img> to be visible
-
+       # await page.wait_for_selector("img", timeout=120000)  # Wait for the <img> to be visible
+        await page.wait_for_timeout(5000)  # Allow sketch to fully render
         print("Capturing screenshot...")
         await page.screenshot(path="img.png")
         await browser.close()