Upload 2 files

app.py

@@ -14,12 +14,15 @@ from flask import Flask, request, Response, jsonify, stream_with_context, render
 from curl_cffi import requests as curl_requests
 from werkzeug.middleware.proxy_fix import ProxyFix
 import get_xid
+import threading
+
 
 current_dir = os.path.dirname(os.path.abspath(__file__))
 env_path = os.path.join(current_dir, '.env')
 
 load_dotenv(env_path)
 
+
 class Logger:
     def __init__(self, level="INFO", colorize=True, format=None):
         logger.remove()
@@ -85,8 +88,8 @@ class Logger:
         caller_info = self._get_caller_info()
         self.logger.bind(**caller_info).info(f"请求: {request.method} {request.path}", "Request")
 
-logger = Logger(level="INFO")
 
+logger = Logger(level="INFO")
 
 CONFIG = {
     "XIDS": None,
@@ -113,11 +116,11 @@ CONFIG = {
     },
     "ADMIN": {
         "MANAGER_SWITCH": os.getenv("MANAGER_SWITCH") or None,
-        "PASSWORD": os.getenv("ADMINPASSWORD") or None 
+        "PASSWORD": os.getenv("ADMINPASSWORD") or None
     },
     "SERVER": {
         "COOKIE": None,
-        "CF_CLEARANCE":os.getenv("CF_CLEARANCE") or None,
+        "CF_CLEARANCE": os.getenv("CF_CLEARANCE") or None,
         "PORT": int(os.getenv("PORT", 5200))
     },
     "RETRY": {
@@ -131,7 +134,6 @@ CONFIG = {
     "ISSHOW_SEARCH_RESULTS": os.getenv("ISSHOW_SEARCH_RESULTS", "true").lower() == "true"
 }
 
-
 DEFAULT_HEADERS = {
     'Accept': '*/*',
     'Accept-Language': 'zh-CN,zh;q=0.9',
@@ -152,6 +154,7 @@ DEFAULT_HEADERS = {
     'Baggage': 'sentry-public_key=b311e0f2690c81f25e2c4cf6d4f7ce1c'
 }
 
+
 class AuthTokenManager:
     def __init__(self):
         self.token_model_map = {}
@@ -187,7 +190,8 @@ class AuthTokenManager:
             if sso not in self.token_status_map:
                 self.token_status_map[sso] = {}
 
-            existing_token_entry = next((entry for entry in self.token_model_map[model] if entry["token"] == token), None)
+            existing_token_entry = next((entry for entry in self.token_model_map[model] if entry["token"] == token),
+                                        None)
 
             if not existing_token_entry:
                 self.token_model_map[model].append({
@@ -224,7 +228,8 @@ class AuthTokenManager:
         try:
             sso = token.split("sso=")[1].split(";")[0]
             for model in self.token_model_map:
-                self.token_model_map[model] = [entry for entry in self.token_model_map[model] if entry["token"] != token]
+                self.token_model_map[model] = [entry for entry in self.token_model_map[model] if
+                                               entry["token"] != token]
 
             if sso in self.token_status_map:
                 del self.token_status_map[sso]
@@ -234,39 +239,41 @@ class AuthTokenManager:
         except Exception as error:
             logger.error(f"令牌删除失败: {str(error)}")
             return False
+
     def reduce_token_request_count(self, model_id, count):
         try:
             normalized_model = self.normalize_model_name(model_id)
-            
+
             if normalized_model not in self.token_model_map:
                 logger.error(f"模型 {normalized_model} 不存在", "TokenManager")
                 return False
-                
+
             if not self.token_model_map[normalized_model]:
                 logger.error(f"模型 {normalized_model} 没有可用的token", "TokenManager")
                 return False
-                
+
             token_entry = self.token_model_map[normalized_model][0]
-            
+
             # 确保RequestCount不会小于0
             new_count = max(0, token_entry["RequestCount"] - count)
             reduction = token_entry["RequestCount"] - new_count
-            
+
             token_entry["RequestCount"] = new_count
-            
+
             # 更新token状态
             if token_entry["token"]:
                 sso = token_entry["token"].split("sso=")[1].split(";")[0]
                 if sso in self.token_status_map and normalized_model in self.token_status_map[sso]:
                     self.token_status_map[sso][normalized_model]["totalRequestCount"] = max(
-                        0, 
+                        0,
                         self.token_status_map[sso][normalized_model]["totalRequestCount"] - reduction
                     )
             return True
-            
+
         except Exception as error:
             logger.error(f"重置校对token请求次数时发生错误: {str(error)}", "TokenManager")
             return False
+
     def get_next_token_for_model(self, model_id, is_return=False):
         normalized_model = self.normalize_model_name(model_id)
 
@@ -289,7 +296,8 @@ class AuthTokenManager:
 
             if token_entry["RequestCount"] > self.model_config[normalized_model]["RequestFrequency"]:
                 self.remove_token_from_model(normalized_model, token_entry["token"])
-                next_token_entry = self.token_model_map[normalized_model][0] if self.token_model_map[normalized_model] else None
+                next_token_entry = self.token_model_map[normalized_model][0] if self.token_model_map[
+                    normalized_model] else None
                 return next_token_entry["token"] if next_token_entry else None
 
             sso = token_entry["token"].split("sso=")[1].split(";")[0]
@@ -428,6 +436,7 @@ class AuthTokenManager:
             for entry in model_tokens:
                 all_tokens.add(entry["token"])
         return list(all_tokens)
+
     def get_current_token(self, model_id):
         normalized_model = self.normalize_model_name(model_id)
 
@@ -440,6 +449,7 @@ class AuthTokenManager:
     def get_token_status_map(self):
         return self.token_status_map
 
+
 class Utils:
     @staticmethod
     def organize_search_results(search_results):
@@ -470,19 +480,20 @@ class Utils:
 
         if proxy:
             logger.info(f"使用代理: {proxy}", "Server")
-            
+
             if proxy.startswith("socks5://"):
                 proxy_options["proxy"] = proxy
-            
+
                 if '@' in proxy:
                     auth_part = proxy.split('@')[0].split('://')[1]
                     if ':' in auth_part:
                         username, password = auth_part.split(':')
                         proxy_options["proxy_auth"] = (username, password)
             else:
-                proxy_options["proxies"] = {"https": proxy, "http": proxy}     
+                proxy_options["proxies"] = {"https": proxy, "http": proxy}
         return proxy_options
 
+
 class GrokApiClient:
     def __init__(self, model_id):
         if model_id not in CONFIG["MODELS"]:
@@ -509,6 +520,7 @@ class GrokApiClient:
             "mimeType": mime_type,
             "fileName": file_name
         }
+
     def upload_base64_file(self, message, model):
         try:
             message_base64 = base64.b64encode(message.encode('utf-8')).decode('utf-8')
@@ -519,13 +531,13 @@ class GrokApiClient:
             }
 
             logger.info("发送文字文件请求", "Server")
-            cookie = f"{Utils.create_auth_headers(model, True)};{CONFIG['SERVER']['CF_CLEARANCE']}" 
+            cookie = f"{Utils.create_auth_headers(model, True)};{CONFIG['SERVER']['CF_CLEARANCE']}"
             proxy_options = Utils.get_proxy_options()
             response = curl_requests.post(
                 "https://grok.com/rest/app-chat/upload-file",
                 headers={
                     **DEFAULT_HEADERS,
-                    "Cookie":cookie
+                    "Cookie": cookie
                 },
                 json=upload_data,
                 impersonate="chrome133a",
@@ -544,6 +556,7 @@ class GrokApiClient:
         except Exception as error:
             logger.error(str(error), "Server")
             raise Exception(f"上传文件失败,状态码:{response.status_code}")
+
     def upload_base64_image(self, base64_data, url):
         try:
             if 'data:image' in base64_data:
@@ -571,7 +584,7 @@ class GrokApiClient:
                 url,
                 headers={
                     **DEFAULT_HEADERS,
-                    "Cookie":CONFIG["SERVER"]['COOKIE']
+                    "Cookie": CONFIG["SERVER"]['COOKIE']
                 },
                 json=upload_data,
                 impersonate="chrome133a",
@@ -593,8 +606,8 @@ class GrokApiClient:
 
     def prepare_chat_request(self, request):
         if ((request["model"] == 'grok-2-imageGen' or request["model"] == 'grok-3-imageGen') and
-            not CONFIG["API"]["PICGO_KEY"] and not CONFIG["API"]["TUMY_KEY"] and
-            request.get("stream", False)):
+                not CONFIG["API"]["PICGO_KEY"] and not CONFIG["API"]["TUMY_KEY"] and
+                request.get("stream", False)):
             raise ValueError("该模型流式输出需要配置PICGO或者TUMY图床密钥!")
 
         todo_messages = request["messages"]
@@ -626,7 +639,9 @@ class GrokApiClient:
                     if item["type"] == 'image_url':
                         text_content += ("[图片]" if not text_content else '\n[图片]')
                     elif item["type"] == 'text':
-                        text_content += (remove_think_tags(item["text"]) if not text_content else '\n' + remove_think_tags(item["text"]))
+                        text_content += (
+                            remove_think_tags(item["text"]) if not text_content else '\n' + remove_think_tags(
+                                item["text"]))
                 return text_content
             elif isinstance(content, dict) and content is not None:
                 if content["type"] == 'image_url':
@@ -657,7 +672,6 @@ class GrokApiClient:
                     if processed_image:
                         file_attachments.append(processed_image)
 
-
             text_content = process_content(current.get("content", ""))
             if is_last_message and convert_to_file:
                 last_message_content = f"{role.upper()}: {text_content or '[图片]'}\n"
@@ -673,7 +687,7 @@ class GrokApiClient:
             message_length += len(messages)
             if message_length >= 40000:
                 convert_to_file = True
-               
+
         if convert_to_file:
             file_id = self.upload_base64_file(messages, request["model"])
             if file_id:
@@ -713,6 +727,7 @@ class GrokApiClient:
             "isReasoning": request["model"] == 'grok-3-reasoning'
         }
 
+
 class MessageProcessor:
     @staticmethod
     def create_chat_response(message, model, is_stream=False):
@@ -748,6 +763,7 @@ class MessageProcessor:
             "usage": None
         }
 
+
 def process_model_response(response, model):
     result = {"token": None, "imageUrl": None}
 
@@ -774,7 +790,8 @@ def process_model_response(response, model):
         elif not response.get("messageStepId") and CONFIG["IS_THINKING"] and response.get("messageTag") == "final":
             result["token"] = "</think>" + response.get("token", "")
             CONFIG["IS_THINKING"] = False
-        elif (response.get("messageStepId") and CONFIG["IS_THINKING"] and response.get("messageTag") == "assistant") or response.get("messageTag") == "final":
+        elif (response.get("messageStepId") and CONFIG["IS_THINKING"] and response.get(
+                "messageTag") == "assistant") or response.get("messageTag") == "final":
             result["token"] = response.get("token")
     elif model == 'grok-3-reasoning':
         if response.get("isThinking") and not CONFIG["SHOW_THINKING"]:
@@ -791,6 +808,7 @@ def process_model_response(response, model):
 
     return result
 
+
 def handle_image_response(image_url):
     max_retries = 2
     retry_count = 0
@@ -803,7 +821,7 @@ def handle_image_response(image_url):
                 f"https://assets.grok.com/{image_url}",
                 headers={
                     **DEFAULT_HEADERS,
-                    "Cookie":CONFIG["SERVER"]['COOKIE']
+                    "Cookie": CONFIG["SERVER"]['COOKIE']
                 },
                 impersonate="chrome133a",
                 **proxy_options
@@ -879,6 +897,7 @@ def handle_image_response(image_url):
                 logger.error(str(error), "Server")
                 return "生图失败，请查看TUMY图床密钥是否设置正确"
 
+
 def handle_non_stream_response(response, model):
     try:
         logger.info("开始处理非流式响应", "Server")
@@ -925,6 +944,8 @@ def handle_non_stream_response(response, model):
     except Exception as error:
         logger.error(str(error), "Server")
         raise
+
+
 def handle_stream_response(response, model):
     def generate():
         logger.info("开始处理流式响应", "Server")
@@ -968,8 +989,19 @@ def handle_stream_response(response, model):
                 continue
 
         yield "data: [DONE]\n\n"
+
     return generate()
 
+def schedule_task():
+    while True:
+        # 创建并启动一个新线程来执行任务
+        xid = get_xid.main()
+        if xid is not None:
+            logger.info(f"成功获取XID: {xid}", "Server")
+            CONFIG["XIDS"] = xid
+
+        time.sleep(3 * 60 * 60)
+
 def initialization():
     sso_array = os.getenv("SSO", "").split(',')
     logger.info("开始加载令牌", "Server")
@@ -983,6 +1015,8 @@ def initialization():
     if CONFIG["API"]["PROXY"]:
         logger.info(f"代理已设置: {CONFIG['API']['PROXY']}", "Server")
 
+    thread = threading.Thread(target=schedule_task)
+    thread.start()
 
     logger.info("初始化完成", "Server")
 
@@ -992,6 +1026,7 @@ app.wsgi_app = ProxyFix(app.wsgi_app)
 app.secret_key = os.getenv('FLASK_SECRET_KEY') or secrets.token_hex(16)
 app.json.sort_keys = False
 
+
 @app.route('/manager/login', methods=['GET', 'POST'])
 def manager_login():
     if CONFIG["ADMIN"]["MANAGER_SWITCH"]:
@@ -1005,21 +1040,25 @@ def manager_login():
     else:
         return redirect('/')
 
+
 def check_auth():
     return session.get('is_logged_in', False)
 
+
 @app.route('/manager')
 def manager():
     if not check_auth():
         return redirect('/manager/login')
     return render_template('manager.html')
 
+
 @app.route('/manager/api/get')
 def get_manager_tokens():
     if not check_auth():
         return jsonify({"error": "Unauthorized"}), 401
     return jsonify(token_manager.get_token_status_map())
 
+
 @app.route('/manager/api/add', methods=['POST'])
 def add_manager_token():
     if not check_auth():
@@ -1033,6 +1072,7 @@ def add_manager_token():
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 
+
 @app.route('/manager/api/delete', methods=['POST'])
 def delete_manager_token():
     if not check_auth():
@@ -1045,8 +1085,9 @@ def delete_manager_token():
         return jsonify({"success": True})
     except Exception as e:
         return jsonify({"error": str(e)}), 500
-    
-@app.route('/manager/api/cf_clearance', methods=['POST'])   
+
+
+@app.route('/manager/api/cf_clearance', methods=['POST'])
 def setCf_Manager_clearance():
     if not check_auth():
         return jsonify({"error": "Unauthorized"}), 401
@@ -1069,6 +1110,7 @@ def get_tokens():
         return jsonify({"error": 'Unauthorized'}), 401
     return jsonify(token_manager.get_token_status_map())
 
+
 @app.route('/add/token', methods=['POST'])
 def add_token():
     auth_token = request.headers.get('Authorization', '').replace('Bearer ', '')
@@ -1084,7 +1126,8 @@ def add_token():
     except Exception as error:
         logger.error(str(error), "Server")
         return jsonify({"error": '添加sso令牌失败'}), 500
-    
+
+
 @app.route('/set/cf_clearance', methods=['POST'])
 def setCf_clearance():
     auth_token = request.headers.get('Authorization', '').replace('Bearer ', '')
@@ -1097,7 +1140,8 @@ def setCf_clearance():
     except Exception as error:
         logger.error(str(error), "Server")
         return jsonify({"error": '设置cf_clearance失败'}), 500
-    
+
+
 @app.route('/delete/token', methods=['POST'])
 def delete_token():
     auth_token = request.headers.get('Authorization', '').replace('Bearer ', '')
@@ -1114,6 +1158,7 @@ def delete_token():
         logger.error(str(error), "Server")
         return jsonify({"error": '删除sso令牌失败'}), 500
 
+
 @app.route('/v1/models', methods=['GET'])
 def get_models():
     return jsonify({
@@ -1129,6 +1174,7 @@ def get_models():
         ]
     })
 
+
 @app.route('/v1/chat/completions', methods=['POST'])
 def chat_completions():
     response_status_code = 500
@@ -1151,7 +1197,6 @@ def chat_completions():
         retry_count = 0
         grok_client = GrokApiClient(model)
         request_payload = grok_client.prepare_chat_request(data)
-        
 
         while retry_count < CONFIG["RETRY"]["MAX_ATTEMPTS"]:
             retry_count += 1
@@ -1161,38 +1206,45 @@ def chat_completions():
                 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']};{CONFIG['SERVER']['CF_CLEARANCE']}" 
+                CONFIG["SERVER"]['COOKIE'] = f"{CONFIG['API']['SIGNATURE_COOKIE']};{CONFIG['SERVER']['CF_CLEARANCE']}"
             else:
                 CONFIG["SERVER"]['COOKIE'] = CONFIG['API']['SIGNATURE_COOKIE']
-            logger.info("有请求到来","Server")
+            logger.info("有请求到来", "Server")
+            if CONFIG['XIDS'] is not None:
+                DEFAULT_HEADERS['x-xai-request-id'] = CONFIG['XIDS']['x-xai-request-id']
+                DEFAULT_HEADERS['x-statsig-id'] = CONFIG['XIDS']['x-statsig-id']
+                DEFAULT_HEADERS['User-Agent'] = CONFIG['XIDS']['user-agent']
+
             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":CONFIG["SERVER"]['COOKIE']
+                        **DEFAULT_HEADERS,
+                        "Cookie": CONFIG["SERVER"]['COOKIE']
                     },
                     json=request_payload,
                     impersonate="chrome133a",
                     verify=False,
                     stream=True,
                     **proxy_options)
-                logger.info(CONFIG["SERVER"]['COOKIE'],"Server")
+                logger.info(CONFIG["SERVER"]['COOKIE'], "Server")
                 if response.status_code == 200:
                     response_status_code = 200
                     logger.info("请求成功", "Server")
-                    logger.info(f"当前{model}剩余可用令牌数: {token_manager.get_token_count_for_model(model)}","Server")
+                    logger.info(f"当前{model}剩余可用令牌数: {token_manager.get_token_count_for_model(model)}",
+                                "Server")
 
                     try:
                         if stream:
                             return Response(stream_with_context(
-                                handle_stream_response(response, model)),content_type='text/event-stream')
+                                handle_stream_response(response, model)), content_type='text/event-stream')
                         else:
                             content = handle_non_stream_response(response, model)
                             return jsonify(
@@ -1207,13 +1259,13 @@ def chat_completions():
                             raise ValueError(f"{model} 次数已达上限，请切换其他模型或者重新对话")
                 elif response.status_code == 403:
                     response_status_code = 403
-                    token_manager.reduce_token_request_count(model,1)#重置去除当前因为错误未成功请��的次数，确保不会因为错误未成功请求的次数导致次数上限
+                    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")
                 elif response.status_code == 429:
                     response_status_code = 429
-                    token_manager.reduce_token_request_count(model,1)
+                    token_manager.reduce_token_request_count(model, 1)
                     if CONFIG["API"]["IS_CUSTOM_SSO"]:
                         raise ValueError(f"自定义SSO令牌当前模型{model}的请求次数已失效")
 
@@ -1226,7 +1278,7 @@ def chat_completions():
                     if CONFIG["API"]["IS_CUSTOM_SSO"]:
                         raise ValueError(f"自定义SSO令牌当前模型{model}的请求次数已失效")
 
-                    logger.error(f"令牌异常错误状态!status: {response.status_code}","Server")
+                    logger.error(f"令牌异常错误状态!status: {response.status_code}", "Server")
                     token_manager.remove_token_from_model(model, CONFIG["API"]["SIGNATURE_COOKIE"])
                     logger.info(
                         f"当前{model}剩余可用令牌数: {token_manager.get_token_count_for_model(model)}",
@@ -1240,7 +1292,7 @@ def chat_completions():
         if response_status_code == 403:
             raise ValueError('IP暂时被封无法破盾，请稍后重试或者更换ip')
         elif response_status_code == 500:
-            raise ValueError('当前模型所有令牌暂无可用，请稍后重试')    
+            raise ValueError('当前模型所有令牌暂无可用，请稍后重试')
 
     except Exception as error:
         logger.error(str(error), "ChatAPI")
@@ -1250,11 +1302,13 @@ def chat_completions():
                 "type": "server_error"
             }}), response_status_code
 
+
 @app.route('/', defaults={'path': ''})
 @app.route('/<path:path>')
 def catch_all(path):
     return 'api运行正常', 200
 
+
 if __name__ == '__main__':
     token_manager = AuthTokenManager()
     initialization()

get_xid.py

@@ -1,132 +1,143 @@
-from playwright.sync_api import sync_playwright
-
-# 用于存储捕获到的请求头信息
-all_request_headers_info = []
-
-
-def handle_request(request):
-    """
-    捕获每个请求的URL, 方法和头部信息
-    """
-    # print(f"Intercepted request to: {request.url}") # 调试时可以取消注释
-    all_request_headers_info.append({
-        "url": request.url,
-        "method": request.method,
-        "headers": request.headers  # request.headers 是一个字典
-    })
-
-
-def main():
-    with sync_playwright() as p:
-        # 启动浏览器，可以是 chromium, firefox, or webkit
-        # headless=False 可以看到浏览器操作，True则为无头模式
-        browser = p.chromium.launch(headless=True,
-                                    args=[
-                '--no-sandbox',
-                '--disable-setuid-sandbox',
-                '--disable-dev-shm-usage' # 有时也需要这个，但 --shm-size 更好
-            ])
-
-        # 创建一个新的浏览器上下文
-        # 可以在这里设置 user_agent, viewport, etc.
-        context = browser.new_context(
-            user_agent="Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:138.0) Gecko/20100101 Firefox/138.0",
-        )
-
-        # 在上下文中创建一个新页面
-        page = context.new_page()
-
-        # 注册请求拦截器，这必须在导航之前完成
-        # 'request' 事件会在每个HTTP请求发起时触发
-        page.on("request", handle_request)
-
-        print(f"Navigating to https://grok.com/ ...")
-        try:
-            # 访问目标网站，设置一个合理的超时时间（例如60秒）
-            page.goto("https://grok.com/", timeout=60000)
-            print("Page loaded. Waiting for 10 seconds for dynamic content or further requests...")
-
-            # 检查是否仍然被 Cloudflare 阻止 (例如，查找特定的标题或元素)
-            title = page.title()
-            print(f"Page title: {title}")
-
-            if "请稍候…" in page.content() or "Just a moment..." in page.content() or "Cloudflare" in title or "Checking your browser" in title:
-                print("Still on a Cloudflare challenge page. Waiting longer or trying interaction...")
-                # 你可能需要在这里添加更长的等待或模拟用户交互
-                # 例如，等待特定的元素出现，表明挑战已通过
-                try:
-                    page.wait_for_selector("body:not(:has-text('请稍候…'))", timeout=60000)
-                    print("Cloudflare challenge likely passed.")
-                    title = page.title()
-                    print(f"New page title: {title}")
-                except Exception as e:
-                    print(f"Failed to pass Cloudflare challenge after extended wait: {e}")
-            else:
-                print("Successfully navigated to the page.")
-                
-            
-            page.wait_for_timeout(10000)
-
-
-            try:
-                textarea_locator = page.get_by_label("向Grok提任何问题")
-                textarea_locator.fill("你好")
-                print("Successfully entered '你好' into the textarea.")
-            except Exception as e:
-                print(f"Could not find or fill the textarea with aria-label '向Grok提任何问题'. Error: {e}")
-                browser.close()
-                return
-
-                # 2. 查找 aria-label 为“提交”的 button 并点击
-                # 使用 get_by_role('button', name='...') 是 Playwright 推荐的方式来查找具有特定可访问名称的按钮
-            try:
-                submit_button_locator = page.get_by_role("button", name="提交")
-                submit_button_locator.click()
-                print("Successfully clicked the '提交' button.")
-            except Exception as e:
-                print(f"Could not find or click the button with aria-label '提交'. Error: {e}")
-                browser.close()
-                return
-
-            # 等待10秒
-            # Playwright 的 page.wait_for_timeout() 是首选，因为它与Playwright的事件循环集成
-            # page.wait_for_timeout(10000)
-            # 或者使用 time.sleep(10) 也可以，但在Playwright脚本中前者更佳
-
-            print("\n--- Cookies ---")
-            # 获取当前上下文中的所有cookies
-            cookies = context.cookies()
-            if cookies:
-                for cookie in cookies:
-                    print(
-                        f"Name: {cookie['name']}, Value: {cookie['value']}, Domain: {cookie['domain']}, Path: {cookie['path']}")
-            else:
-                print("No cookies found.")
-
-            print("\n--- Request Headers (collected during the session) ---")
-            if all_request_headers_info:
-                # 打印捕获到的每个请求的头部信息
-                # 注意：这里会包含所有资源的请求（HTML, CSS, JS, XHR, 图片等）
-                for i, req_info in enumerate(all_request_headers_info):
-                    if req_info['url'] == 'https://grok.com/rest/app-chat/conversations/new':
-                        datas = {
-                            'x-xai-request-id': req_info['headers']['x-xai-request-id'],
-                            'x-statsig-id':req_info['headers']['x-statsig-id'],
-                            'user-agent': req_info['headers']['user-agent'],
-                        }
-                        print(datas)
-                        return datas
-            else:
-                print("No requests were intercepted (this is unlikely if the page loaded).")
-
-        except Exception as e:
-            print(f"An error occurred: {e}")
-        finally:
-            # 确保浏览器关闭
-            print("\nClosing browser...")
-            browser.close()
-        return None
-
-
-if __name__ == "__main__":
+from playwright.sync_api import sync_playwright
+
+# 用于存储捕获到的请求头信息
+all_request_headers_info = []
+
+
+def handle_request(request):
+    """
+    捕获每个请求的URL, 方法和头部信息
+    """
+    # print(f"Intercepted request to: {request.url}") # 调试时可以取消注释
+    all_request_headers_info.append({
+        "url": request.url,
+        "method": request.method,
+        "headers": request.headers  # request.headers 是一个字典
+    })
+
+
+def main():
+    with sync_playwright() as p:
+        # 启动浏览器，可以是 chromium, firefox, or webkit
+        # headless=False 可以看到浏览器操作，True则为无头模式
+        browser = p.chromium.launch(headless=True,
+                                    args=[
+                                        '--no-sandbox',
+                                        '--disable-setuid-sandbox',
+                                        '--disable-dev-shm-usage'  # 有时也需要这个，但 --shm-size 更好
+                                    ])
+
+        # 创建一个新的浏览器上下文
+        # 可以在这里设置 user_agent, viewport, etc.
+        context = browser.new_context(
+            user_agent="Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:138.0) Gecko/20100101 Firefox/138.0",
+        )
+
+        # 在上下文中创建一个新页面
+        page = context.new_page()
+
+        # 注册请求拦截器，这必须在导航之前完成
+        # 'request' 事件会在每个HTTP请求发起时触发
+        page.on("request", handle_request)
+
+        print(f"Navigating to https://grok.com/ ...")
+        try:
+            # 访问目标网站，设置一个合理的超时时间（例如60秒）
+            page.goto("https://grok.com/", timeout=60000)
+            page.wait_for_timeout(5000)
+            print("Page loaded. Waiting for 10 seconds for dynamic content or further requests...")
+
+            # 检查是否仍然被 Cloudflare 阻止 (例如，查找特定的标题或元素)
+            title = page.title()
+            print(f"Page title: {title}")
+
+            if "请稍候…" in page.content() or "Just a moment..." in page.content() or "Cloudflare" in title or "Checking your browser" in title:
+                print("Still on a Cloudflare challenge page. Waiting longer or trying interaction...")
+                # 你可能需要在这里添加更长的等待或模拟用户交互
+                # 例如，等待特定的元素出现，表明挑战已通过
+                try:
+                    page.wait_for_selector("body:not(:has-text('请稍候…'))", timeout=60000)
+                    print("Cloudflare challenge likely passed.")
+                    title = page.title()
+                    print(f"New page title: {title}")
+                    page.screenshot(path="cf_passed.png")
+                except Exception as e:
+                    print(f"Failed to pass Cloudflare challenge after extended wait: {e}")
+                    page.screenshot(path="cf_failed.png")
+            else:
+                print("Successfully navigated to the page.")
+                page.screenshot(path="cf_success.png")
+
+            page.evaluate("""
+                function(){
+                    const element = document.getElementById('turnstile-widget');
+                    if (element) {
+                      element.style.display = 'none';
+                    }
+                }
+            """)
+            page.wait_for_timeout(10000)
+
+            try:
+                textarea_locator = page.get_by_label("Ask Grok anything")
+                textarea_locator.fill("你好")
+                print("Successfully entered '你好' into the textarea.")
+            except Exception as e:
+                print(f"Could not find or fill the textarea with aria-label 'Ask Grok anything'. Error: {e}")
+                browser.close()
+                return
+
+                # 2. 查找 aria-label 为“提交”的 button 并点击
+                # 使用 get_by_role('button', name='...') 是 Playwright 推荐的方式��查找具有特定可访问名称的按钮
+            try:
+                submit_button_locator = page.get_by_role("button", name="Submit")
+                submit_button_locator.click()
+                print("Successfully clicked the 'Submit' button.")
+            except Exception as e:
+                print(f"Could not find or click the button with aria-label 'Submit'. Error: {e}")
+                browser.close()
+                return
+
+            # 等待10秒
+            # Playwright 的 page.wait_for_timeout() 是首选，因为它与Playwright的事件循环集成
+            # page.wait_for_timeout(10000)
+            # 或者使用 time.sleep(10) 也可以，但在Playwright脚本中前者更佳
+
+            print("\n--- Cookies ---")
+            # 获取当前上下文中的所有cookies
+            cookies = context.cookies()
+            if cookies:
+                for cookie in cookies:
+                    print(
+                        f"Name: {cookie['name']}, Value: {cookie['value']}, Domain: {cookie['domain']}, Path: {cookie['path']}")
+            else:
+                print("No cookies found.")
+
+            print("\n--- Request Headers (collected during the session) ---")
+            if all_request_headers_info:
+                # 打印捕获到的每个请求的头部信息
+                # 注意：这里会包含所有资源的请求（HTML, CSS, JS, XHR, 图片等）
+                for i, req_info in enumerate(all_request_headers_info):
+                    if req_info['url'] == 'https://grok.com/rest/app-chat/conversations/new':
+                        datas = {
+                            'x-xai-request-id': req_info['headers']['x-xai-request-id'],
+                            'x-statsig-id': req_info['headers']['x-statsig-id'],
+                            'user-agent': req_info['headers']['user-agent'],
+                        }
+                        print(datas)
+                        return datas
+            else:
+                print("No requests were intercepted (this is unlikely if the page loaded).")
+
+        except Exception as e:
+            print(f"An error occurred: {e}")
+        finally:
+            # 确保浏览器关闭
+            print("\nClosing browser...")
+            page.close()
+            browser.close()
+        return None
+
+
+if __name__ == "__main__":
     main()