Upload 2 files

api.py

@@ -82,8 +82,10 @@ def solver_captcha():
         return jsonify(msg="Unauthorized Request"), 403
     dir_path=generate_uuid()
     resp=asyncio.run(hcaptcha_solver.bytedance(data["host"], data["site_key"], dir_path))
-    shutil.rmtree(dir_path)
-    shutil.rmtree("tmp_dir")
+    if os.path.exists(dir_path):
+        shutil.rmtree(dir_path)
+    if os.path.exists("tmp_dir"):
+        shutil.rmtree("tmp_dir")
     return resp
 
 

hcaptcha_solver.py

@@ -6,9 +6,8 @@ from playwright.async_api import BrowserContext as ASyncContext, async_playwrigh
 import hcaptcha_challenger as solver
 from hcaptcha_challenger.agents import AgentT, Malenia
 
-
 # Init local-side of the ModelHub
-solver.install(upgrade=True)
+solver.install(upgrade=True, clip=True)
 
 # Save dataset to current working directory
 tmp_dir = Path(__file__).parent.joinpath("tmp_dir")
@@ -16,11 +15,9 @@ tmp_dir = Path(__file__).parent.joinpath("tmp_dir")
 
 @logger.catch
 async def hit_challenge(context: ASyncContext, host, sitekey, user_data_dir, times: int = 8):
-    import hcaptcha_challenger as solver
-    from hcaptcha_challenger.agents import AgentT, Malenia
     await context.route('**/*', lambda route, request: route_continuation(route, request, host, sitekey))
     page = context.pages[0]
-    agent = AgentT.from_page(page=page, tmp_dir=tmp_dir,self_supervised=True)
+    agent = AgentT.from_page(page=page, tmp_dir=tmp_dir, self_supervised=True)
     await page.goto(f"https://{host}")
 
     await agent.handle_checkbox()
@@ -124,5 +121,3 @@ async def bytedance(host, sitekey, user_data_dirs):
         traceback.print_exc()
         print(e)
         return traceback.format_exc()
-
-