import csv 

def nctid2label_dict():

    nctid2outcome = dict() 

    outcome2label = dict() 

    nctid2label = dict() 

    with open("trialtrove/outcome2label.txt", 'r') as fin: 

        lines = fin.readlines() 

        for line in lines:

            outcome = line.split('\t')[0]

            label = line.strip().split('\t')[1]

            outcome2label[outcome] = label 

    with open("trialtrove/trial_outcomes_v1.csv", 'r') as csvfile: 

        csvreader = list(csv.reader(csvfile))[1:]

        for row in csvreader:

            nctid = row[0]

            outcome = row[1]

            nctid2outcome[nctid] = outcome 

    for nctid,outcome in nctid2outcome.items():

        nctid2label[nctid] = outcome2label[outcome]

    return nctid2label 

nctid2label = nctid2label_dict() 

print(nctid2label)