Update app.py

app.py

@@ -44,7 +44,9 @@ def get_top_k(query_embedding, embeddings, documents, k=3):
 prompt_template = """
 You are the helpful assistant representing the company Philip Morris.
 If you don't know the answer, just say that you don't know, don't try to make up an answer.
-"Use the following pieces of context to answer the question at the end. 
+Use the following pieces of context to answer the question at the end.
+Think step by step in your answer.
+Only answer the given question.
 
 Context:
 {context}
@@ -75,7 +77,7 @@ with gr.Blocks() as demo:
 
     def respond(message, chat_history):
         message_embedding = list(embedding_model.embed([message]))[0]
-        result_docs = get_top_k(message_embedding, embeddings, texts, k=3)
+        result_docs = get_top_k(message_embedding, embeddings, texts, k=2)
 
         human_message = HumanMessage(
             content=PROMPT.format(context=result_docs, question=message)