Update app.py

app.py

@@ -27,6 +27,40 @@ memory = ConversationBufferWindowMemory(
 )
 
 
+
+
+from langchain import PromptTemplate
+query_text = "when did alice see mad hatter"
+
+results = db.similarity_search_with_relevance_scores(query_text, k=3)
+if len(results) == 0 or results[0][1] < 0.5:
+    print(f"Unable to find matching results.")
+
+
+context_text = "\n\n---\n\n".join([doc.page_content for doc, _score in results ])
+
+template = """
+The following is a conversation between a human an AI. Answer  question based only on the conversation.
+
+Current conversation:
+{history}
+
+"""
+
+
+
+s="""
+
+\n question: {input}
+
+\n answer:""".strip()
+
+
+prompt = PromptTemplate(input_variables=["history", "input"], template=template+context_text+'\n'+s)
+
+
+
+
 chain = ConversationChain(
     llm=llm,