Update app.py

app.py

@@ -35,11 +35,11 @@ def load_pdf_and_generate_embeddings(pdf_doc, open_ai_key):
         Return the key fields from the question followed by : and the answer :"""
         PROMPT = PromptTemplate(template=prompt_template, input_variables=["context", "question"])
         chain_type_kwargs = {"prompt": PROMPT}
-        pdf_qa = RetrievalQA.from_chain_type(llm=ChatOpenAI(temperature=0, model_name="gpt-4"),chain_type="stuff", retriever=vectordb.as_retriever(), chain_type_kwargs=chain_type_kwargs, return_source_documents=False)
+        pdf_qa = RetrievalQA.from_chain_type(llm=ChatOpenAI(temperature=0, model_name="gpt-4"),chain_type="stuff", retriever=vectordb.as_retriever(search_kwargs={"k": 5}), chain_type_kwargs=chain_type_kwargs, return_source_documents=False)
                 
         return "Ready"
     else:
-        return "Please provide an OpenAI API key"
+        return "Please provide an OpenAI gpt-4 API key"
         
 
 def answer_predefined_questions(document_type):
@@ -63,11 +63,11 @@ def answer_predefined_questions(document_type):
         field1 = "Borrower"
         query2 = "what is the Property Address?"
         field2 = "Property Address"
-        query3 = "what is the loan term?"
+        query3 = "what is the Loan Term?"
         field3 = "Loan Term"
-        query4 = "What is the base income?"
+        query4 = "What is the Base Income?"
         field4 = "Base Income"
-        query5 = "what is the borrower's SSN?"
+        query5 = "what is the Borrower's SSN?"
         field5 = "Borrower SSN"
 
     else:
@@ -99,8 +99,8 @@ css="""
 title = """
 <div style="text-align: center;max-width: 700px;">
     <h1>Chatbot for PDFs - GPT-4</h1>
-    <p style="text-align: center;">Upload a .PDF, click the "Load PDF" button, <br />
-    Wait for the Status to show Ready, start typing your questions. <br />
+    <p style="text-align: center;">Upload a .PDF, click the "Upload PDF and generate embeddings" button, <br />
+    Wait for the Status to show Ready. You can chose to get answers to the pre-defined question set OR ask your own question <br />
     The app is built on GPT-4 and leverages PromptTemplate</p>
 </div>
 """