Update main.py

main.py

@@ -1,3 +1,4 @@
+import os
 import cloudscraper
 from fastapi import FastAPI, HTTPException, Request, Response
 from fastapi.middleware.cors import CORSMiddleware
@@ -48,37 +49,25 @@ async def proxy(request: Request, path: str, target_url: Optional[str] = None):
         # 移除可能导致问题的头
         headers.pop("host", None)
         headers.pop("content-length", None)
-        headers.update({
-            "sec-fetch-dest": "document",
-        })
+        
         # 检查是否请求流式响应
         stream_request = "stream" in request.query_params and request.query_params["stream"].lower() in ["true", "1", "yes"]
         
+        # 创建带有代理的 scraper
         # 创建cloudscraper实例
-        scraper = cloudscraper.create_scraper(   
-            browser={
-                'browser': 'chrome',
-                'platform': 'windows',
-                'desktop': True
-            },
-            delay=10
-        )
-        grok_response = scraper.get("https://grok.com/",headers={
-            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/96.0.4664.110 Safari/537.36',
-            'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,*/*;q=0.8',
-            'Accept-Language': 'en-US,en;q=0.5',
-            'Accept-Encoding': 'gzip, deflate, br',
-            'Referer': 'https://grok.com/',  # 添加一个合理的来源
-            'DNT': '1',
-            'Connection': 'keep-alive',
-            'Upgrade-Insecure-Requests': '1',
-            'Sec-Fetch-Dest': 'document',
-            'Sec-Fetch-Mode': 'navigate',
-            'Sec-Fetch-Site': 'cross-site',
-            'Sec-Fetch-User': '?1',
-            'Cache-Control': 'max-age=0',
-        })
-        print(f"grok主页状态码: {grok_response.status_code}")
+        scraper = cloudscraper.create_scraper()
+        # 检查环境变量PROXY是否存在
+        proxy = os.environ.get('PROXY')
+        if proxy:
+            # 如果环境变量存在，则设置代理
+            scraper.proxies = {
+                'http': proxy,
+                'https': proxy
+            }
+        # 测试代理是否生效
+        response = scraper.get('https://httpbin.org/ip')
+        print(response.text)
+        
         # 获取请求体
         body = await request.body()