From b1dafc0ef79bdd94f69c783877217d8a5524d460 Mon Sep 17 00:00:00 2001
From: Heiner Lohaus <hlohaus@users.noreply.github.com>
Date: Sat, 18 May 2024 07:37:37 +0200
Subject: Improve Liabots provider, Add image api support

---
 g4f/Provider/openai/proofofwork.py | 48 +++++++++++++++++++++++++-------------
 1 file changed, 32 insertions(+), 16 deletions(-)

(limited to 'g4f/Provider/openai/proofofwork.py')

diff --git a/g4f/Provider/openai/proofofwork.py b/g4f/Provider/openai/proofofwork.py
index 51d96bc4..cbce153f 100644
--- a/g4f/Provider/openai/proofofwork.py
+++ b/g4f/Provider/openai/proofofwork.py
@@ -2,35 +2,51 @@ import random
 import hashlib
 import json
 import base64
-from datetime import datetime, timedelta, timezone
+from datetime import datetime, timezone
 
-def generate_proof_token(required: bool, seed: str, difficulty: str, user_agent: str):
+proof_token_cache: dict = {}
+
+def generate_proof_token(required: bool, seed: str = None, difficulty: str = None, user_agent: str = None, proofTokens: list = None):
     if not required:
         return
-
-    cores = [8, 12, 16, 24]
-    screens = [3000, 4000, 6000]
-
-    core = random.choice(cores)
-    screen = random.choice(screens)
+    if seed is not None and seed in proof_token_cache:
+        return proof_token_cache[seed]
 
     # Get current UTC time
     now_utc = datetime.now(timezone.utc)
     parse_time = now_utc.strftime('%a, %d %b %Y %H:%M:%S GMT')
 
-    config = [core + screen, parse_time, None, 0, user_agent, "https://tcr9i.chat.openai.com/v2/35536E1E-65B4-4D96-9D97-6ADB7EFF8147/api.js","dpl=53d243de46ff04dadd88d293f088c2dd728f126f","en","en-US",442,"plugins−[object PluginArray]","","alert"]
-
-    diff_len = len(difficulty) // 2
-
+    if proofTokens:
+        config = random.choice(proofTokens)
+    else:
+        screen = random.choice([3008, 4010, 6000]) * random.choice([1, 2, 4])
+        config = [
+            screen, parse_time,
+            None, 0, user_agent,
+            "https://tcr9i.chat.openai.com/v2/35536E1E-65B4-4D96-9D97-6ADB7EFF8147/api.js",
+            "dpl=1440a687921de39ff5ee56b92807faaadce73f13","en","en-US",
+            None,
+            "plugins−[object PluginArray]",
+            random.choice(["_reactListeningcfilawjnerp", "_reactListening9ne2dfo1i47", "_reactListening410nzwhan2a"]),
+            random.choice(["alert", "ontransitionend", "onprogress"])
+        ]
+
+    config[1] = parse_time
+    config[4] = user_agent
+    config[7] = random.randint(101, 2100)
+
+    diff_len = None if difficulty is None else len(difficulty)
     for i in range(100000):
         config[3] = i
         json_data = json.dumps(config)
         base = base64.b64encode(json_data.encode()).decode()
-        hash_value = hashlib.sha3_512((seed + base).encode()).digest()
+        hash_value = hashlib.sha3_512((seed or "" + base).encode()).digest()
 
-        if hash_value.hex()[:diff_len] <= difficulty:
-            result = "gAAAAAB" + base
-            return result
+        if difficulty is None or hash_value.hex()[:diff_len] <= difficulty:
+            if seed is None:
+                return "gAAAAAC" + base
+            proof_token_cache[seed] = "gAAAAAB" + base
+            return proof_token_cache[seed]
 
     fallback_base = base64.b64encode(f'"{seed}"'.encode()).decode()
     return "gAAAAABwQ8Lk5FbGpA2NcR9dShT6gYjU7VxZ4D" + fallback_base
-- 
cgit v1.2.3