Upload 2 files

Dockerfile

@@ -1,27 +1,67 @@
-# 最简 HF Spaces Dockerfile - 修复版
-FROM ghcr.io/puppeteer/puppeteer:21.5.2
+# HF Spaces Dockerfile - 修复 Chrome 路径问题
+FROM node:18-slim
 
-# 直接设置工作目录
-WORKDIR /usr/src/app
+# 安装必要的系统依赖
+RUN apt-get update && apt-get install -y \
+    wget \
+    gnupg \
+    ca-certificates \
+    apt-transport-https \
+    --no-install-recommends
+
+# 添加 Google Chrome 官方仓库
+RUN wget -q -O - https://dl.google.com/linux/linux_signing_key.pub | apt-key add - \
+    && echo "deb [arch=amd64] http://dl.google.com/linux/chrome/deb/ stable main" > /etc/apt/sources.list.d/google-chrome.list
+
+# 安装 Chrome 和相关依赖
+RUN apt-get update && apt-get install -y \
+    google-chrome-stable \
+    fonts-liberation \
+    libappindicator3-1 \
+    libasound2 \
+    libatk-bridge2.0-0 \
+    libdrm2 \
+    libgtk-3-0 \
+    libnspr4 \
+    libnss3 \
+    libx11-xcb1 \
+    libxcomposite1 \
+    libxdamage1 \
+    libxrandr2 \
+    libgbm1 \
+    libxss1 \
+    libgconf-2-4 \
+    --no-install-recommends \
+    && rm -rf /var/lib/apt/lists/*
+
+# 创建非 root 用户
+RUN groupadd -r pptruser && useradd -r -g pptruser -G audio,video pptruser \
+    && mkdir -p /home/pptruser/Downloads \
+    && chown -R pptruser:pptruser /home/pptruser
+
+# 设置工作目录
+WORKDIR /app
 
 # 复制 package 文件
 COPY package*.json ./
 
-# 切换到 root 用户安装依赖
-USER root
-
-# 现在有了 package-lock.json，可以使用 npm ci
+# 安装 Node.js 依赖
 RUN npm ci --only=production && npm cache clean --force
 
 # 复制应用代码
 COPY . .
 
-# 切换回非 root 用户
+# 更改所有权
+RUN chown -R pptruser:pptruser /app
+
+# 切换到非 root 用户
 USER pptruser
 
 # 设置环境变量
 ENV NODE_ENV=production
 ENV PORT=7860
+ENV PUPPETEER_SKIP_CHROMIUM_DOWNLOAD=true
+ENV PUPPETEER_EXECUTABLE_PATH=/usr/bin/google-chrome-stable
 
 # 暴露端口
 EXPOSE 7860

server.js

@@ -73,9 +73,10 @@ app.post('/screenshot', async (req, res) => {
   try {
     console.log('Launching browser...');
     
-    // HF Spaces 优化配置 - 使用 Puppeteer 镜像的默认 Chrome
-    browser = await puppeteer.launch({
+    // HF Spaces 优化配置 - 明确指定 Chrome 路径
+    const browserOptions = {
       headless: 'new',
+      executablePath: '/usr/bin/google-chrome-stable',
       args: [
         '--no-sandbox',
         '--disable-setuid-sandbox',
@@ -89,7 +90,9 @@ app.post('/screenshot', async (req, res) => {
         '--disable-backgrounding-occluded-windows',
         '--disable-renderer-backgrounding'
       ]
-    });
+    };
+
+    browser = await puppeteer.launch(browserOptions);
 
     const page = await browser.newPage();
     
@@ -151,6 +154,8 @@ app.post('/screenshot', async (req, res) => {
       errorResponse.suggestion = 'Try a simpler webpage or reduce timeout';
     } else if (error.message.includes('net::')) {
       errorResponse.suggestion = 'Check if the URL is accessible';
+    } else if (error.message.includes('Chrome')) {
+      errorResponse.suggestion = 'Chrome installation issue, contact admin';
     }
     
     res.status(500).json(errorResponse);