Update main.py

main.py

@@ -57,14 +57,20 @@ logger.info(f"OPENAI_API_KEY value: {OPENAI_API_KEY}")
 
 def get_cookie():
     try:
+        logger.info("Starting cookie retrieval process...")
+        
         # 创建配置对象
         options = ChromiumOptions()
         
-        # 设置浏览器路径（如果需要）
-        # options.set_browser_path('/usr/bin/google-chrome')
+        # 设置浏览器路径
+        chrome_path = os.getenv('CHROME_PATH', '/usr/bin/google-chrome-stable')
+        logger.info(f"Using Chrome path: {chrome_path}")
+        options.set_browser_path(chrome_path)
         
         # 设置用户数据目录
-        options.set_argument('--user-data-dir=/tmp/chrome-data')
+        user_data_dir = '/tmp/chrome-data'
+        logger.info(f"Using user data directory: {user_data_dir}")
+        options.set_argument(f'--user-data-dir={user_data_dir}')
         
         # 设置无头模式和其他参数
         options.set_argument('--headless=new')
@@ -78,28 +84,28 @@ def get_cookie():
         options.set_argument('--no-zygote')
         options.set_argument('--single-process')
         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)
+        logger.info("Creating ChromiumPage instance...")
+        page = ChromiumPage(addr_or_opts=options, show_log=True)
         
-        # 访问目标网站
+        logger.info("Navigating to target website...")
         page.get("https://chat.akash.network/")
+        
+        logger.info("Waiting for page load...")
         time.sleep(10)
         
-        # 获取 cookies
+        logger.info("Getting cookies...")
         cookies = page.cookies()
         if not cookies:
+            logger.error("No cookies found")
             page.quit()
             return None
             
         cookie_dict = {cookie['name']: cookie['value'] for cookie in cookies}
         if 'cf_clearance' not in cookie_dict:
+            logger.error("cf_clearance cookie not found")
             page.quit()
             return None
             
@@ -113,11 +119,15 @@ def get_cookie():
         else:
             global_data["cookie_expires"] = time.time() + 3600
         
+        logger.info("Successfully retrieved cookies")
         page.quit()
         return cookie_str
                 
     except Exception as e:
-        logger.error(f"Error fetching cookie: {e}")
+        logger.error(f"Error fetching cookie: {str(e)}")
+        logger.error(f"Error type: {type(e)}")
+        import traceback
+        logger.error(f"Traceback: {traceback.format_exc()}")
         return None
 
 # 添加刷新 cookie 的函数