Update main.py

main.py

@@ -181,18 +181,18 @@ async def llm_response(chain,id,mode):
                 same question again, and don't provide treatment and diagnosis ask next small and short question with yes or no format , 
                 always don't ask same question again and again , always only ask next single small question""")
             try:
-                if "Patient:" in str(question['result']) or "Patient response:" in str(question['result']):
+                if "Patient:" in str(question['result']) or "Patient response:" in str(question['result']) or "Patient Response" in str(question['result']):
                     return str((str(question['result']).split("\n\n")[-1]).split(":")[-1])
                 else:
                     return str(question['result']).split("\n\n")[1]
 
             except:
-                if "Patient:" in str(question['result']) or "Patient response:" in str(question['result']):
+                if "Patient:" in str(question['result']) or "Patient response:" in str(question['result']) or "Patient Response" in str(question['result']):
                    return str(question['result']).split(":")[-1]
                 else:
                    return str(question['result'])
 
-    if id==16:
+    if id==13:
         diagnosis_and_treatment = qa_chain(sys+chain+end_sys_prompts)
         diagnosis_and_treatment = str(diagnosis_and_treatment['result'])