fix

Dockerfile

@@ -66,5 +66,5 @@ COPY . .
 # 暴露端口
 EXPOSE 7860
 
-# 启动应用 - 先初始化playwright，然后运行应用
-CMD ["sh", "-c", "python init_playwright.py && python app.py"] 
+CMD ["sh", "-c", "python init_playwright.py && python app.py"]
+# CMD ["python", "app.py"]

init_playwright.py

@@ -2,11 +2,32 @@
 """
 Playwright初始化脚本
 确保浏览器正确安装和配置
+针对Hugging Face Space环境优化
 """
 
 import subprocess
 import sys
 import os
+from pathlib import Path
+
+def setup_environment():
+    """设置环境变量，解决权限问题"""
+    # 设置缓存目录到用户主目录
+    home_dir = Path.home()
+    cache_dir = home_dir / ".cache"
+    
+    # 确保缓存目录存在
+    cache_dir.mkdir(parents=True, exist_ok=True)
+    
+    # 设置 Playwright 相关环境变量
+    os.environ["PLAYWRIGHT_BROWSERS_PATH"] = str(cache_dir / "ms-playwright")
+    os.environ["PLAYWRIGHT_SKIP_BROWSER_GC"] = "1"
+    
+    # 设置系统缓存目录
+    os.environ["XDG_CACHE_HOME"] = str(cache_dir)
+    
+    print(f"📁 设置缓存目录: {cache_dir}")
+    return str(cache_dir)
 
 def check_playwright_installation():
     """检查playwright是否正确安装"""
@@ -30,21 +51,35 @@ 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)
+        
+        # 使用 --with-deps 选项一次性安装浏览器和依赖
+        result = subprocess.run([
+            sys.executable, "-m", "playwright", "install", 
+            "chromium", "--with-deps"
+        ], capture_output=True, text=True)
+        
+        if result.returncode != 0:
+            print(f"安装输出: {result.stdout}")
+            print(f"错误输出: {result.stderr}")
+            return False
+            
         print("✅ Playwright 浏览器安装完成!")
         return True
     except subprocess.CalledProcessError as e:
         print(f"❌ 浏览器安装失败: {e}")
         return False
+    except Exception as e:
+        print(f"❌ 安装过程中出现异常: {e}")
+        return False
 
 def main():
     """主函数"""
-    print("🚀 初始化 Playwright...")
+    print("🚀 初始化 Playwright (Hugging Face Space 优化版)...")
+    
+    # 首先设置环境变量
+    cache_dir = setup_environment()
     
-    # 首先检查是否已正确安装
+    # 检查是否已正确安装
     if check_playwright_installation():
         return True
     
@@ -60,5 +95,6 @@ if __name__ == "__main__":
     success = main()
     if not success:
         print("❌ Playwright 初始化失败!")
+        print("💡 提示：如果仍有问题，可能需要在系统级别安装浏览器依赖")
         sys.exit(1)
     print("🎉 Playwright 初始化成功!")