output cleaning

app.py

@@ -45,8 +45,8 @@ def get_result(text):
         response_dia = GPT_answer(prompt_dia)
         answer = 'diagnoses: '+response_dia
         #creating ICD codes based on found diagnoses
-        diagnoses = response_dia.lower().split('\n')
-        diagnoses_cleaned = "\n".join([dia for dia in diagnoses if 'kein' not in dia])
+        diagnoses = response_dia.split('\n')
+        diagnoses_cleaned = "\n".join([dia for dia in diagnoses if 'kein' not in dia.lower()])
         prompt_icd = prompt_creation_icd(diagnoses_cleaned)
         response_icd = GPT_answer(prompt_icd)
         answer += 'Found DIAGNOSES:'+diagnoses_cleaned+'\nICD-10 CODES:'+response_icd