Update app.py

app.py

@@ -28,7 +28,7 @@ However, if the user does not require legal assistance in Pakistan, you will imm
 say goodbye, ending the conversation. Remember to base your responses on the user's needs, providing accurate and
 concise information regarding the Pakistan legal law and rights where applicable. Your interactions should be professional and
 focused, ensuring the user's queries are addressed efficiently without deviating from the set flows.
-
+CONTEXT: {context}
 CHAT HISTORY: {chat_history}
 QUESTION: {question}
 ANSWER:
@@ -84,14 +84,13 @@ if "messages" not in st.session_state:
     st.session_state.messages = []
 
 if "memory" not in st.session_state:
-    st.session_state.memory = ConversationBufferWindowMemory(k=2, memory_key="chat_history",return_messages=True) 
+    st.session_state.memory = ConversationBufferWindowMemory(k=5, memory_key="chat_history",return_messages=True,output_key='answer') 
 
 #embeddings = HuggingFaceEmbeddings(model_name="nomic-ai/nomic-embed-text-v1",model_kwargs={"trust_remote_code":True,"revision":"289f532e14dbbbd5a04753fa58739e9ba766f3c7"})
 #db=FAISS.load_local("/content/ipc_vector_db", embeddings, allow_dangerous_deserialization=True)
 
 
-prompt = PromptTemplate(template=custom_template,
-                        input_variables=[ 'question', 'chat_history'])
+prompt = PromptTemplate(template=custom_template)
 
 # You can also use other LLMs options from https://python.langchain.com/docs/integrations/llms. Here I have used TogetherAI API