Update app.py

app.py

@@ -314,28 +314,43 @@ EXAMPLE:
 
         final_output = context_management_chain({"context":context,"retriever_query":query,"query":query})
 
-        #st.subheader('final_output["relevancy_response"]')
-        #st.write(final_output["relevancy_response"] )
-
-        st.write("📌 **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')}**")
-                st.write(f"**Reasoning:** {item.get('Reasoning', 'No explanation provided.')}")
-        else:
-            st.write("⚠️ No relevance analysis available.")
-
-
-        st.subheader('final_output["context_number"]')
-        st.write(final_output["context_number"])
-
-        st.subheader('final_output["relevant_contexts"]')
-        st.write(final_output["relevant_contexts"])
-
-        #st.subheader('final_output["final_response"]')
-        #st.write(final_output["final_response"])
+        # 🎯 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.")
+
+        # 📂 Expandable Detailed Processing Section
+        st.markdown("### 📂 **Detailed Processing Breakdown**")
+        st.info("Expand the sections below to see retrieved contexts, relevance scores, and other details.")
+
+        # 📖 Relevant Contexts Used
+        with st.expander("📌 **View Relevant Contexts Used**"):
+            st.markdown("### 📖 **Relevant Contexts Extracted**")
+            if isinstance(final_output.get("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.")
+
+        # 📊 Relevance Analysis
+        with st.expander("📌 **View Relevance Analysis**"):
+            st.markdown("### 📊 **Context Relevancy Scores & Analysis**")
+            if isinstance(final_output.get("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.")
+
+        # 🔢 Relevant Context Numbers
+        with st.expander("🔍 **View Relevant Context Numbers**"):
+            st.markdown("### 🔢 **Context Numbers Used in Processing**")
+            context_numbers = final_output.get("context_number", "⚠️ No context numbers available.")
+            
+            # 🛠 Handle <think> debug output (if any)
+            if isinstance(context_numbers, str) and "<think>" in context_numbers:
+                context_numbers = "⚠️ No valid context numbers available."
+            
+            st.write(context_numbers)
 
-        st.subheader("📢 **Final Answer:**")
-        st.write(final_output["final_response"] if final_output["final_response"] else "⚠️ No clear answer found based on retrieved content.")