Update app.py

app.py

@@ -314,22 +314,13 @@ if query:
 
         final_output = context_management_chain({"context":context,"retriever_query":query,"query":query})
 
-        st.markdown("## 💡 **Final Answer**")
+        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("### 📂 **Detailed Processing Breakdown**")
-        st.info("Expand the sections below to see retrieved contexts, relevance scores, and other details.")
-
-        with st.expander("📌 **View Relevant Contexts Used**"):
-            st.markdown("### 📖 **Relevant Contexts Extracted**")
-            if isinstance(final_output["relevant_contexts"], list):
-                for idx, doc in enumerate(final_output["relevant_contexts"], 1):
-                    st.write(f"🔹 **Context {idx}:** {doc}")
-            else:
-                st.warning("⚠️ No relevant contexts available.")
-
-        with st.expander("📌 **View Relevance Analysis**"):
-            st.markdown("### 📊 **Context Relevancy Scores & Analysis**")
+        # 📂 Expand for Details
+        with st.expander("📖 **View Detailed Processing (Relevance, Retrieved Contexts, Analysis)**"):
+            # 📌 Relevance Analysis
+            st.markdown("### 📌 **Relevance Analysis**")
             if isinstance(final_output["relevancy_response"], list):
                 for item in final_output["relevancy_response"]:
                     st.write(f"✅ **Context {item.get('content', 'N/A')} - Score: {item.get('score', 'N/A')}**")
@@ -337,8 +328,16 @@ if query:
             else:
                 st.warning("⚠️ No relevance analysis available.")
 
-        with st.expander("🔍 **View Relevant Context Numbers**"):
-            st.markdown("### 🔢 **Context Numbers Used in Processing**")
+            # 📚 Retrieved Contexts
+            st.markdown("### 📚 **Retrieved Contexts**")
+            if isinstance(final_output["relevant_contexts"], list):
+                for idx, doc in enumerate(final_output["relevant_contexts"], 1):
+                    st.write(f"🔹 **Context {idx}:** {doc}")
+            else:
+                st.warning("⚠️ No retrieved context available.")
+
+            # 🔍 Retrieved Context Numbers
+            st.markdown("### 🔍 **Relevant Context Numbers**")
             st.write(final_output["context_number"])