playwright

Dockerfile

@@ -0,0 +1,65 @@
+FROM python:3.10-slim
+
+# 安装系统依赖
+RUN apt-get update && apt-get install -y \
+    wget \
+    gnupg \
+    ca-certificates \
+    fonts-liberation \
+    libasound2 \
+    libatk-bridge2.0-0 \
+    libatk1.0-0 \
+    libc6 \
+    libcairo2 \
+    libcups2 \
+    libdbus-1-3 \
+    libexpat1 \
+    libfontconfig1 \
+    libgcc1 \
+    libgconf-2-4 \
+    libgdk-pixbuf2.0-0 \
+    libglib2.0-0 \
+    libgtk-3-0 \
+    libnspr4 \
+    libnss3 \
+    libpango-1.0-0 \
+    libpangocairo-1.0-0 \
+    libstdc++6 \
+    libx11-6 \
+    libx11-xcb1 \
+    libxcb1 \
+    libxcomposite1 \
+    libxcursor1 \
+    libxdamage1 \
+    libxext6 \
+    libxfixes3 \
+    libxi6 \
+    libxrandr2 \
+    libxrender1 \
+    libxss1 \
+    libxtst6 \
+    lsb-release \
+    wget \
+    xdg-utils \
+    && rm -rf /var/lib/apt/lists/*
+
+WORKDIR /app
+
+# 复制requirements文件
+COPY requirements.txt .
+
+# 安装Python依赖
+RUN pip install --no-cache-dir -r requirements.txt
+
+# 安装playwright浏览器
+RUN playwright install chromium
+RUN playwright install-deps chromium
+
+# 复制应用代码
+COPY . .
+
+# 暴露端口
+EXPOSE 7860
+
+# 启动应用
+CMD ["python", "app.py"] 

README.md

@@ -3,9 +3,8 @@ title: P2P Paper-to-Poster Generator
 emoji: 🎓
 colorFrom: blue
 colorTo: purple
-sdk: gradio
-sdk_version: 5.32.1
-app_file: app.py
+sdk: docker
+app_port: 7860
 pinned: false
 license: mit
 ---
@@ -19,10 +18,15 @@ license: mit
 
 ## 🚀 Try it on Hugging Face Spaces
 
-This application is deployed on Hugging Face Spaces! You can try it directly in your browser without any installation:
+This application is deployed on Hugging Face Spaces using Docker SDK to support advanced dependencies like Playwright! You can try it directly in your browser without any installation:
 
 **🎓 [Launch P2P Paper-to-Poster Generator](https://huggingface.co/spaces/ASC8384/P2P)**
 
+### Docker Deployment on HF Spaces:
+1. **SDK**: Uses `docker` instead of `gradio` to support system-level dependencies
+2. **Playwright Support**: Automatically installs and configures Playwright browsers
+3. **Pre-built Environment**: No manual setup required for complex dependencies
+
 ### Quick Start on Spaces:
 1. Upload your PDF research paper
 2. Enter your OpenAI API key and base URL (if using proxy)
@@ -41,6 +45,7 @@ This application is deployed on Hugging Face Spaces! You can try it directly in
 - Direct JSON structure display
 - Support for multiple AI models
 - Flexible API configuration
+- Advanced layout optimization with Playwright
 
 ## Overview
 

main.py

@@ -11,7 +11,6 @@ from poster.poster import (
     generate_poster_v3,
     replace_figures_in_poster,
     replace_figures_size_in_poster,
-    take_screenshot,
 )
 
 

poster/poster.py

@@ -451,15 +451,15 @@ Now there are {proportion:.0%} blank spaces. Please regenerate the content to cr
         ).content
 
 
-def take_screenshot(output: str, html: str):
-    with sync_playwright() as p:
-        browser = p.chromium.launch(headless=True)
-        page = browser.new_page(viewport={"width": 1280, "height": 100})
-        page.set_content(html)
-        page.screenshot(
-            type="png", path=output.replace(".json", ".png"), full_page=True
-        )
-        browser.close()
+# def take_screenshot(output: str, html: str):
+#     with sync_playwright() as p:
+#         browser = p.chromium.launch(headless=True)
+#         page = browser.new_page(viewport={"width": 1280, "height": 100})
+#         page.set_content(html)
+#         page.screenshot(
+#             type="png", path=output.replace(".json", ".png"), full_page=True
+#         )
+#         browser.close()
 
 
 def replace_figures_in_svg(svg: str, figures: list[str]) -> str: