Your commit message

app.py

@@ -12,7 +12,7 @@ from langchain.embeddings import HuggingFaceEmbeddings
 from langchain.vectorstores import FAISS
 from langchain.chains import RetrievalQA
 from langchain_community.llms import HuggingFacePipeline
-
+from langchain.prompts import PromptTemplate
 import tempfile
 import os
 
@@ -50,7 +50,17 @@ def process_pdf(pdf_file):
 #    with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp:
 #        tmp.write(pdf_file.read())
 #        tmp_path = tmp.name
+    custom_prompt = PromptTemplate(
+    input_variables=["context", "question"],
+    template="""
+      Answer the following question based only on the provided context.
+
+      Context:
+      {context}
 
+      Question: {question}
+      Answer:"""
+      )
     loader = PyPDFLoader(pdf_file)
     documents = loader.load()
 
@@ -64,7 +74,8 @@ def process_pdf(pdf_file):
     chain = RetrievalQA.from_chain_type(
         llm=llm,
         retriever=retriever,
-        return_source_documents=False
+        return_source_documents=False,
+        chain_type_kwargs={"prompt": custom_prompt}
     )
 
     # Clean up temp file