import os
from huggingface_hub import hf_hub_download

MODEL_DIR = "models"

def get_model(filename: str):
    local_path = os.path.join(MODEL_DIR, filename)

    # 🟢 Local dev: model exists
    if os.path.exists(local_path):
        print(f"[LOCAL] Using model: {local_path}")
        return local_path

    # 🔵 HF Space: download model
    print(f"[HF] Downloading model: {filename}")
    os.makedirs(MODEL_DIR, exist_ok=True)

    return hf_hub_download(
        repo_id="rajitb/geoai-models",   # CHANGE IF NEEDED
        filename=filename,
        local_dir=MODEL_DIR
    )