diff --git a/wrp.go b/wrp.go index 475ef96..fe3298d 100644 --- a/wrp.go +++ b/wrp.go @@ -314,7 +314,6 @@ func (w wrpReq) capture() { var pngcap []byte chromedp.Run(ctx, emulation.SetDeviceMetricsOverride(int64(float64(w.width)/w.scale), 10, w.scale, false), - chromedp.Sleep(time.Second*2), chromedp.Location(&w.url), chromedp.ComputedStyle("body", &styles, chromedp.ByQuery), chromedp.ActionFunc(func(ctx context.Context) error { @@ -338,7 +337,10 @@ func (w wrpReq) capture() { } chromedp.Run(ctx, emulation.SetDeviceMetricsOverride(int64(float64(w.width)/w.scale), height, w.scale, false)) // Capture screenshot... - err = chromedp.Run(ctx, chromedp.CaptureScreenshot(&pngcap)) + err = chromedp.Run(ctx, + chromedp.Sleep(time.Second*2), + chromedp.CaptureScreenshot(&pngcap), + ) if err != nil { if err.Error() == "context canceled" { log.Printf("%s Contex cancelled, try again", w.req.RemoteAddr)