def findDisease(cui, lang):
look_umls = UMLSLookup()
meaning_umls = look_umls.lookup_code_meaning(cui,
lat=lang,
preferred=False)
if meaning_umls:
return str(meaning_umls[0])
else:
#disease = str(all_csv.loc[all_csv['Disease'] == cui]["Disease_UMLS"].unique()[0].encode('utf8'))
print("Using fallback CSV for: " + cui)
return cui
def findSymptom(cui, lang):
look_umls = UMLSLookup()
meaning_umls = look_umls.lookup_code_meaning(cui,
lat=lang,
preferred=False)
if meaning_umls:
return str(meaning_umls[0])
else:
print("Using fallback CSV for: " + cui)
return cui
def findFeatures(cui, lang):
symptom_strings = []
look_umls = UMLSLookup()
for s_cui in getRelatedSymptomsForDisease(cui):
meaning_umls = look_umls.lookup_code_meaning(s_cui,
lat=lang,
preferred=False)
if meaning_umls:
symptom_strings.append(meaning_umls[0])
else:
symptom_strings.append(
all_csv.loc[all_csv['Disease'] == cui]["Symptom"].unique()[0])
return symptom_strings