Upload 3 files

Dockerfile

@@ -1,6 +1,6 @@
 FROM python:3.9
 
-# システムのアップデート & Playwright 用依存ライブラリのインストール
+# 必要なライブラリのインストール
 RUN apt-get update && apt-get install -y \
     libnss3 \
     libatk1.0-0 \
@@ -14,28 +14,23 @@ RUN apt-get update && apt-get install -y \
     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 でブラウザをインストール
+# Playwright ブラウザ本体をインストール
 RUN python -m playwright install
 
-# Hugging Face Spaces で実行ユーザーを合わせるための設定
+# Hugging Face Spaces の実行ユーザーと作業ディレクトリ
 RUN useradd -m -u 1000 user
 USER user
 ENV HOME=/home/user \
     PATH=/home/user/.local/bin:$PATH
-WORKDIR $HOME/app
 
-# ソースコードをコピー
+WORKDIR $HOME/app
 COPY --chown=user . $HOME/app
 
-# コンテナ起動時に FastAPI を立ち上げる
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

main.py

@@ -1,47 +1,32 @@
-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 として保存。
-    """
+@app.on_event("startup")
+def on_startup():
+    # 起動時に自動的に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"})
+            cdp = page.context.new_cdp_session(page)
+            snapshot = cdp.send("Page.captureSnapshot", {"format": "mhtml"})
 
             browser.close()
 
-        # ファイルに書き出し
         with open(output_path, "w", encoding="utf-8") as f:
             f.write(snapshot)
+        print(f"[INFO] MHTML saved to: {output_path}")
 
-        return {
-            "status": "success",
-            "message": f"MHTML saved to {output_path}",
-        }
     except Exception as e:
-        return {
-            "status": "error",
-            "message": str(e),
-        }
+        print(f"[ERROR] Failed to save MHTML: {e}")
+
+@app.get("/")
+def read_root():
+    return {"message": "Startup event attempted to save MHTML to akashicpulse.mhtml."}