Update alz_companion/agent.py

alz_companion/agent.py

@@ -809,19 +809,23 @@ def make_rag_chain(vs_general: FAISS, vs_personal: FAISS, *, for_evaluation: boo
             user_prompt = template.format(personal_context=personal_context, general_context=general_context, question=query, language=language, patient_name=p_name, caregiver_name=c_name, context=personal_context, role=role)
             print("[DEBUG] Personal Route Factual / Sum / Multi PROMPT") 
         else: # caregiving_scenario
-            # NEXT evolution
-            # if settings.get("role") == "patient":
-            # use template = ANSWER_TEMPLATE_PATIENT
-            # else template = ANSWER_TEMPLATE_ADQ
             
             if disease_stage == "Advanced Stage": template = ANSWER_TEMPLATE_ADQ_ADVANCED
             elif disease_stage == "Moderate Stage": template = ANSWER_TEMPLATE_ADQ_MODERATE
-            else: template = ANSWER_TEMPLATE_ADQ
-                
+            else: 
+                # template = ANSWER_TEMPLATE_ADQ
+                # NEXT evolution
+                if settings.get("role") == "patient":
+                    template = ANSWER_TEMPLATE_PATIENT
+                    print("[DEBUG] Use ANSWER_TEMPLATE_PATIENT")
+                else 
+                    template = ANSWER_TEMPLATE_ADQ
+                    print("[DEBUG] Use ANSWER_TEMPLATE_ADQ")
+            
             emotions_context = render_emotion_guidelines(kwargs.get("emotion_tag"))
             # NEXT evolution
             # if emotion in ["confusion", "sadness", "anxiety", "orientation_check"]:
-            #    template = ANSWER_TEMPLATE_CAL
+            #    template = ANSWER_TEMPLATE_CALM
             user_prompt = template.format(general_context=general_context, personal_context=personal_context, question=query, scenario_tag=kwargs.get("scenario_tag"), emotions_context=emotions_context, role=role, language=language, patient_name=p_name, caregiver_name=c_name, emotion_tag=kwargs.get("emotion_tag"))
             print("[DEBUG] Caregiving Scenario PROMPT") 
         # end