Upload 3 files

Dockerfile

@@ -0,0 +1,41 @@
+FROM python:3.9
+
+# システムのアップデート & Playwright 用依存ライブラリのインストール
+RUN apt-get update && apt-get install -y \
+    libnss3 \
+    libatk1.0-0 \
+    libatk-bridge2.0-0 \
+    libcups2 \
+    libdrm2 \
+    libxkbcommon0 \
+    libwayland-egl1 \
+    libwayland-cursor0 \
+    libwayland-server0 \
+    libgbm1 \
+    fonts-liberation \
+    libasound2 \
+    wget \
+    git \
+    && rm -rf /var/lib/apt/lists/*
+
+WORKDIR /code
+
+# requirements.txt をコピーしてインストール
+COPY ./requirements.txt /code/requirements.txt
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+# Playwright でブラウザをインストール
+RUN python -m playwright install
+
+# Hugging Face Spaces で実行ユーザーを合わせるための設定
+RUN useradd -m -u 1000 user
+USER user
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:$PATH
+WORKDIR $HOME/app
+
+# ソースコードをコピー
+COPY --chown=user . $HOME/app
+
+# コンテナ起動時に FastAPI を立ち上げる
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

main.py

@@ -0,0 +1,47 @@
+import os
+
+from fastapi import FastAPI
+from playwright.sync_api import sync_playwright
+
+app = FastAPI()
+
+@app.get("/")
+def read_root():
+    """
+    動作確認用のエンドポイント。
+    """
+    return {"message": "Hello from Hugging Face Spaces. Visit /get_mhtml to save MHTML."}
+
+@app.get("/get_mhtml")
+def get_mhtml():
+    """
+    https://civitai.com/models/1055452/akashicpulse を開き、
+    ブラウザレンダリング後のページを MHTML として保存。
+    """
+    url = "https://civitai.com/models/1055452/akashicpulse"
+    output_path = "akashicpulse.mhtml"
+    try:
+        with sync_playwright() as p:
+            browser = p.chromium.launch(headless=True)
+            page = browser.new_page()
+            page.goto(url, wait_until="networkidle")
+
+            # Chrome DevTools Protocol セッションを作成し、MHTML 取得
+            cdp_session = page.context.new_cdp_session(page)
+            snapshot = cdp_session.send("Page.captureSnapshot", {"format": "mhtml"})
+
+            browser.close()
+
+        # ファイルに書き出し
+        with open(output_path, "w", encoding="utf-8") as f:
+            f.write(snapshot)
+
+        return {
+            "status": "success",
+            "message": f"MHTML saved to {output_path}",
+        }
+    except Exception as e:
+        return {
+            "status": "error",
+            "message": str(e),
+        }

requirements.txt

@@ -0,0 +1,3 @@
+fastapi
+uvicorn
+playwright