Update app.py

app.py

@@ -174,11 +174,32 @@ if query:
 
     response_chain = LLMChain(llm=rag_llm,prompt=final_prompt,output_key="final_response")
 
+
+    #temp
     st.subheader("Response Chain")
     st.json(response_chain)
 
+    response = chain.invoke({"query":query,"context":contexts['relevant_contexts']})
 
-
+    #temp
+    st.subheader("blah response")
+    st.json(response.content)
+
+    # Orchestrate using SequentialChain
+    context_management_chain = SequentialChain(
+        chains=[context_relevancy_evaluation_chain ,pick_relevant_context_chain, relevant_contexts_chain,response_chain],
+        input_variables=["context","retriever_query","query"],
+        output_variables=["relevancy_response", "context_number","relevant_contexts","final_response"]
+    )
+
+   final_output = context_management_chain({"context":context,"retriever_query":query,"query":query})
+   st.subheader("Final Output from Context Management chain")
+   st.json(final_output)
+
+   st.subheader("Context of Final Output from Context Management chain")
+   st.json(final_output['context'])
+   
+   
 
 --------