Update prompts.py

prompts.py

@@ -2,6 +2,20 @@
 
 from langchain.prompts import PromptTemplate
 
+contextualize_prompt_str = """
+You are a helpful assistant for DailyWellnessAI. Based on the current question and the previous conversation, reformat or rephrase the question to make it more suitable for processing by the classification or RAG chains.
+
+Here's the current query: 
+{input}
+
+Conversation history:
+{chat_history}
+
+Please rephrase the question or format it as needed. Keep it relevant to the context and make it suitable for classification or further processing.
+"""
+
+
+
 classification_prompt_str = """
 You are a helpful assistant that classifies user questions into three categories:
 1) "Wellness" if the question involves health, nutrition, fitness, mental well-being, self-care, or research related to these.
@@ -52,6 +66,13 @@ Write a short, polite refusal that gently explains we only handle daily wellness
 """
 
 # Modify prompt templates to work with chat history if needed
+
+contextualize_prompt = PromptTemplate(
+    template=contextualize_prompt_str,
+    input_variables=["input", "chat_history"]
+)
+
+
 classification_prompt = PromptTemplate(
     template=classification_prompt_str,
     input_variables=["query"]