output cleaning

app.py

@@ -43,13 +43,12 @@ def get_result(text):
     prompt_dia = prompt_creation_dia(clean_text)
     try:
         response_dia = GPT_answer(prompt_dia)
-        answer = 'diagnoses: '+response_dia
         #creating ICD codes based on found diagnoses
         diagnoses = response_dia.split('\n')
-        diagnoses_cleaned = "\n".join([dia for dia in diagnoses if 'kein' not in dia.lower()])
+        diagnoses_cleaned = "\n".join([dia for dia in diagnoses if 'kein' or 'nicht' 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
+        answer = 'Found DIAGNOSES:'+diagnoses_cleaned+'\nICD-10 CODES:'+response_icd
         return answer
     except openai.error.InvalidRequestError as e:
         # Handle the InvalidRequestError
@@ -67,4 +66,3 @@ iface = gr.Interface(
 )
 iface.launch()
 
-#%%