Update app.py

app.py

@@ -104,6 +104,8 @@ CONFIG = {
         "PROXY": os.environ.get("PROXY") or None
     },
     "SERVER": {
+        "COOKIE": None,
+        "CF_CLEARANCE":os.environ.get("CF_CLEARANCE") or None,
         "PORT": int(os.environ.get("PORT", 5200))
     },
     "RETRY": {
@@ -443,6 +445,7 @@ class Utils:
 
         if proxy:
             logger.info(f"使用代理: {proxy}", "Server")
+            
             if proxy.startswith("socks5://"):
                 proxy_options["proxy"] = proxy
             
@@ -510,7 +513,7 @@ class GrokApiClient:
                 url,
                 headers={
                     **DEFAULT_HEADERS,
-                    "Cookie": CONFIG["API"]["SIGNATURE_COOKIE"]
+                    "Cookie":CONFIG["SERVER"]['COOKIE']
                 },
                 json=upload_data,
                 impersonate="chrome133a",
@@ -724,7 +727,7 @@ def handle_image_response(image_url):
                 f"https://assets.grok.com/{image_url}",
                 headers={
                     **DEFAULT_HEADERS,
-                    "Cookie": CONFIG["API"]["SIGNATURE_COOKIE"]
+                    "Cookie":CONFIG["SERVER"]['COOKIE']
                 },
                 impersonate="chrome120",
                 **proxy_options
@@ -974,6 +977,7 @@ def get_models():
 
 @app.route('/v1/chat/completions', methods=['POST'])
 def chat_completions():
+    response_status_code = 500
     try:
         auth_token = request.headers.get('Authorization',
                                          '').replace('Bearer ', '')
@@ -993,35 +997,36 @@ def chat_completions():
         retry_count = 0
         grok_client = GrokApiClient(model)
         request_payload = grok_client.prepare_chat_request(data)
-        response_status_code = 500
+        
 
         while retry_count < CONFIG["RETRY"]["MAX_ATTEMPTS"]:
             retry_count += 1
-            CONFIG["API"]["SIGNATURE_COOKIE"] = Utils.create_auth_headers(
-                model)
+            CONFIG["API"]["SIGNATURE_COOKIE"] = Utils.create_auth_headers(model)
 
             if not CONFIG["API"]["SIGNATURE_COOKIE"]:
                 raise ValueError('该模型无可用令牌')
 
             logger.info(
-                f"当前令牌: {json.dumps(CONFIG['API']['SIGNATURE_COOKIE'], indent=2)}",
-                "Server")
+                f"当前令牌: {json.dumps(CONFIG['API']['SIGNATURE_COOKIE'], indent=2)}","Server")
             logger.info(
-                f"当前可用模型的全部可用数量: {json.dumps(token_manager.get_remaining_token_request_capacity(), indent=2)}",
-                "Server")
-
+                f"当前可用模型的全部可用数量: {json.dumps(token_manager.get_remaining_token_request_capacity(), indent=2)}","Server")
+            if CONFIG['SERVER']['CF_CLEARANCE']:
+                CONFIG["SERVER"]['COOKIE'] = f"{CONFIG['API']['SIGNATURE_COOKIE']};cf_clearance={CONFIG['SERVER']['CF_CLEARANCE']}" 
+            else:
+                CONFIG["SERVER"]['COOKIE'] = CONFIG['API']['SIGNATURE_COOKIE']
             try:
                 proxy_options = Utils.get_proxy_options()
                 response = curl_requests.post(
                     f"{CONFIG['API']['BASE_URL']}/rest/app-chat/conversations/new",
                     headers={
                         **DEFAULT_HEADERS, 
-                        "Cookie":f"{CONFIG['API']['SIGNATURE_COOKIE']}; cf_clearance=ndxDqSPPQQQicT0vsP6JSSudjOUBQ_G7fufmUuOnsm4-1741270561-1.2.1.1-dlueDEYSJbFuyAUcbUdn0xFpXSXqRppz8euI.sQhX0ZQyKgjLQBm2kL32e6.FpA8pgcPyLxfXYYkHJg_.5iS4C3T7lwDJDNjE7AIEAXOYxyhas6jvb71l7FBPpnDnCF97kVPxlAOLMHMUZegxCm7Ln8EGftiMb9qCDSVxyqO14eEVQyRJK6ziLnEdg3Vk9_aAaf6E3Dbk6nM7X1eRa2SBGehoLy8FNR7Jw.Uz2TA1VxRIPoR8tyvcl19PxEtPwr9snvquFuGLUfF8QxrOU.bi2JdSOYVD9P28oaL4xy_OgN9EL.CY2R4TARzlcho7vJUA4HHRcWVDB0X3OM96ZRCvocqXqf7_r3GZFzypNTphkLK3TvNC6NGH2HbDMlJ_BHhlBFL2Hy3ap5h4rGXfpmRnfo01XDoQCGq9enjYw6QQujNzNc.2KN7u5oNyUyUdRp9rHPhGDY.BNcFOKV33bFG2sMwmNarEEUQ2Z5hx3nIkfI;"
+                        "Cookie":CONFIG["SERVER"]['COOKIE']
                     },
                     data=json.dumps(request_payload),
                     impersonate="chrome133a",
                     stream=True,
                     **proxy_options)
+                print(CONFIG["SERVER"]['COOKIE'])
                 if response.status_code == 200:
                     response_status_code = 200
                     logger.info("请求成功", "Server")
@@ -1055,7 +1060,7 @@ def chat_completions():
                     token_manager.reduce_token_request_count(model,1)#重置去除当前因为错误未成功请求的次数，确保不会因为错误未成功请求的次数导致次数上限
                     if token_manager.get_token_count_for_model(model) == 0:
                         raise ValueError(f"{model} 次数已达上限，请切换其他模型或者重新对话")
-                    raise ValueError(f"IP暂时被封黑无法破盾，请稍后重试或者更换ip")
+                    raise ValueError(f"IP暂时被封无法破盾，请稍后重试或者更换ip")
                 elif response.status_code == 429:
                     response_status_code = 429
                     token_manager.reduce_token_request_count(model,1)
@@ -1085,7 +1090,7 @@ def chat_completions():
                     raise
                 continue
         if response_status_code == 403:
-            raise ValueError('IP暂时被封黑无法破盾，请稍后重试或者更换ip')
+            raise ValueError('IP暂时被封无法破盾，请稍后重试或者更换ip')
         elif response_status_code == 500:
             raise ValueError('当前模型所有令牌暂无可用，请稍后重试')