Update main.py

main.py

@@ -23,7 +23,8 @@ baseurl = os.environ.get("BASEURL", "https://duckduckgo.com")
 key = os.environ.get('KEY', 'sk-123456')
 # 使用 rstrip('/') 去掉尾部的斜杠
 baseurl = baseurl.rstrip('/')
-
+device_id = ''.join(random.choices(string.ascii_uppercase + string.digits, k=32))
+print(f'device_id: {device_id}')
 apiToken =''
 
 async def chat_chunk(body, headers):
@@ -486,25 +487,28 @@ def get_deviceId():
     # 生成YvcHTwwgEdYz56yQgbesjUTH4Ugm8MWT位随机数作为设备ID
         # 从所有大小写字母和数字中生成32位随机数
     characters = string.ascii_letters + string.digits
+    global device_id
     device_id = ''.join(random.choices(characters, k=32))
     print(f'{formatted_now()}: device_id: {device_id}')
     return device_id
 
 
 def get_token():
-    device_id = get_deviceId()
-    url = f'https://api2.immersivetranslate.com/big-model/get-token?deviceId={device_id}'
-    headers = {'Content-Type': 'application/json',
-               'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/123.0.0.0 Safari/537.36'}
-    response = httpx.get(url, headers=headers)
-    if response.status_code == 200:
-        global apiToken
-        apiToken = response.json().get('apiToken')
-        print(f'apiToken: {apiToken}')
-        return apiToken
-    else:
-        print(f'Error getting API token: {response.text}')
-        return None
+    global device_id
+    while True:
+        url = f'https://api2.immersivetranslate.com/big-model/get-token?deviceId={device_id}'
+        headers = {'Content-Type': 'application/json',
+                   'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/123.0.0.0 Safari/537.36'}
+        response = httpx.get(url, headers=headers)
+        if response.status_code == 200:
+            global apiToken
+            apiToken = response.json().get('apiToken')
+            print(f'apiToken: {apiToken}')
+            return apiToken
+        else:
+            print(f'{formatted_now()} Error getting API token: {response.text}')
+            device_id = get_deviceId()
+            continue
 
 
 def get_external_ip():