Define the prompt template for both general conversation and weather retrieval

workflow.py

@@ -11,12 +11,28 @@ def get_session_history(memory: BaseMemory):
 def get_workflow():
     """Set up the chatbot workflow with memory and prompt template."""
 
-    # Define the prompt template for conversation
+    # Define the prompt template for both general conversation and weather retrieval
     prompt_template = PromptTemplate(
-        input_variables=["input"],
-        template="You are a helpful assistant. The user just said: {input}. What would be a good response?"
-    )
+        input_variables=["input", "previous_conversation"],
+        template="""
+You are a helpful assistant. You should answer the user's question or have a normal conversation. If the user asks about the weather, 
+please respond with the current weather information based on their input location. Otherwise, answer to the best of your ability.
+
+If the user's input is about the weather, you should respond with details about the weather.
+For example:
+- "What is the weather in Paris?"
+- "How's the weather in New York?"
+
+Example conversation flow:
+User: What's the weather like today in London?
+Assistant: Let me check the weather for you. The current weather in London is [weather details].
+
+If the input is not weather-related, just respond with a conversational response.
 
+The user said: {input}
+Previous conversation: {previous_conversation}
+""")
+    
     # Create memory for conversation
     memory = ConversationBufferMemory(memory_key="input", return_messages=True)