import os
import gdown
from appdirs import user_cache_dir
import stanza

# stanza
stanza.download('en', processors='tokenize,ner')
stanza.download('en', package='radiology')


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

extras = [
("1UxNmUDX7wJUS-I3M815igCZZokh_VUAS", "chexbert.pth"),
("1RF4GmjyeB2vYy6Y8S3wx-G25ASj2j4HA", "radgraph.tar.gz"),
("17ZKYgS7boxMB9xHwrp0hOzpl96oLlF7T","model_medrad_19k.gz"),

]

for id, name in extras:
    output = os.path.join(EXTRA_CACHE_DIR, name)
    if not os.path.exists(output):
        gdown.download(url="https://drive.google.com/uc?id={}".format(id),
                       output=output,
                       quiet=False)