Upload 4 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,10 +1,7 @@
 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
@@ -85,4 +82,14 @@ def solver_captcha():
     return asyncio.run(hcaptcha_solver.bytedance(data["host"], data["site_key"], generate_uuid()))
 
 
+@app.route("/api/update", methods=["POST"])
+def update_model():
+    require_data = ["token"]
+    data = request.get_json(force=True, silent=True)
+    if not check_request(require_data, data):
+        return jsonify(msg="Unauthorized Request"), 403
+    hcaptcha_solver.solver.install(upgrade=True)
+    return jsonify(statue="ok"), 200
+
+
 app.run(host="0.0.0.0", port=7860)

get_all_model.py

@@ -0,0 +1,29 @@
+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)
+            models_path = models_dir + asset_name
+            if os.path.exists(models_path):
+                continue
+            print(f'Downloading {asset_name}')
+            r = session.get(asset_url, stream=True)
+            with open(models_path, 'wb') as f:
+                f.write(r.content)
+    print("Done!")

hcaptcha_solver.py

@@ -1,10 +1,12 @@
 from pathlib import Path
 import os
 from loguru import logger
+import get_all_model
 from playwright.async_api import BrowserContext as ASyncContext, async_playwright
 import hcaptcha_challenger as solver
 from hcaptcha_challenger.agents import AgentT, Malenia
 
+get_all_model.download_all()
 # Init local-side of the ModelHub
 solver.install(upgrade=True)