Update app.py

app.py

@@ -33,12 +33,14 @@ db = lcvs.Chroma.from_documents(splits, sentence_embeddings)
 retriever = db.as_retriever(search_type="mmr")
 
 template = """
-You are a novel writer. Write response of question based on following context:
+Based of following Context, complete the conversation followed afterwards:
 
 -----------------------------
 Context : {context}
 -----------------------------
 
+Conversation:
+
 {question}
 """
 
@@ -52,8 +54,8 @@ qa_chain = lcchains.RetrievalQA.from_chain_type(
 )
 
 def rag_interface(ques, history):
-    prev_chat = " ".join([f"Question: {q} \nAnswer: {a} \n" for q, a in history])
-    current_ques = f'\nQuestion: {ques} \nAnswer:'
+    prev_chat = " ".join([f"Human: {q} \nAI: {a} \n" for q, a in history])
+    current_ques = f'\nHuman: {ques} \nAI:'
 
     query = prev_chat + current_ques