Update app.py

app.py

@@ -63,18 +63,34 @@ def answer_predefined_questions(document_type):
         #Create a list of questions around the relevant fields of a Deed of Trust(DOT) document
         query1 = "what is the Loan Number?"
         field1 = "Loan Number"
-        query2 = "who is the Borrower?"
+        query2 = "Who is the Borrower?"
         field2 = "Borrower"
         query3 = "what is the Case Number?"
         field3 = "Case Number"
         query4 = "what is the Mortgage Identification number?"
         field4 = "MIN Number"
         query5 = "DOT signed date?"
-        field5 = "Signed Date"        
+        field5 = "Signed Date"
+        query6 = "Who is the Lender?"
+        field6 = "Lender"
+        query7 = "what is the VA/FHA Number?"
+        field7 = "VA/FHA Number"
+        query8 = "Who is the Co-Borrower?"
+        field8 = "Co-Borrower"
+        query9 = "What is the property type - single family, multi family?"
+        field9 = "Property Type"
+        query10 = "what is the Property Address?"
+        field10 = "Property Address"
+        query11 = "In what County is the property located?"
+        field11 = "Property County"
+        query12 = "what is the Electronically recorded date"
+        field12 = "Electronic Recording Date"
+
+    
         
     elif document_type == "Transmittal Summary":
         #Create a list of questions around the relevant fields of a TRANSMITTAL SUMMARY document
-        query1 = "who is the Borrower?"
+        query1 = "Who is the Borrower?"
         field1 = "Borrower"
         query2 = "what is the Property Address?"
         field2 = "Property Address"
@@ -83,13 +99,29 @@ def answer_predefined_questions(document_type):
         query4 = "What is the Base Income?"
         field4 = "Base Income"
         query5 = "what is the Borrower's SSN?"
-        field5 = "Borrower SSN"
+        field5 = "Borrower's SSN"
+        query6 = "Who is the Co-Borrower?"
+        field6 = "Co-Borrower"
+        query7 = "What is the Original Loan Amount?"
+        field7 = "Original Loan Amount"
+        query8 = "What is the Initial P&I payment?"
+        field8 = "Initial P&I payment"
+        query9 = "What is the Co-Borrower's SSN?"
+        field9 = "Co-Borrower’s SSN"
+        query10 = "Number of units?"
+        field10 = "Units#"
+        query11 = "Who is the Seller?"
+        field11 = "Seller"
+        query12 = "Document signed date?"
+        field12 = "Signed Date"
+
+
 
     else:
         return "Please choose your Document Type"
 
-    queryList = [query1, query2, query3, query4, query5]
-    fieldList = [field1, field2, field3, field4, field5]
+    queryList = [query1, query2, query3, query4, query5, query6, query7, query8, query9, query10, query11,query12]
+    fieldList = [field1, field2, field3, field4, field5, field6, field7, field8, field9, field10, field11,field12]
     responseList =[]
 
     i = 0
@@ -98,7 +130,9 @@ def answer_predefined_questions(document_type):
         responseList.append(pdf_qa.run(question))
         i = i+1
 
-    return pd.DataFrame({"Field": [fieldList[0], fieldList[1], fieldList[2], fieldList[3], fieldList[4]], "Question to gpt-4": [queryList[0], queryList[1], queryList[2], queryList[3], queryList[4]], "Response from gpt-4": [responseList[0],responseList[1],responseList[2],responseList[3],responseList[4]]})
+    return pd.DataFrame({"Field": [fieldList[0],fieldList[1],fieldList[2],fieldList[3],fieldList[4],fieldList[5],fieldList[6],fieldList[7],fieldList[8],fieldList[9],fieldList[10],fieldList[11]], 
+                         "Question to gpt-4": [queryList[0],queryList[1],queryList[2],queryList[3],queryList[4],queryList[5],queryList[6],queryList[7],queryList[8],queryList[9],queryList[10],queryList[11]], 
+                         "Response from gpt-4": [responseList[0],responseList[1],responseList[2],responseList[3],responseList[4],responseList[5],responseList[6],responseList[7],responseList[8],responseList[9],responseList[10],responseList[11]]})