ICD prediction based on found dia

app.py

@@ -30,7 +30,7 @@ def text_preparation(text):
     return text
 
 def prompt_creation_icd(clean_text):
-    prompt = ' Exctract ICD-10 codes from this text:\n'+clean_text+'.\n'
+    prompt = ' What are ICD-10 codes for these diagnoses:\n'+clean_text+'.\n'
     return prompt
 
 def prompt_creation_dia(clean_text):
@@ -39,11 +39,12 @@ def prompt_creation_dia(clean_text):
 
 def get_result(text):
     clean_text = text_preparation(text)
-    prompt_icd = prompt_creation_icd(clean_text)
     prompt_dia = prompt_creation_dia(clean_text)
     try:
-        response_icd = GPT_answer(prompt_icd)
         response_dia = GPT_answer(prompt_dia)
+        #creating ICD codes based on found diagnoses
+        prompt_icd = prompt_creation_icd(response_dia)
+        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)
         return answer
     except openai.error.InvalidRequestError as e: