import os

import modal


HF_SECRET_NAME = os.environ.get("VC_HF_SECRET_NAME", "hf-token")
MODEL_ID = os.environ.get("VC_HF_CHECK_MODEL", "google/gemma-4-12B-it")

image = modal.Image.debian_slim(python_version="3.12").uv_pip_install("huggingface-hub>=0.36.0")
app = modal.App("virtual-characters-hf-check", image=image)


@app.function(secrets=[modal.Secret.from_name(HF_SECRET_NAME)], timeout=60)
def check_hf_token() -> dict:
    from huggingface_hub import HfApi

    token = os.environ.get("HF_TOKEN")
    api = HfApi(token=token)
    user = api.whoami()
    model = api.model_info(MODEL_ID)
    return {
        "ok": True,
        "hf_user": user.get("name") or user.get("email"),
        "model_id": model.modelId,
        "private": model.private,
        "gated": getattr(model, "gated", None),
    }


@app.local_entrypoint()
def main():
    print(check_hf_token.remote())