# app.py
import os
from fastapi import FastAPI
from huggingface_hub import login

app = FastAPI()

@app.get("/")
def root():
    login(os.environ.get("HUGGINGFACE_HUB_TOKEN", ""))
    return {"status": "ok"}

# app.py
import os
from fastapi import FastAPI
from huggingface_hub import hf_hub_download

# 必ず import の前に環境変数を設定！
os.environ["HF_HOME"] = "/tmp/hf"
os.environ["HF_HUB_CACHE"] = "/tmp/hf/hub"
os.environ["HUGGINGFACE_HUB_CACHE"] = "/tmp/hf/hub"
os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf/transformers"
os.environ["XDG_CACHE_HOME"] = "/tmp/hf/xdg"

app = FastAPI()

@app.get("/")
def root():
    model_path = hf_hub_download(
        repo_id="revi13/ip-adapter-faceid-private",
        filename="ip-adapter-faceid-plusv2_sd15.bin",
        local_dir="/tmp/hf/models",  # 保存先を手動指定
        local_dir_use_symlinks=False,  # シンボリックリンクではなく実体保存
        token=os.environ["HUGGINGFACE_HUB_TOKEN"]
    )
    return {"model_path": model_path}