from pathlib import Path
import json
from huggingface_hub import hf_hub_download

ROOT = Path(__file__).resolve().parent
CFG = json.loads((ROOT / "config.json").read_text(encoding="utf-8-sig"))

llm_dir = ROOT / "models" / "llm"
llm_dir.mkdir(parents=True, exist_ok=True)

target = ROOT / CFG["llm_model_path"]
repo_id = CFG["llm_repo_id"]
filename = CFG["llm_filename"]

print("LOCAL S2S SHELL - MODEL DOWNLOAD")
print("Repo:", repo_id)
print("File:", filename)
print("Target:", target)

if target.exists():
    print("GREEN: LLM already exists.")
    raise SystemExit(0)

downloaded = hf_hub_download(
    repo_id=repo_id,
    filename=filename,
    local_dir=str(llm_dir),
    local_dir_use_symlinks=False
)

downloaded_path = Path(downloaded)

if downloaded_path.resolve() != target.resolve():
    if target.exists():
        target.unlink()
    downloaded_path.rename(target)

print("GREEN: downloaded", target)