Update app.py

app.py

@@ -28,9 +28,22 @@ class AdvancedPdfChatbot:
         os.environ["OPENAI_API_KEY"] = openai_api_key
         self.embeddings = OpenAIEmbeddings()
         self.text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=200)
-        self.llm =  OpenAIChat(temperature=0,model_name='gpt-3.5-turbo',prompt = PromptTemplate(template=template)
+        self.llm =  OpenAIChat(temperature=0,model_name='gpt-3.5-turbo')
+        
         self.memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
         self.qa_chain = None
+        self.template = """
+        You are a study partner assistant, students give you pdfs
+        and you help them to answer their questions.
+        
+        Answer the question based on the most recent provided resources only.
+        Give the most relevant answer.
+        
+        Context: {context}
+        Question: {question}
+        Answer:
+        """
+        self.prompt = PromptTemplate(template=self.template, input_variables=["context", "question"])
 
     def load_and_process_pdf(self, pdf_path):
         loader = PyPDFLoader(pdf_path)
@@ -44,6 +57,7 @@ class AdvancedPdfChatbot:
             self.llm,
             retriever=self.db.as_retriever(),
             memory=self.memory
+            combine_docs_chain_kwargs={"prompt": self.prompt}
         )
 
     def chat(self, query):