Update alz_companion/agent.py

alz_companion/agent.py

@@ -479,6 +479,9 @@ def make_rag_chain(vs_general: FAISS, vs_personal: FAISS, *, for_evaluation: boo
     # NEW
     def _answer_fn(query: str, query_type: str, chat_history: List[Dict[str, str]], **kwargs) -> Dict[str, Any]:
 
+        print(f"[DEBUG] The Query is: {query}")
+        print(f"[DEBUG] The Query Type is: {query_type}")
+
         # --- ADD THIS LINE FOR VERIFICATION ---
         print(f"DEBUG: RAG chain received disease_stage = '{disease_stage}'")
         # --- END OF ADDITION ---
@@ -710,20 +713,26 @@ def make_rag_chain(vs_general: FAISS, vs_personal: FAISS, *, for_evaluation: boo
             if disease_stage in ["Moderate Stage", "Advanced Stage"]:
                 # --- STAGE-AWARE LOGIC FOR CAREGIVING SCENARIOS ---
                 if filtered_general_docs:
+                    print("[DEBUG] Stage-Aware Logic: Prioritizing general documents.")
                     all_retrieved_docs = filtered_general_docs
                 elif filtered_personal_docs:
+                    print("[DEBUG] Stage-Aware Logic: Falling back to personal documents.")
                     all_retrieved_docs = filtered_personal_docs
                 else:
+                    print("[DEBUG] Stage-Aware Logic: No relevant documents found.")
                     all_retrieved_docs = []
                 # --- END STAGE-AWARE BLOCK ---
             else:
                 # --- NORMAL ROUTING LOGIC ---
                 # Conditional Blending logic for caregiving remains.
                 if abs(best_personal_score - best_general_score) <= SCORE_MARGIN:
+                    print("[DEBUG] Normal Logic: Blending personal and general docs (scores are close).")
                     all_retrieved_docs = list({doc.page_content: doc for doc in filtered_personal_docs + filtered_general_docs}.values())[:4]
                 elif best_personal_score < best_general_score:
+                    print("[DEBUG] Normal Logic: Prioritizing personal docs (better score).")
                     all_retrieved_docs = filtered_personal_docs
                 else:
+                    print("[DEBUG] Normal Logic: Prioritizing general docs (better score).")
                     all_retrieved_docs = filtered_general_docs
         
         # --- Prompt Generation and LLM Call ---