Update app.py

app.py

@@ -164,6 +164,19 @@ if query:
 
     relevant_response = pick_relevant_context_chain.invoke({"relevancy_response":response_crisis['relevancy_response']})
 
+    relevant_contexts_chain = LLMChain(llm=llm_judge, prompt=context_prompt, output_key="relevant_contexts")
+
+    contexts = relevant_contexts_chain.invoke({"context_number":relevant_response['context_number'],"context":context})
+
+    #temp
+    st.subheader("Relevant Contexts")
+    st.json(contexts['relevant_contexts']) 
+
+    response_chain = LLMChain(llm=rag_llm,prompt=final_prompt,output_key="final_response")
+
+    st.subheader("Response Chain")
+    st.json(response_chain)
+
 
 
 
@@ -176,6 +189,9 @@ if query:
 
     st.subheader("response_crisis['relevancy_response']")
     st.json((response_crisis['relevancy_response']))
+
+
+print(relevant_response['context_number'])
     
     st.markdown("### Context Relevancy Evaluation")
     st.json(response_crisis["relevancy_response"])