Update app.py

app.py

@@ -459,7 +459,7 @@ def update_knowledge_base(user_input: str, assistant_response: str):
 full_pipeline = hybrid_chain | RunnableAssign({"validation": validation_chain}) | answer_chain
 
 
-def chat_interface(message, history):
+def chat_interface(message, history):    
     inputs = {
         "query": message,
         "all_queries": [message],
@@ -469,10 +469,14 @@ def chat_interface(message, history):
         "vectorstore": vectorstore,
         "bm25_retriever": bm25_retriever,
     }
+
+    hybrid_result = hybrid_chain.invoke(inputs)
+    hybrid_result["validation"] = validation_chain.invoke(hybrid_result)
+    
     full_response = ""
 
     # Stream the response to user
-    for chunk in full_pipeline.stream(inputs):
+    for chunk in answer_chain.stream(hybrid_result):
         if isinstance(chunk, dict) and "answer" in chunk:
             full_response += chunk["answer"]
             yield full_response