Upload 2 files

Dockerfile

@@ -1,22 +1,18 @@
-# HF Spaces Dockerfile - 修复 Chrome 路径问题
+# HF Spaces Dockerfile - 精简全球字体支持
 FROM node:18-slim
 
-# 安装必要的系统依赖
+# 安装必要的系统依赖和 Chrome
 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 \
+    && 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 \
+    && apt-get update && apt-get install -y \
     google-chrome-stable \
-    fonts-liberation \
+    fonts-noto-core \
+    fonts-noto-cjk \
+    fonts-noto-color-emoji \
     libappindicator3-1 \
     libasound2 \
     libatk-bridge2.0-0 \
@@ -32,7 +28,8 @@ RUN apt-get update && apt-get install -y \
     libxss1 \
     libgconf-2-4 \
     --no-install-recommends \
-    && rm -rf /var/lib/apt/lists/*
+    && rm -rf /var/lib/apt/lists/* \
+    && fc-cache -fv
 
 # 创建非 root 用户
 RUN groupadd -r pptruser && useradd -r -g pptruser -G audio,video pptruser \

server.js

@@ -3,142 +3,39 @@ const puppeteer = require('puppeteer');
 const cors = require('cors');
 const helmet = require('helmet');
 const rateLimit = require('express-rate-limit');
-const os = require('os');
 
 const app = express();
 const PORT = process.env.PORT || 7860;
 
-// 请求队列管理
-const requestQueue = [];
-let activeRequests = 0;
-const MAX_CONCURRENT_REQUESTS = 3;
-
-// CPU 监控
-let lastCpuUsage = 0;
-let cpuMonitorInterval;
-
-// 中间件配置
-app.use(helmet({
-  contentSecurityPolicy: false
-}));
+// 安全中间件
+app.use(helmet());
 app.use(cors());
-app.use(express.json({ limit: '10mb' }));
+app.use(express.json());
 
-// 简化的速率限制 - 所有用户统一限制
+// 速率限制
 const limiter = rateLimit({
-  windowMs: 15 * 60 * 1000,
-  max: 100, // 统一的较高限制
+  windowMs: 15 * 60 * 1000, // 15分钟
+  max: 100, // 限制每个IP 100次请求
   message: {
-    error: 'Too many requests, please try again later.'
-  },
-  keyGenerator: (req) => {
-    return req.ip;
+    error: 'Too many requests',
+    message: 'Rate limit exceeded. Please try again later.'
   }
 });
-app.use('/screenshot', limiter);
+app.use(limiter);
 
-// CPU 监控功能
-function startCpuMonitoring() {
-  const cpus = os.cpus();
-  
-  cpuMonitorInterval = setInterval(() => {
-    const cpus = os.cpus();
-    let totalIdle = 0;
-    let totalTick = 0;
-    
-    cpus.forEach(cpu => {
-      for (const type in cpu.times) {
-        totalTick += cpu.times[type];
-      }
-      totalIdle += cpu.times.idle;
-    });
-    
-    const idle = totalIdle / cpus.length;
-    const total = totalTick / cpus.length;
-    const usage = 100 - ~~(100 * idle / total);
-    
-    lastCpuUsage = usage;
-  }, 2000);
-}
-
-// 请求队列处理
-function processQueue() {
-  if (requestQueue.length === 0 || activeRequests >= MAX_CONCURRENT_REQUESTS) {
-    return;
-  }
-
-  if (lastCpuUsage >= 95) {
-    return;
-  }
-
-  const { req, res, next } = requestQueue.shift();
-  activeRequests++;
-  
-  const originalSend = res.send;
-  res.send = function(...args) {
-    activeRequests--;
-    setImmediate(processQueue);
-    return originalSend.apply(this, args);
-  };
+// 队列管理
+let activeRequests = 0;
+const MAX_CONCURRENT = 3;
+const requestQueue = [];
 
-  next();
+// CPU 监控
+function getCpuUsage() {
+  const usage = process.cpuUsage();
+  const total = usage.user + usage.system;
+  return Math.round((total / 1000000) % 100); // 简化的CPU使用率
 }
 
-// 队列中间件
-const queueMiddleware = (req, res, next) => {
-  if (lastCpuUsage >= 95) {
-    return res.status(503).json({
-      status: 'busy',
-      error: 'Server is currently overloaded',
-      message: 'CPU usage is too high. Please try again later.',
-      cpuUsage: `${lastCpuUsage}%`,
-      queueLength: requestQueue.length
-    });
-  }
-
-  if (activeRequests >= MAX_CONCURRENT_REQUESTS) {
-    requestQueue.push({ req, res, next });
-    
-    const timeout = setTimeout(() => {
-      const index = requestQueue.findIndex(item => item.res === res);
-      if (index !== -1) {
-        requestQueue.splice(index, 1);
-        res.status(503).json({
-          status: 'busy',
-          error: 'Request queue timeout',
-          message: 'Request was queued too long and timed out.',
-          queueLength: requestQueue.length
-        });
-      }
-    }, 30000);
-
-    req.on('close', () => {
-      clearTimeout(timeout);
-      const index = requestQueue.findIndex(item => item.res === res);
-      if (index !== -1) {
-        requestQueue.splice(index, 1);
-      }
-    });
-
-    return;
-  }
-
-  activeRequests++;
-  
-  const originalSend = res.send;
-  res.send = function(...args) {
-    activeRequests--;
-    setImmediate(processQueue);
-    return originalSend.apply(this, args);
-  };
-
-  next();
-};
-
-// 启动 CPU 监控
-startCpuMonitoring();
-
-// 健康检查端点
+// 健康检查
 app.get('/', (req, res) => {
   res.json({
     message: 'Page Screenshot API - Hugging Face Spaces',
@@ -159,26 +56,22 @@ app.get('/', (req, res) => {
   });
 });
 
-// 服务器状态端点
+// 状态检查
 app.get('/status', (req, res) => {
-  const memUsage = process.memoryUsage();
+  const cpuUsage = getCpuUsage();
   
   res.json({
     status: 'running',
     timestamp: new Date().toISOString(),
     system: {
-      cpuUsage: `${lastCpuUsage}%`,
-      memoryUsage: {
-        rss: `${Math.round(memUsage.rss / 1024 / 1024)}MB`,
-        heapUsed: `${Math.round(memUsage.heapUsed / 1024 / 1024)}MB`,
-        heapTotal: `${Math.round(memUsage.heapTotal / 1024 / 1024)}MB`
-      },
+      cpuUsage: `${cpuUsage}%`,
+      memoryUsage: '', // HF Spaces 不提供详细内存信息
       uptime: `${Math.floor(process.uptime())}s`
     },
     queue: {
       activeRequests,
       queuedRequests: requestQueue.length,
-      maxConcurrent: MAX_CONCURRENT_REQUESTS
+      maxConcurrent: MAX_CONCURRENT
     },
     authentication: {
       type: 'HuggingFace System Level',
@@ -187,40 +80,146 @@ app.get('/status', (req, res) => {
   });
 });
 
-// 截图API端点 - 移除所有认证检查
-app.post('/screenshot', queueMiddleware, async (req, res) => {
+// 演示页面
+app.get('/demo', (req, res) => {
+  res.send(`
+<!DOCTYPE html>
+<html>
+<head>
+    <title>Page Screenshot API Demo</title>
+    <style>
+        body { font-family: -apple-system, BlinkMacSystemFont, sans-serif; max-width: 800px; margin: 0 auto; padding: 20px; }
+        .form-group { margin: 15px 0; }
+        input, button { padding: 12px; margin: 5px; border: 1px solid #ddd; border-radius: 6px; }
+        input[type="text"] { width: 400px; }
+        button { background: #007bff; color: white; border: none; cursor: pointer; }
+        button:hover { background: #0056b3; }
+        #result { margin-top: 20px; }
+        img { max-width: 100%; border: 1px solid #ccc; border-radius: 6px; }
+        .loading { color: #666; font-style: italic; }
+        .error { color: #dc3545; }
+        .success { color: #28a745; }
+    </style>
+</head>
+<body>
+    <h1>📸 Page Screenshot API Demo</h1>
+    <p>Test the screenshot API with global font support</p>
+    
+    <div class="form-group">
+        <label>URL to screenshot:</label><br>
+        <input type="text" id="url" value="https://www.baidu.com" placeholder="Enter website URL">
+    </div>
+    
+    <div class="form-group">
+        <label>Width:</label>
+        <input type="number" id="width" value="1280" min="100" max="1600">
+        <label>Height:</label>
+        <input type="number" id="height" value="720" min="100" max="1200">
+        <label>Quality:</label>
+        <input type="number" id="quality" value="75" min="10" max="100">
+    </div>
+    
+    <button onclick="takeScreenshot()">📸 Take Screenshot</button>
+    <button onclick="testHealth()">🔍 Health Check</button>
+    
+    <div id="result"></div>
+    
+    <script>
+        async function testHealth() {
+            try {
+                const response = await fetch('/');
+                const data = await response.json();
+                document.getElementById('result').innerHTML = 
+                    '<h3>✅ Health Check Result:</h3><pre>' + JSON.stringify(data, null, 2) + '</pre>';
+            } catch (error) {
+                document.getElementById('result').innerHTML = '<p class="error">❌ Error: ' + error.message + '</p>';
+            }
+        }
+        
+        async function takeScreenshot() {
+            const url = document.getElementById('url').value;
+            const width = parseInt(document.getElementById('width').value);
+            const height = parseInt(document.getElementById('height').value);
+            const quality = parseInt(document.getElementById('quality').value);
+            
+            if (!url) {
+                alert('Please enter a URL');
+                return;
+            }
+            
+            document.getElementById('result').innerHTML = '<p class="loading">📸 Taking screenshot...</p>';
+            
+            try {
+                const response = await fetch('/screenshot', {
+                    method: 'POST',
+                    headers: {
+                        'Content-Type': 'application/json',
+                    },
+                    body: JSON.stringify({ url, width, height, quality })
+                });
+                
+                if (response.ok) {
+                    const blob = await response.blob();
+                    const imageUrl = URL.createObjectURL(blob);
+                    const size = (blob.size / 1024).toFixed(1);
+                    
+                    document.getElementById('result').innerHTML = 
+                        '<h3 class="success">✅ Screenshot Success!</h3>' +
+                        '<p>Size: ' + size + ' KB | Dimensions: ' + width + 'x' + height + '</p>' +
+                        '<img src="' + imageUrl + '" alt="Screenshot"><br><br>' +
+                        '<a href="' + imageUrl + '" download="screenshot.jpg">💾 Download Image</a>';
+                } else {
+                    const error = await response.json();
+                    document.getElementById('result').innerHTML = 
+                        '<p class="error">❌ Error: ' + (error.error || error.message) + '</p>';
+                }
+            } catch (error) {
+                document.getElementById('result').innerHTML = 
+                    '<p class="error">❌ Network Error: ' + error.message + '</p>';
+            }
+        }
+    </script>
+</body>
+</html>
+  `);
+});
+
+// 截图端点
+app.post('/screenshot', async (req, res) => {
   const { url, width = 1280, height = 720, quality = 75 } = req.body;
 
+  // 验证参数
   if (!url) {
-    return res.status(400).json({
-      error: 'URL is required',
-      example: { url: 'https://example.com', width: 1280, height: 720 }
-    });
+    return res.status(400).json({ error: 'URL is required' });
   }
 
-  try {
-    const urlObj = new URL(url);
-    if (!['http:', 'https:'].includes(urlObj.protocol)) {
-      return res.status(400).json({
-        error: 'Only HTTP and HTTPS URLs are supported'
-      });
-    }
-  } catch (error) {
-    return res.status(400).json({
-      error: 'Invalid URL format'
+  if (!url.startsWith('http://') && !url.startsWith('https://')) {
+    return res.status(400).json({ error: 'URL must start with http:// or https://' });
+  }
+
+  const cpuUsage = getCpuUsage();
+  if (cpuUsage > 95) {
+    return res.status(503).json({
+      status: 'busy',
+      error: 'Server is currently overloaded',
+      cpuUsage: `${cpuUsage}%`,
+      queueLength: requestQueue.length
     });
   }
 
-  if (width < 100 || width > 1600 || height < 100 || height > 1200) {
-    return res.status(400).json({
-      error: 'Width must be 100-1600px, height must be 100-1200px for HF Spaces'
+  if (activeRequests >= MAX_CONCURRENT) {
+    return res.status(503).json({
+      status: 'busy',
+      error: 'Too many concurrent requests',
+      activeRequests,
+      maxConcurrent: MAX_CONCURRENT
     });
   }
 
+  activeRequests++;
   let browser;
+
   try {
-    console.log(`Taking screenshot of: ${url}`);
-    
     const browserOptions = {
       headless: 'new',
       executablePath: '/usr/bin/google-chrome-stable',
@@ -235,304 +234,68 @@ app.post('/screenshot', queueMiddleware, async (req, res) => {
         '--disable-extensions',
         '--disable-background-timer-throttling',
         '--disable-backgrounding-occluded-windows',
-        '--disable-renderer-backgrounding'
+        '--disable-renderer-backgrounding',
+        // 字体优化设置
+        '--font-render-hinting=none',
+        '--disable-font-subpixel-positioning',
+        '--force-color-profile=srgb',
+        '--disable-features=VizDisplayCompositor'
       ]
     };
 
     browser = await puppeteer.launch(browserOptions);
     const page = await browser.newPage();
     
+    // 设置字体回退CSS - 支持全球字体
+    await page.addStyleTag({
+      content: `
+        * {
+          font-family: "Noto Sans", "Noto Sans CJK SC", "Noto Sans CJK TC", 
+                       "Noto Sans CJK JP", "Noto Sans CJK KR", 
+                       "Noto Color Emoji", system-ui, -apple-system, 
+                       BlinkMacSystemFont, sans-serif !important;
+        }
+      `
+    });
+    
     await page.setViewport({
       width: parseInt(width),
       height: parseInt(height),
       deviceScaleFactor: 1
     });
 
-    await page.setUserAgent('Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36');
-    
-    // 拦截资源以提高性能
-    await page.setRequestInterception(true);
-    page.on('request', (req) => {
-      const resourceType = req.resourceType();
-      if (['font', 'media'].includes(resourceType)) {
-        req.abort();
-      } else {
-        req.continue();
-      }
-    });
-
-    await page.goto(url, {
-      waitUntil: 'domcontentloaded',
-      timeout: 15000
+    await page.goto(url, { 
+      waitUntil: 'domcontentloaded', 
+      timeout: 15000 
     });
 
-    await page.waitForTimeout(1000);
-
     const screenshot = await page.screenshot({
       type: 'jpeg',
-      quality: Math.max(10, Math.min(100, parseInt(quality))),
+      quality: parseInt(quality),
       fullPage: false
     });
 
-    console.log(`Screenshot completed: ${screenshot.length} bytes`);
-
-    res.set({
-      'Content-Type': 'image/jpeg',
-      'Content-Length': screenshot.length,
-      'Cache-Control': 'no-cache',
-      'Content-Disposition': `inline; filename="screenshot-${Date.now()}.jpg"`
-    });
-
+    res.set('Content-Type', 'image/jpeg');
     res.send(screenshot);
 
   } catch (error) {
-    console.error('Screenshot error:', error.message);
-    const errorResponse = {
-      error: 'Failed to capture screenshot',
-      message: error.message
-    };
-    
-    if (error.message.includes('timeout')) {
-      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);
+    console.error('Screenshot error:', error);
+    res.status(500).json({ 
+      error: 'Failed to capture screenshot', 
+      message: error.message 
+    });
   } finally {
     if (browser) {
-      try {
-        await browser.close();
-      } catch (closeError) {
-        console.error('Error closing browser:', closeError.message);
-      }
+      await browser.close();
     }
+    activeRequests--;
   }
 });
 
-// 简化的演示界面
-app.get('/demo', (req, res) => {
-  const demoHtml = `
-    <!DOCTYPE html>
-    <html>
-    <head>
-        <title>📸 Page Screenshot API Demo</title>
-        <meta charset="utf-8">
-        <meta name="viewport" content="width=device-width, initial-scale=1">
-        <style>
-            body { 
-                font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', system-ui, sans-serif; 
-                max-width: 800px; margin: 0 auto; padding: 20px; 
-                background: #f8f9fa;
-            }
-            .container { background: white; padding: 30px; border-radius: 10px; box-shadow: 0 2px 10px rgba(0,0,0,0.1); }
-            .form-group { margin: 20px 0; }
-            label { display: block; margin-bottom: 8px; font-weight: 600; color: #333; }
-            input[type="text"], input[type="number"] { 
-                width: 100%; padding: 12px; border: 2px solid #e1e5e9; 
-                border-radius: 6px; font-size: 16px; box-sizing: border-box;
-            }
-            input:focus { border-color: #007bff; outline: none; }
-            .input-row { display: flex; gap: 15px; }
-            .input-row > div { flex: 1; }
-            button { 
-                background: linear-gradient(135deg, #007bff, #0056b3); 
-                color: white; border: none; padding: 14px 28px; 
-                border-radius: 6px; cursor: pointer; font-size: 16px; font-weight: 600;
-                transition: transform 0.2s; margin: 5px;
-            }
-            button:hover { transform: translateY(-1px); }
-            button:disabled { background: #6c757d; cursor: not-allowed; transform: none; }
-            #result { margin-top: 30px; }
-            .loading { color: #007bff; font-weight: 500; }
-            .error { color: #dc3545; background: #f8d7da; padding: 15px; border-radius: 6px; }
-            .success img { max-width: 100%; border-radius: 6px; box-shadow: 0 4px 15px rgba(0,0,0,0.1); }
-            .examples { margin: 20px 0; }
-            .example-btn { 
-                background: #e9ecef; color: #495057; border: none; 
-                padding: 8px 12px; margin: 5px; border-radius: 4px; cursor: pointer; font-size: 14px;
-            }
-            .example-btn:hover { background: #dee2e6; }
-            .status-info { background: #d1ecf1; padding: 15px; border-radius: 6px; margin-bottom: 20px; border-left: 4px solid #17a2b8; }
-            .hf-info { background: #d4edda; padding: 15px; border-radius: 6px; margin-bottom: 20px; border-left: 4px solid #28a745; }
-        </style>
-    </head>
-    <body>
-        <div class="container">
-            <h1>📸 Page Screenshot API Demo</h1>
-            <p>Professional screenshot service powered by Hugging Face Spaces.</p>
-            
-            <div class="hf-info">
-                <strong>🔐 Access Control</strong><br>
-                This service uses Hugging Face system-level authentication. If this Space is private, access is automatically controlled by HF platform.
-            </div>
-            
-            <div class="status-info">
-                <strong>📊 Server Status:</strong> <span id="serverStatus">Loading...</span>
-            </div>
-            
-            <div class="examples">
-                <strong>Try these examples:</strong><br>
-                <button class="example-btn" onclick="setExample('https://www.google.com')">Google</button>
-                <button class="example-btn" onclick="setExample('https://www.github.com')">GitHub</button>
-                <button class="example-btn" onclick="setExample('https://www.wikipedia.org')">Wikipedia</button>
-                <button class="example-btn" onclick="setExample('https://news.ycombinator.com')">Hacker News</button>
-            </div>
-            
-            <div class="form-group">
-                <label for="url">URL:</label>
-                <input type="text" id="url" placeholder="https://example.com" value="https://www.google.com">
-            </div>
-            
-            <div class="input-row">
-                <div>
-                    <label for="width">Width (px):</label>
-                    <input type="number" id="width" value="1280" min="100" max="1600">
-                </div>
-                <div>
-                    <label for="height">Height (px):</label>
-                    <input type="number" id="height" value="720" min="100" max="1200">
-                </div>
-                <div>
-                    <label for="quality">Quality:</label>
-                    <input type="number" id="quality" value="75" min="10" max="100">
-                </div>
-            </div>
-            
-            <button onclick="takeScreenshot()" id="captureBtn">Take Screenshot</button>
-            <button onclick="checkStatus()" id="statusBtn">Check Status</button>
-            
-            <div id="result"></div>
-        </div>
-        
-        <script>
-            async function checkStatus() {
-                try {
-                    const response = await fetch('/status');
-                    const data = await response.json();
-                    document.getElementById('serverStatus').innerHTML = 
-                        \`CPU: \${data.system.cpuUsage} | Queue: \${data.queue.queuedRequests} | Active: \${data.queue.activeRequests}\`;
-                    
-                    document.getElementById('result').innerHTML = 
-                        '<h3>Server Status:</h3><pre>' + JSON.stringify(data, null, 2) + '</pre>';
-                } catch (error) {
-                    document.getElementById('serverStatus').innerHTML = 'Error loading status';
-                    document.getElementById('result').innerHTML = '<p style="color: red;">Error: ' + error.message + '</p>';
-                }
-            }
-            
-            function setExample(url) {
-                document.getElementById('url').value = url;
-            }
-            
-            async function takeScreenshot() {
-                const url = document.getElementById('url').value;
-                const width = parseInt(document.getElementById('width').value);
-                const height = parseInt(document.getElementById('height').value);
-                const quality = parseInt(document.getElementById('quality').value);
-                const btn = document.getElementById('captureBtn');
-                
-                if (!url) {
-                    document.getElementById('result').innerHTML = '<div class="error">Please enter a URL</div>';
-                    return;
-                }
-                
-                btn.disabled = true;
-                btn.textContent = 'Taking Screenshot...';
-                document.getElementById('result').innerHTML = '<div class="loading">📸 Capturing screenshot, please wait...</div>';
-                
-                try {
-                    const response = await fetch('/screenshot', {
-                        method: 'POST',
-                        headers: {
-                            'Content-Type': 'application/json'
-                        },
-                        body: JSON.stringify({ url, width, height, quality })
-                    });
-                    
-                    if (response.ok) {
-                        const blob = await response.blob();
-                        const imageUrl = URL.createObjectURL(blob);
-                        const size = (blob.size / 1024).toFixed(1);
-                        
-                        document.getElementById('result').innerHTML = 
-                            '<div class="success"><h3>Screenshot Result:</h3>' +
-                            '<p>Size: ' + size + ' KB | Dimensions: ' + width + 'x' + height + '</p>' +
-                            '<img src="' + imageUrl + '" alt="Screenshot"><br><br>' +
-                            '<a href="' + imageUrl + '" download="screenshot.jpg" style="background: #28a745; color: white; padding: 10px 20px; text-decoration: none; border-radius: 5px;">Download Image</a></div>';
-                    } else {
-                        const error = await response.json();
-                        let errorMsg = '<div class="error"><strong>Error:</strong> ' + error.error;
-                        
-                        if (error.status === 'busy') {
-                            errorMsg += '<br><strong>Status:</strong> Server is currently busy (CPU: ' + (error.cpuUsage || 'N/A') + ')';
-                            errorMsg += '<br><strong>Queue Length:</strong> ' + (error.queueLength || 0);
-                        }
-                        
-                        if (error.suggestion) {
-                            errorMsg += '<br><strong>Suggestion:</strong> ' + error.suggestion;
-                        }
-                        
-                        errorMsg += '</div>';
-                        document.getElementById('result').innerHTML = errorMsg;
-                    }
-                } catch (error) {
-                    document.getElementById('result').innerHTML = 
-                        '<div class="error"><strong>Network Error:</strong> ' + error.message + '</div>';
-                } finally {
-                    btn.disabled = false;
-                    btn.textContent = 'Take Screenshot';
-                }
-            }
-            
-            // 页面加载时检查状态
-            checkStatus();
-            
-            // 每30秒更新一次状态
-            setInterval(checkStatus, 30000);
-            
-            // Enter key support
-            document.getElementById('url').addEventListener('keypress', function(e) {
-                if (e.key === 'Enter') {
-                    takeScreenshot();
-                }
-            });
-        </script>
-    </body>
-    </html>
-  `;
-  
-  res.send(demoHtml);
-});
-
-// 错误处理中间件
-app.use((error, req, res, next) => {
-  console.error('Unhandled error:', error);
-  res.status(500).json({
-    error: 'Internal server error'
-  });
-});
-
-// 404处理
-app.use((req, res) => {
-  res.status(404).json({
-    error: 'Endpoint not found'
-  });
-});
-
-// 优雅关闭
-process.on('SIGTERM', () => {
-  console.log('SIGTERM received, shutting down gracefully...');
-  if (cpuMonitorInterval) {
-    clearInterval(cpuMonitorInterval);
-  }
-  process.exit(0);
+app.listen(PORT, () => {
+  console.log(`🚀 Page Screenshot API running on port ${PORT}`);
+  console.log(`📸 Demo: http://localhost:${PORT}/demo`);
+  console.log(`🔍 Health: http://localhost:${PORT}/`);
 });
 
-app.listen(PORT, '0.0.0.0', () => {
-  console.log(`Screenshot API server running on port ${PORT} for Hugging Face Spaces`);
-  console.log('Authentication: Handled by HuggingFace platform (system-level)');
-  console.log('No application-level authentication required');
-});
+module.exports = app;