Update app.py

app.py

@@ -154,13 +154,29 @@ if query:
         st.write(context, len(context))
     # ----------------- Run Individual Chains Explicitly -----------------
 
+    context_relevancy_checker_prompt = PromptTemplate(input_variables=["retriever_query","context"],template=relevancy_prompt)
 
+    context_relevancy_evaluation_chain = LLMChain(llm=llm_judge, prompt=context_relevancy_checker_prompt, output_key="relevancy_response")
 
+    response_crisis = context_relevancy_evaluation_chain.invoke({"context":context,"retriever_query":query})
 
+    pick_relevant_context_chain = LLMChain(llm=llm_judge, prompt=relevant_prompt, output_key="context_number")
 
+    relevant_response = pick_relevant_context_chain.invoke({"relevancy_response":response_crisis['relevancy_response']})
+
+
+
+
+--------
 
 
     # ----------------- Display All Outputs -----------------
+    st.subheader("response_crisis")
+    st.json((response_crisis))
+
+    st.subheader("response_crisis['relevancy_response']")
+    st.json((response_crisis['relevancy_response']))
+    
     st.markdown("### Context Relevancy Evaluation")
     st.json(response_crisis["relevancy_response"])