Update app.py

app.py

@@ -15,9 +15,18 @@ from langchain.embeddings import OpenAIEmbeddings
 from langchain.vectorstores import Chroma
 
 from langchain.chains import ConversationalRetrievalChain
+from langchain.prompts.prompt import PromptTemplate
 
 os.environ['OPENAI_API_KEY'] = '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'
 
+_template = """Assume you are He Yingxu, please complete the following conversations:
+Chat History:
+{chat_history}
+Follow Up Input: {question}
+"""
+
+CUSTOM_QUESTION_PROMPT = PromptTemplate.from_template(_template)
+
 def loading_pdf():
     return "Loading..."
 
@@ -34,6 +43,7 @@ def pdf_changes():
     qa = ConversationalRetrievalChain.from_llm(
         llm=OpenAI(temperature=0.5), 
         retriever=retriever, 
+        condense_question_prompt=CUSTOM_QUESTION_PROMPT,
         return_source_documents=False)
     return "Ready"