Update main.py

main.py

@@ -199,11 +199,17 @@ async def llm_response(chain,id,mode):
         diagnosis_and_treatment = str(diagnosis_and_treatment['result'])
 
         if mode.lower()!="h&p":
+            print(diagnosis_and_treatment)
+            print("dirrect answer")
             return diagnosis_and_treatment
+
         else:
             report = qa_chain_a(report_prompt_template+sys+chain+"\n\ntreatment & diagnosis with ICD code below\n"+diagnosis_and_treatment)
+            print(str(report['result']))
+            print("h&P")
             return str(report['result'])
             
+            
     result_ex = qa_chain(sys+chain+"""\n\n\nalways give small and single response based on the patient 
         response. don't give multiline response always give response based on last patient response""")
     if "Patient:" in str(result_ex['result']) or "Patient response:" in str(result_ex['result']) or "Patient Response" in str(result_ex['result']):