import os
import gdown
from appdirs import user_cache_dir

# Chexbert
CACHE_DIR = user_cache_dir("vilmedic")
EXTRA_CACHE_DIR = os.path.join(CACHE_DIR, "extras")
os.makedirs(EXTRA_CACHE_DIR, exist_ok=True)

gdown.download(url="https://drive.google.com/uc?id=1UxNmUDX7wJUS-I3M815igCZZokh_VUAS",
               output=os.path.join(EXTRA_CACHE_DIR, "chexbert.pth"),
               quiet=False)

gdown.download(url="https://drive.google.com/uc?id=1RF4GmjyeB2vYy6Y8S3wx-G25ASj2j4HA",
               output=os.path.join(EXTRA_CACHE_DIR, "radgraph.tar.gz"),
               quiet=False)