import cPickle as pickle

pickle_file = "./waims/src/waims2.pickle"

mdict = pickle.load(open(pickle_file))

def get_candidate_image(word,pos):
    try:
        key = u"{}_{}".format(word.decode('utf-8'),pos) 
    except UnicodeDecodeError:
        key = "{}_{}".format(word,pos)
    if key in mdict.keys():
        return mdict[key]
    else:
        return None