dia prompt changes, negations removal

app.py

@@ -34,7 +34,8 @@ def prompt_creation_icd(clean_text):
     return prompt
 
 def prompt_creation_dia(clean_text):
-    prompt = ' Exctract diagnoses names from this text:\n'+clean_text+'.\n'
+    prompt = " Extract patient's diagnoses from this text:\n"+clean_text+".\n"
+    prompt += "Print each diagnose from the new line."
     return prompt
 
 def get_result(text):
@@ -43,9 +44,11 @@ def get_result(text):
     try:
         response_dia = GPT_answer(prompt_dia)
         #creating ICD codes based on found diagnoses
-        prompt_icd = prompt_creation_icd(response_dia)
+        diagnoses = response_dia.lower().split('\n')
+        diagnoses_cleaned = "\n".join([dia for dia in diagnoses if 'kein' not in dia])
+        prompt_icd = prompt_creation_icd(diagnoses_cleaned)
         response_icd = GPT_answer(prompt_icd)
-        answer = 'ICD-10 CODES:\n'+re.sub('\n', '', response_icd)+'\n\nFound DIAGNOSES:\n'+re.sub('\n', '', response_dia)
+        answer = 'Found DIAGNOSES:\n'+re.sub('\n', '', diagnoses_cleaned)+'\nICD-10 CODES:\n'+re.sub('\n', '', response_icd)
         return answer
     except openai.error.InvalidRequestError as e:
         # Handle the InvalidRequestError
@@ -62,3 +65,5 @@ iface = gr.Interface(
     outputs=gr.outputs.Textbox(label="Found ICD-10 codes and diagnoses"),
 )
 iface.launch()
+
+#%%