Update pipeline.py

pipeline.py

@@ -178,15 +178,15 @@ def build_rag_chain(llm_model: LiteLLMModel, vectorstore: FAISS) -> RetrievalQA:
 def run_pipeline(query: str) -> str:
     try:
         query = sanitize_message(query)
-        
+        topic=extract_main_topic(query)
         moderation_result = moderate_text(query)
         if not moderation_result.is_safe:
             return "Sorry, this query contains harmful or inappropriate content."
 
         classification = classify_query(moderation_result.original_text)
-
+        
         if classification == "OutOfScope":
-            refusal_text = refusal_chain.run({"topic": "this topic"})
+            refusal_text = refusal_chain.run({"topic": topic})
             return tailor_chain.run({"response": refusal_text}).strip()
 
         if classification == "Wellness":
@@ -201,7 +201,7 @@ def run_pipeline(query: str) -> str:
             csv_answer = rag_result["result"].strip()
             final_merged = merge_responses(csv_answer, "")
             return tailor_chain.run({"response": final_merged}).strip()
-
+        
         refusal_text = refusal_chain.run({"topic": "this topic"})
         return tailor_chain.run({"response": refusal_text}).strip()
     except Exception as e: