Update hcaptcha_solver.py

hcaptcha_solver.py

@@ -124,30 +124,34 @@ async def on_response(response, page):
 
 
 async def classify_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')]")
-    samples = frame_challenge.locator("//div[@class='task-image']")
-    await frame_challenge.locator("//div[@tabindex='0']").nth(0).wait_for()
-    count = await samples.count()
-    print(count)
-    clicks = nopecha.Recognition.solve(
-        type='hcaptcha',
-        task=data.get("requester_question").get("en"),
-        image_urls=[data.get("tasklist")[i]["datapoint_uri"] for i in (range(9) if round0 == 0 else range(9, 18))],
-        image_examples=examples
-    )
-    for i in range(count):
-        sample = samples.nth(i)
-        await sample.wait_for()
-        if clicks[i]:
-            print("try to click")
-            await sample.click(delay=200)
-    print(clicks)
-    await asyncio.sleep(random.uniform(0.1, 0.3))
-    fl = frame_challenge.locator("//div[@class='button-submit button']")
-    await fl.click()
-    if round0 == 0:
-        await classify_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')]")
+        samples = frame_challenge.locator("//div[@class='task-image']")
+        await frame_challenge.locator("//div[@tabindex='0']").nth(0).wait_for()
+        count = await samples.count()
+        print(count)
+        clicks = nopecha.Recognition.solve(
+            type='hcaptcha',
+            task=data.get("requester_question").get("en"),
+            image_urls=[data.get("tasklist")[i]["datapoint_uri"] for i in (range(9) if round0 == 0 else range(9, 18))],
+            image_examples=examples
+        )
+        for i in range(count):
+            sample = samples.nth(i)
+            await sample.wait_for()
+            if clicks[i]:
+                print("try to click")
+                await sample.click(delay=200)
+        print(clicks)
+        await asyncio.sleep(random.uniform(0.1, 0.3))
+        fl = frame_challenge.locator("//div[@class='button-submit button']")
+        await fl.click()
+        if round0 == 0:
+            await classify_click(page, data, 1, examples)
+    except Exception as e:
+        print(e)
+        await classify_click(page, data, round0, examples)
 
 
 async def area_click(page, data, round0, examples):