'''

icd code maps to description 

input: "icdcode/ICD-10-Medical-Diagnosis-Codes.pdf"

output:  "icdcode/description2icd10.pkl" 

'''

import icd10, PyPDF2, pickle, os, csv 

# def code2description(code):

#   # code = "A1803"

#   code = icd10.find(code)

#   description = code.description 

#   return description 

# def extract_icdcode(file, pkl_file):

#   f = open(file, 'rb')

#   description2icd = dict()

#   fileReader = PyPDF2.PdfFileReader(file)

#   for pagenum in range(fileReader.numPages):

#       pageObj = fileReader.getPage(pagenum)

#       text = pageObj.extractText()

#       text = text.split('\n')

#       text = list(filter(lambda x:len(x.strip())>0, text))

#       if pagenum == 0:

#           text = list(filter(lambda x:x[0]=='A', text))

#           print(text)

#       for i in text:

#           icd = i.split()[0]

#           description = ' '.join(i.strip().split()[1:]).lower()

#           description2icd[description] = icd 

#           # print(description, icd)

#   print("code number is", str(len(description2icd)))

#   # code number is 76331

#   pickle.dump(description2icd, open(pkl_file, 'wb'))

###  csv_file = 'icdcode/icd_10_direct_mapping.csv'

def extract_icdcode(csv_file, pkl_file):

    with open(csv_file, 'r') as csvfile:

        rows = list(csv.reader(csvfile, delimiter=','))

    description2icd = dict() 

    for row in rows:

        icd = row[0]

        description = row[1]

        print(icd)

        description2icd[description] = icd

    print("code number is", str(len(description2icd)))  

    pickle.dump(description2icd, open(pkl_file, 'wb'))

if __name__ == "__main__":

    # file = "icdcode/ICD-10-Medical-Diagnosis-Codes.pdf"

    csv_file = 'icdcode/icd_10_direct_mapping.csv'

    pkl_file = "icdcode/description2icd10.pkl"

    if not os.path.exists(pkl_file):

        extract_icdcode(csv_file, pkl_file)

    description2icd10 = pickle.load(open(pkl_file, 'rb'))

    # for description, icd in description2icd10.items():

    #   #if len(description.split())==1:

    #   print(description, "  -->  ", icd)