Update main.py

main.py

@@ -57,10 +57,17 @@ logger.info(f"OPENAI_API_KEY value: {OPENAI_API_KEY}")
 
 def get_cookie():
     try:
+        # 创建配置对象
         options = ChromiumOptions()
-        # 设置无头模式
+        
+        # 设置浏览器路径（如果需要）
+        # options.set_browser_path('/usr/bin/google-chrome')
+        
+        # 设置用户数据目录
+        options.set_argument('--user-data-dir=/tmp/chrome-data')
+        
+        # 设置无头模式和其他参数
         options.set_argument('--headless=new')
-        # 添加容器环境所需的启动参数
         options.set_argument('--no-sandbox')
         options.set_argument('--disable-dev-shm-usage')
         options.set_argument('--disable-gpu')
@@ -70,16 +77,22 @@ def get_cookie():
         options.set_argument('--no-first-run')
         options.set_argument('--no-zygote')
         options.set_argument('--single-process')
-        options.set_argument('--disable-dev-shm-usage')
         options.set_argument('--remote-debugging-port=9222')
         
+        # 设置窗口大小
+        options.set_argument('--window-size=1920,1080')
+        
+        # 设置用户代理
+        options.set_argument('--user-agent=Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36')
+        
+        # 创建页面对象
         page = ChromiumPage(addr_or_opts=options)
-        page.set.window.size(1920, 1080)
-        page.set.user_agent = "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36"
         
+        # 访问目标网站
         page.get("https://chat.akash.network/")
         time.sleep(10)
         
+        # 获取 cookies
         cookies = page.cookies()
         if not cookies:
             page.quit()