Upload 3 files

Dockerfile

@@ -40,6 +40,7 @@ RUN playwright install firefox --with-deps
 USER foxer
 RUN playwright install firefox
 COPY hcaptcha_solver.py .
+COPY get_all_model.py .
 COPY api.py .
 EXPOSE 7860
 

api.py

@@ -1,8 +1,11 @@
 import os
 import asyncio
+import time
+import multiprocessing
 from flask import Flask, jsonify, request, logging as flog
 from flask_limiter.util import get_remote_address
 import hcaptcha_solver
+import get_all_model
 
 app = Flask(__name__)
 
@@ -36,6 +39,12 @@ def check_request(required_data, data):
     return True
 
 
+def update_models():
+    while True:
+        get_all_model.download_all()
+        time.sleep(30 * 60)
+
+
 @app.errorhandler(429)
 def rate_limit_exceeded(e):
     print(get_remote_address())
@@ -73,5 +82,6 @@ def solver_captcha():
         return jsonify(msg="Unauthorized Request"), 403
     return asyncio.run(hcaptcha_solver.bytedance(data["host"], data["site_key"]))
 
-
-app.run(host="0.0.0.0", port=7860)
+process = multiprocessing.Process(target=update_models)
+process.start()
+app.run(host="0.0.0.0", port=7860)

get_all_model.py

@@ -0,0 +1,28 @@
+import os
+from pathlib import Path
+import requests
+import re
+
+username = "QIN2DIM"
+repo = "hcaptcha-challenger"
+
+url = f"https://api.github.com/repos/{username}/{repo}/releases"
+models_dir = "/usr/local/lib/python3.11/lib/site-packages/hcaptcha_challenger/onnx/models/"
+
+
+def download_all():
+    session = requests.session()
+    response = session.get(url)
+    data = response.json()
+    os.makedirs(models_dir, exist_ok=True)
+    for i in data:
+        for j in i['assets']:
+            asset_url = j['browser_download_url']
+            asset_name = re.sub('.*/', '', asset_url)
+            print(f'Downloading {asset_name}')
+            models_path = models_dir + asset_name
+            if os.path.exists(models_path):
+                continue
+            r = session.get(asset_url, stream=True)
+            with open(models_path, 'wb') as f:
+                f.write(r.content)