Update app.py

app.py

@@ -313,31 +313,36 @@ EXAMPLE:
         )
 
         final_output = context_management_chain({"context":context,"retriever_query":query,"query":query})
-
-        # 📌 Display Final Answer
+        # 📌 Display Final Answer (Always Visible)
         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("### 🟥 Relevance Analysis (context_relevancy_evaluation_chain Statement)")
-        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("### 🟦 Picked Relevant Contexts (pick_relevant_context_chain Statement)")
-        if isinstance(final_output["context_number"], list) and final_output["context_number"]:
-            st.write(final_output["context_number"])
-        else:
-            st.warning("⚠️ No relevant context numbers available.")
-
-        st.markdown("### 🟥 Extracted Relevant Contexts (relevant_contexts_chain Statement)")
-        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("---")
+
+        # 🟥 Relevance Analysis (context_relevancy_evaluation_chain Statement)
+        with st.expander("🟥 **Relevance Analysis (context_relevancy_evaluation_chain Statement)**"):
+            if final_output["relevancy_response"]:  # Don't restrict to list, print whatever is there
+                st.write(final_output["relevancy_response"])
+            else:
+                st.warning("⚠️ No relevance analysis available.")
+
+        st.markdown("---")
+
+        # 🟦 Picked Relevant Contexts (pick_relevant_context_chain Statement)
+        with st.expander("🟦 **Picked Relevant Contexts (pick_relevant_context_chain Statement)**"):
+            if final_output["context_number"]:  
+                st.write(final_output["context_number"])
+            else:
+                st.warning("⚠️ No relevant context numbers available.")
+
+        st.markdown("---")
+
+        # 🟥 Extracted Relevant Contexts (relevant_contexts_chain Statement)
+        with st.expander("🟥 **Extracted Relevant Contexts (relevant_contexts_chain Statement)**"):
+            if final_output["relevant_contexts"]:
+                st.write(final_output["relevant_contexts"])
+            else:
+                st.warning("⚠️ No relevant contexts available.")