Update app.py

app.py

@@ -42,7 +42,7 @@ rag_llm = ChatGroq(model="mixtral-8x7b-32768")
 llm_judge.verbose = True
 rag_llm.verbose = True
 
-# ----------------- PDF Selection -----------------
+# ----------------- PDF Selection (Upload or URL) -----------------
 st.sidebar.subheader("📂 PDF Selection")
 pdf_source = st.radio("Choose a PDF source:", ["Upload a PDF file", "Enter a PDF URL"], index=0, horizontal=True)
 
@@ -131,11 +131,28 @@ if query:
         ).invoke({"context": context, "retriever_query": query, "query": query})
 
     # ----------------- Display All Outputs -----------------
-    st.subheader("🟥 Context Relevancy Evaluation")
+    st.markdown("### 🟥 Context Relevancy Evaluation")
     st.json(final_output["relevancy_response"])
-    st.subheader("🟦 Picked Relevant Contexts")
+
+    st.markdown("### 🟦 Picked Relevant Contexts")
+    st.json(final_output["context_number"])
+
+    st.markdown("### 🟥 Extracted Relevant Contexts")
+    st.json(final_output["relevant_contexts"])
+
+    st.markdown("## 🟥 RAG Final Response")
+    st.write(final_output["final_response"])
+
+    # ----------------- Streamlit-Friendly Debugging (Replacing print statements) -----------------
+    st.markdown("### Debug Logs:")
+    st.text("\n-------- 🟥 Context Relevancy Evaluation Statement 🟥 --------\n")
+    st.json(final_output["relevancy_response"])
+
+    st.text("\n-------- 🟦 Picked Relevant Context Statement 🟦 --------\n")
     st.json(final_output["context_number"])
-    st.subheader("🟥 Extracted Relevant Contexts")
+
+    st.text("\n-------- 🟥 Relevant Contexts Statement 🟥 --------\n")
     st.json(final_output["relevant_contexts"])
-    st.subheader("🟥 RAG Final Response")
-    st.write(final_output["final_response"])
+
+    st.text("\n-------- 🟥 RAG Response Statement 🟥 --------\n")
+    st.write(final_output["final_response"])