info

Dockerfile

@@ -41,6 +41,12 @@ RUN apt-get update && apt-get install -y \
     lsb-release \
     wget \
     xdg-utils \
+    libnss3-dev \
+    libxss1 \
+    libasound2-dev \
+    libxrandr2 \
+    libxcomposite1 \
+    libxdamage1 \
     && rm -rf /var/lib/apt/lists/*
 
 WORKDIR /app
@@ -60,5 +66,5 @@ COPY . .
 # 暴露端口
 EXPOSE 7860
 
-# 启动应用
-CMD ["python", "app.py"] 
+# 启动应用 - 先初始化playwright，然后运行应用
+CMD ["sh", "-c", "python init_playwright.py && python app.py"] 

app.py

@@ -9,7 +9,7 @@ import logging
 from main import generate_paper_poster
 
 
-logging.basicConfig(level=logging.DEBUG)
+logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 

init_playwright.py

@@ -0,0 +1,64 @@
+#!/usr/bin/env python3
+"""
+Playwright初始化脚本
+确保浏览器正确安装和配置
+"""
+
+import subprocess
+import sys
+import os
+
+def check_playwright_installation():
+    """检查playwright是否正确安装"""
+    try:
+        # 尝试导入playwright
+        from playwright.sync_api import sync_playwright
+        
+        # 尝试启动浏览器
+        with sync_playwright() as p:
+            browser = p.chromium.launch(headless=True)
+            browser.close()
+            
+        print("✅ Playwright 浏览器检查通过!")
+        return True
+        
+    except Exception as e:
+        print(f"❌ Playwright 浏览器检查失败: {e}")
+        return False
+
+def install_browsers():
+    """安装playwright浏览器"""
+    try:
+        print("🔄 正在安装 Playwright 浏览器...")
+        subprocess.run([sys.executable, "-m", "playwright", "install", "chromium"], 
+                      check=True)
+        subprocess.run([sys.executable, "-m", "playwright", "install-deps", "chromium"], 
+                      check=True)
+        print("✅ Playwright 浏览器安装完成!")
+        return True
+    except subprocess.CalledProcessError as e:
+        print(f"❌ 浏览器安装失败: {e}")
+        return False
+
+def main():
+    """主函数"""
+    print("🚀 初始化 Playwright...")
+    
+    # 首先检查是否已正确安装
+    if check_playwright_installation():
+        return True
+    
+    # 如果检查失败，尝试重新安装
+    print("🔧 检测到问题，正在重新安装浏览器...")
+    if install_browsers():
+        # 重新检查
+        return check_playwright_installation()
+    
+    return False
+
+if __name__ == "__main__":
+    success = main()
+    if not success:
+        print("❌ Playwright 初始化失败!")
+        sys.exit(1)
+    print("🎉 Playwright 初始化成功!")