Upload screenshotService.js

backend/src/services/screenshotService.js

@@ -25,8 +25,8 @@ class ScreenshotService {
         
         const launchOptions = {
           headless: 'new',
-          timeout: 30000, // 减少超时时间
-          protocolTimeout: 30000,
+          timeout: 60000, // 增加超时时间
+          protocolTimeout: 60000,
           args: [
             '--no-sandbox',
             '--disable-setuid-sandbox',
@@ -55,22 +55,53 @@ class ScreenshotService {
             '--disable-hang-monitor',
             '--disable-prompt-on-repost',
             '--memory-pressure-off',
-            '--max_old_space_size=512' // 减少内存使用
+            '--max_old_space_size=1024', // 增加内存限制
+            // 添加更多Hugging Face Space兼容性选项
+            '--disable-background-media-suspend',
+            '--disable-backgrounding-occluded-windows',
+            '--disable-renderer-backgrounding',
+            '--disable-field-trial-config',
+            '--disable-component-extensions-with-background-pages',
+            '--disable-permissions-api',
+            '--disable-client-side-phishing-detection',
+            '--no-zygote',
+            '--disable-web-security',
+            '--allow-running-insecure-content',
+            '--disable-features=VizDisplayCompositor,AudioServiceOutOfProcess',
+            '--disable-software-rasterizer',
+            '--disable-canvas-aa',
+            '--disable-2d-canvas-clip-aa',
+            '--disable-gl-drawing-for-tests',
+            '--use-gl=swiftshader'
           ]
         };
 
-        // 如果是Hugging Face Space环境，使用更保守的配置
+        // 如果是Hugging Face Space环境，使用单进程模式
         if (this.isHuggingFaceSpace) {
-          console.log('检测到Hugging Face Space环境，使用保守配置');
+          console.log('检测到Hugging Face Space环境，使用单进程模式');
           launchOptions.args.push(
             '--single-process',
-            '--no-zygote',
-            '--disable-web-security',
             '--disable-features=site-per-process',
-            '--disable-site-isolation-trials'
+            '--disable-site-isolation-trials',
+            '--disable-features=BlockInsecurePrivateNetworkRequests'
           );
         }
 
+        // 尝试找到Chrome可执行文件路径
+        try {
+          const chromePath = process.env.CHROME_BIN || 
+                           process.env.GOOGLE_CHROME_BIN || 
+                           '/usr/bin/google-chrome-stable' ||
+                           '/usr/bin/chromium-browser';
+          
+          if (chromePath && require('fs').existsSync(chromePath)) {
+            launchOptions.executablePath = chromePath;
+            console.log(`使用Chrome路径: ${chromePath}`);
+          }
+        } catch (pathError) {
+          console.log('未找到自定义Chrome路径，使用默认配置');
+        }
+
         this.browser = await puppeteer.launch(launchOptions);
         console.log('✅ Puppeteer浏览器初始化成功');
 
@@ -88,7 +119,7 @@ class ScreenshotService {
         
         if (this.browserLaunchRetries <= this.maxBrowserLaunchRetries) {
           console.log(`尝试重新初始化浏览器 (${this.browserLaunchRetries}/${this.maxBrowserLaunchRetries})`);
-          await this.delay(2000); // 等待2秒后重试
+          await this.delay(3000); // 增加等待时间
           return this.initBrowser();
         } else {
           // 如果Puppeteer完全失败，返回null，使用fallback方法