Update hcaptcha_solver.py

hcaptcha_solver.py

@@ -155,38 +155,42 @@ async def classify_click(page, data, round0, examples):
 
 
 async def area_click(page, data, round0, examples):
-    await page.wait_for_selector("//iframe[contains(@title, 'hCaptcha challenge')]")
-    frame_challenge = page.frame_locator("//iframe[contains(@title, 'hCaptcha challenge')]")
-    locator = frame_challenge.locator("//div[@class='challenge-view']//canvas")
-    await locator.wait_for(state="visible")
-    image = await locator.screenshot()
-    if examples:
-        clicks = nopecha.Recognition.solve(
-            type='hcaptcha_area_select',
-            task=data.get("requester_question").get("en"),
-            image_data=[url_to_base64(data.get("tasklist")[0]["datapoint_uri"])],
-            image_examples=examples
-        )
-    else:
-        clicks = nopecha.Recognition.solve(
-            type='hcaptcha_area_select',
-            task=data.get("requester_question").get("en"),
-            image_data=[url_to_base64(data.get("tasklist")[0]["datapoint_uri"])]
-        )
-    print(clicks)
-    print(clicks["x"], clicks["y"])
-    print("try to click")
-    bounds = await locator.bounding_box()
-    print(bounds)
-    await locator.click(delay=200, position={"x": int(bounds["width"] * clicks["x"] / 100),
-                                             "y": int(bounds["height"] * clicks["y"] / 100)})
-    print("done")
-
-    fl = frame_challenge.locator("//div[@class='button-submit button']")
-    await fl.click()
-    await asyncio.sleep(random.uniform(0.1, 0.3))
-    if round0 == 0:
-        await area_click(page, data, 1, examples)
+    try:
+        await page.wait_for_selector("//iframe[contains(@title, 'hCaptcha challenge')]")
+        frame_challenge = page.frame_locator("//iframe[contains(@title, 'hCaptcha challenge')]")
+        locator = frame_challenge.locator("//div[@class='challenge-view']//canvas")
+        await locator.wait_for(state="visible")
+        image = await locator.screenshot()
+        if examples:
+            clicks = nopecha.Recognition.solve(
+                type='hcaptcha_area_select',
+                task=data.get("requester_question").get("en"),
+                image_data=[url_to_base64(data.get("tasklist")[0]["datapoint_uri"])],
+                image_examples=examples
+            )
+        else:
+            clicks = nopecha.Recognition.solve(
+                type='hcaptcha_area_select',
+                task=data.get("requester_question").get("en"),
+                image_data=[url_to_base64(data.get("tasklist")[0]["datapoint_uri"])]
+            )
+        print(clicks)
+        print(clicks["x"], clicks["y"])
+        print("try to click")
+        bounds = await locator.bounding_box()
+        print(bounds)
+        await locator.click(delay=200, position={"x": int(bounds["width"] * clicks["x"] / 100),
+                                                 "y": int(bounds["height"] * clicks["y"] / 100)})
+        print("done")
+    
+        fl = frame_challenge.locator("//div[@class='button-submit button']")
+        await fl.click()
+        await asyncio.sleep(random.uniform(0.1, 0.3))
+        if round0 == 0:
+            await area_click(page, data, 1, examples)
+    except Exception as e:
+        print(e)
+        await area_click(page, data, round0, examples)
 
 
 async def hit_challenge(context: ASyncContext, host, sitekey, times: int = 8):