corrected the

pipeline.py

@@ -35,11 +35,11 @@ from langchain_core.callbacks import Callbacks
 
 # Custom chain imports
 # from groq_client import GroqClient
-from HealthyAiExpert.chain.classification_chain import get_classification_chain
-from HealthyAiExpert.chain.refusal_chain import get_refusal_chain
-from HealthyAiExpert.chain.tailor_chain import get_tailor_chain
-from HealthyAiExpert.chain.cleaner_chain import get_cleaner_chain
-from HealthyAiExpert.chain.tailor_chain_wellnessBrand import get_tailor_chain_wellnessBrand
+from chain.classification_chain import get_classification_chain
+from chain.refusal_chain import get_refusal_chain
+from chain.tailor_chain import get_tailor_chain
+from chain.cleaner_chain import get_cleaner_chain
+from chain.tailor_chain_wellnessBrand import get_tailor_chain_wellnessBrand
 
 # Mistral moderation
 from mistralai import Mistral
@@ -66,10 +66,8 @@ from smolagents import (
 )
 
 # Import new prompts
-from HealthyAiExpert.chain.prompts import (
-    selfharm_prompt, frustration_prompt, ethical_conflict_prompt,
-    classification_prompt, refusal_prompt, tailor_prompt, cleaner_prompt
-)
+from chain.prompts import selfharm_prompt, frustration_prompt, ethical_conflict_prompt,classification_prompt, refusal_prompt, tailor_prompt, cleaner_prompt
+
 
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
@@ -488,20 +486,20 @@ class PipelineState:
         self.cleaner_chain        = get_cleaner_chain()
 
         # Specialized chain for self-harm
-        from HealthyAiExpert.chain.prompts import selfharm_prompt
+        from chain.prompts import selfharm_prompt
         # self.self_harm_chain = LLMChain(llm=gemini_llm, prompt=selfharm_prompt, verbose=False)
         
         self.self_harm_chain = LLMChain(llm=groq_fallback_llm, prompt=selfharm_prompt, verbose=False)
         
         
         # NEW: chain for frustration/harsh queries
-        from HealthyAiExpert.chain.prompts import frustration_prompt
+        from chain.prompts import frustration_prompt
         # self.frustration_chain = LLMChain(llm=gemini_llm, prompt=frustration_prompt, verbose=False)
         self.frustration_chain = LLMChain(llm=groq_fallback_llm, prompt=frustration_prompt, verbose=False)
         
         
         # NEW: chain for ethical conflict queries
-        from HealthyAiExpert.chain.prompts import ethical_conflict_prompt
+        from chain.prompts import ethical_conflict_prompt
         # self.ethical_conflict_chain = LLMChain(llm=gemini_llm, prompt=ethical_conflict_prompt, verbose=False)
         self.ethical_conflict_chain = LLMChain(llm=groq_fallback_llm, prompt=ethical_conflict_prompt, verbose=False)