Update app.py

app.py

@@ -42,8 +42,7 @@ def load_pdf_and_generate_embeddings(pdf_doc, open_ai_key):
         
 
 def answer_predefined_questions(document_type):
-    queryList=[]
-    fieldList=[]
+    
     if document_type == "Deed of Trust":
         #Create a list of questions around the relevant fields of a Deed of Trust(DOT) document
         query0 = "what is the Lender's Name?"
@@ -70,12 +69,9 @@ def answer_predefined_questions(document_type):
         field10 = "Property County"
         query11 = "what is the Electronically recorded date"
         field11 = "Electronic Recording Date"
-        queryListForDOT = [query0, query1, query2, query3, query4, query5, query6, query7, query8, query9, query10, query11]
-        fieldListForDOT = [field0, field1, field2, field3, field4, field5, field6, field7, field8, field9, field10, field11]
-        for query in queryListForDOT:
-            queryList.append(query)
-        for field in fieldListForDOT:
-            fieldList.append(field)
+        queryList = [query0, query1, query2, query3, query4, query5, query6, query7, query8, query9, query10, query11]
+        fieldList= [field0, field1, field2, field3, field4, field5, field6, field7, field8, field9, field10, field11]
+        
         
     elif document_type == "Transmittal Summary":
         #Create a list of questions around the relevant fields of a TRANSMITTAL SUMMARY document
@@ -103,16 +99,14 @@ def answer_predefined_questions(document_type):
         fieldA10 = "Seller"
         queryA11 = "Document signed date?"
         fieldA11 = "Singed Date"
-        queryListForTransmittalSummary = [queryA0, queryA1, queryA2, queryA3, queryA4, queryA5, queryA6, queryA7, queryA8, queryA9, queryA10, queryA11]
-        fieldListForTransmittalSummary = [fieldA0, fieldA1, fieldA2, fieldA3, fieldA4, fieldA5, fieldA6, fieldA7, fieldA8, fieldA9, fieldA10, fieldA11]
-        for query in queryListForTransmittalSummary:
-            queryList.append(query)
-        for field in fieldListForTransmittalSummary:
-            fieldList.append(field)
+        queryList = [queryA0, queryA1, queryA2, queryA3, queryA4, queryA5, queryA6, queryA7, queryA8, queryA9, queryA10, queryA11]
+        fieldList = [fieldA0, fieldA1, fieldA2, fieldA3, fieldA4, fieldA5, fieldA6, fieldA7, fieldA8, fieldA9, fieldA10, fieldA11]
+
+        
 
     i = 0
     while i < len(queryList):
-        question = questionList[i]
+        question = queryList[i]
         field = fieldList[i]
         response += "Field Name:", field, "; Question sent to gpt-4: ", question, "; Response from gpt-4:",pdf_qa.run(question)
     return response