Upload 2 files

api.py

@@ -1,7 +1,10 @@
+import hashlib
 import os
 import asyncio
 import time
 import multiprocessing
+import uuid
+
 from flask import Flask, jsonify, request, logging as flog
 from flask_limiter.util import get_remote_address
 import hcaptcha_solver
@@ -20,6 +23,12 @@ def get_ipaddr():
 handler = flog.default_handler
 
 
+def generate_uuid():
+    unique_identifier = str(uuid.uuid4())
+    hashed_string = hashlib.sha256(unique_identifier.encode()).hexdigest()
+    return hashed_string
+
+
 def get_token():
     default_token = "init_token"
     if os.path.exists("token"):
@@ -37,6 +46,7 @@ def check_request(required_data, data):
         return False
     return True
 
+
 @app.errorhandler(429)
 def rate_limit_exceeded(e):
     print(get_remote_address())
@@ -72,7 +82,7 @@ def solver_captcha():
     data = request.get_json(force=True, silent=True)
     if not check_request(require_data, data):
         return jsonify(msg="Unauthorized Request"), 403
-    return asyncio.run(hcaptcha_solver.bytedance(data["host"], data["site_key"]))
+    return asyncio.run(hcaptcha_solver.bytedance(data["host"], data["site_key"], generate_uuid()))
 
 
 app.run(host="0.0.0.0", port=7860)

hcaptcha_solver.py

@@ -9,7 +9,6 @@ from hcaptcha_challenger.agents import AgentT, Malenia
 solver.install(upgrade=True)
 
 # Save dataset to current working directory
-user_data_dir = Path(__file__).parent.joinpath("user_data_dir")
 context_dir = user_data_dir.joinpath("context")
 tmp_dir = Path(__file__).parent.joinpath("tmp_dir")
 
@@ -134,14 +133,14 @@ async def route_continuation(route, request, host, sitekey):
         await route.continue_()
 
 
-async def bytedance(host, sitekey):
+async def bytedance(host, sitekey, user_data_dirs):
     # playwright install firefox --with-deps
     async with async_playwright() as p:
         context = await p.firefox.launch_persistent_context(
-            user_data_dir=context_dir,
+            user_data_dir=Path(__file__).parent.joinpath(user_data_dirs),
             headless=True,
             locale="en-US"
         )
         await Malenia.apply_stealth(context)
         token = await hit_challenge(context, host, sitekey)
-        return token
+        return token