Update api/validate.py

api/validate.py

@@ -1,14 +1,26 @@
-# api/validate.py
-
 import re
 import time
 import asyncio
-import httpx
-from typing import Optional  # Added import for Optional
+import aiohttp
+from typing import Optional
 
 base_url = "https://www.blackbox.ai"
 headers = {
-    "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36"
+    'accept': '*/*',
+    'accept-language': 'en-US,en;q=0.9',
+    'cache-control': 'no-cache',
+    'content-type': 'application/json',
+    'origin': base_url,
+    'pragma': 'no-cache',
+    'priority': 'u=1, i',
+    'referer': f'{base_url}/',
+    'sec-ch-ua': '"Not?A_Brand";v="99", "Chromium";v="130"',
+    'sec-ch-ua-mobile': '?0',
+    'sec-ch-ua-platform': '"Linux"',
+    'sec-fetch-dest': 'empty',
+    'sec-fetch-mode': 'cors',
+    'sec-fetch-site': 'same-origin',
+    'user-agent': 'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/130.0.0.0 Safari/537.36'
 }
 
 # Cache variables
@@ -26,45 +38,32 @@ async def getHid(force_refresh: bool = False) -> Optional[str]:
         return cached_hid
 
     try:
-        async with httpx.AsyncClient() as client:
-            # Retrieve the initial HTML content.
-            response = await client.get(base_url, headers=headers)
-            response.raise_for_status()
-            content = response.text
-
-            # Use a regular expression to find specific `static/chunks` paths.
-            pattern = r"static/chunks/app/layout-[a-zA-Z0-9]+\.js"
-            match = re.search(pattern, content)
-
-            if match:
-                # Construct the full URL of the JS file.
-                js_path = match.group()
-                full_url = f"{base_url}/_next/{js_path}"
+        async with aiohttp.ClientSession(headers=headers) as session:
+            async with session.get(base_url) as response:
+                if response.status != 200:
+                    print("Failed to load the page.")
+                    return None
 
-                # Request the content of the JS file.
-                js_response = await client.get(full_url, headers=headers)
-                js_response.raise_for_status()
+                page_content = await response.text()
+                js_files = re.findall(r'static/chunks/\d{4}-[a-fA-F0-9]+\.js', page_content)
 
-                # In JS content, use a regular expression to search for h-value.
-                h_pattern = r'h="([0-9a-f-]+)"'
-                h_match = re.search(h_pattern, js_response.text)
+                key_pattern = re.compile(r'w="([0-9a-fA-F]{8}-[0-9a-fA-F]{4}-[0-9a-fA-F]{4}-[0-9a-fA-F]{4}-[0-9a-fA-F]{12})"')
 
-                if h_match:
-                    h_value = h_match.group(1)
-                    print("Found the h-value:", h_value)
-                    # Update the cache.
-                    cached_hid = h_value
-                    cache_time = current_time
-                    return h_value
-                else:
-                    print("The h-value was not found in the JS content.")
-                    return None
-            else:
-                print("The specified JS file path was not found in the HTML content.")
+                for js_file in js_files:
+                    js_url = f"{base_url}/_next/{js_file}"
+                    async with session.get(js_url) as js_response:
+                        if js_response.status == 200:
+                            js_content = await js_response.text()
+                            match = key_pattern.search(js_content)
+                            if match:
+                                h_value = match.group(1)
+                                print("Found the h-value:", h_value)
+                                # Update the cache
+                                cached_hid = h_value
+                                cache_time = current_time
+                                return h_value
+                print("The h-value was not found in any JS content.")
                 return None
-    except httpx.RequestError as e:
+    except Exception as e:
         print(f"An error occurred during the request: {e}")
         return None
-    except httpx.HTTPStatusError as e:
-        print(f"HTTP error occurred: {e}")
-        return None