Update app.py

app.py

@@ -314,10 +314,40 @@ EXAMPLE:
 
         final_output = context_management_chain({"context":context,"retriever_query":query,"query":query})
 
-        # 🎯 Final Answer Section
         st.markdown("## 💡 **Final Answer**")
         st.success(final_output["final_response"] if final_output["final_response"] else "⚠️ No clear answer found based on retrieved content.")
 
+        st.markdown("---")
+
+        # Display Relevant Contexts Extracted
+        st.markdown("### 📌 **Relevant Contexts Used**")
+        if isinstance(final_output["relevant_contexts"], list) and final_output["relevant_contexts"]:
+            for idx, doc in enumerate(final_output["relevant_contexts"], 1):
+                st.write(f"🔹 **Context {idx}:** {doc}")
+        else:
+            st.warning("⚠️ No relevant contexts available.")
+
+        st.markdown("---")
+
+        # Display Relevance Analysis
+        st.markdown("### 📊 **Relevance Analysis**")
+        if isinstance(final_output["relevancy_response"], list) and final_output["relevancy_response"]:
+            for item in final_output["relevancy_response"]:
+                st.write(f"✅ **Context {item.get('content', 'N/A')} - Score: {item.get('score', 'N/A')}**")
+                st.write(f"**Reasoning:** {item.get('Reasoning', 'No explanation provided.')}")
+        else:
+            st.warning("⚠️ No relevance analysis available.")
+
+        st.markdown("---")
+
+        # Display Relevant Context Numbers
+        st.markdown("### 🔍 **Relevant Context Numbers**")
+        if final_output["context_number"]:
+            st.write(final_output["context_number"])
+        else:
+            st.warning("⚠️ No relevant context numbers available.")
+
+