Added reasoning to response

agents/rag_agent.py

@@ -49,7 +49,11 @@ def load_rag_agent(vectorstore_path: str = "vectorstore/"):
 
     # Load FAISS index
     if os.path.exists("vectorstore/index.faiss"):
-        vectorstore = FAISS.load_local("vectorstore", embeddings)
+        vectorstore = FAISS.load_local(
+            "vectorstore",
+             embeddings,
+             allow_dangerous_deserialization=True
+    )
     else:
         documents = load_documents()   #PDF loader
 
@@ -75,9 +79,22 @@ def load_rag_agent(vectorstore_path: str = "vectorstore/"):
 
     # Grounded prompt
     prompt_template = """You are a helpful HDFC Bank policy assistant.
+
 Use ONLY the context below to answer the customer's question.
-If the answer is not in the context, say "I don't have enough information 
-in the policy documents to answer this. Please contact HDFC Bank directly."
+
+IMPORTANT:
+- Always include the sources at the end of your answer.
+- Also include a short explanation titled "Why this answer?"
+- Explain briefly how the answer was derived from context
+- The sources are provided in the context.
+- Format sources exactly like this:
+
+Sources:
+- file1.pdf
+- file2.pdf
+
+If the answer is not in the context, say:
+"I don't have enough information in the policy documents to answer this. Please contact HDFC Bank directly."
 
 Context:
 {context}
@@ -93,18 +110,19 @@ Answer:"""
 
     def format_docs(docs):
         formatted = []
-        sources = set()
+        sources = []
 
         for doc in docs:
             source = doc.metadata.get("source", "Unknown")
             filename = os.path.basename(source)
 
-            sources.add(filename)
+            sources.append(filename)
             formatted.append(doc.page_content)
 
         context = "\n\n".join(formatted)
 
-        source_text = "\n\nSources:\n" + "\n".join(f"- {s}" for s in sources)
+        unique_sources = list(set(sources))
+        source_text = "\n\nSources:\n" + "\n".join(f"- {s}" for s in unique_sources)
 
         return context + source_text
 

app.py

@@ -87,7 +87,25 @@ if query := st.chat_input("Ask your banking question here..."):
         else:
             st.caption(" Answered by: Data Agent (SQL)")
 
-        st.markdown(response)
+        if "Sources:" in response:
+            answer, sources = response.split("Sources:")
+
+        # Show answer
+            st.markdown(answer)
+
+        # Show sources nicely
+            st.markdown("### Sources")
+            for s in sources.strip().split("\n"):
+                if s.strip():
+                   st.markdown(f"- {s.replace('-', '').strip()}")
+        else:
+            st.markdown(response)
+
+    if "Why this answer?" in response:
+        explanation = response.split("Why this answer?")[1]
+
+        st.markdown("### Why this answer?")
+        st.markdown(explanation)
 
     # Save assistant message
     st.session_state.messages.append({