Update app.py

app.py

@@ -41,7 +41,73 @@ def load_pdf_and_generate_embeddings(pdf_doc, open_ai_key):
         return "Please provide an OpenAI API key"
         
 
- 
+def answer_predefined_questions(document_type):
+    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?"
+        field0 = "Lender"
+        query1 = "what is the Loan Number?"
+        field1 = "Loan Number"
+        query2 = "what is the Case Number?"
+        field2 = "Case Number"
+        query3 = "what is the VA/FHA Number?"
+        field3 = "VA/FHA Number"
+        query4 = "who is the Borrower?"
+        field4 = "Borrower"
+        query5 = "who is the Co-Borrower?"
+        field5 = "Co-Borrower"
+        query6 = "what is the Mortgage Identification number?"
+        field6 = "MIN Number"
+        query7 = "what is the property type - single fmaily, multi family?"
+        field7 = "Property Type"
+        query8 = "what is the Property Address?"
+        field8 = "Property Address"
+        query9 = "On what date was the Deed of Trust signed?"
+        field9 = "Signed Date"
+        query10 = "What is the Property County?"
+        field10 = "Property County"
+        query11 = "what is the Electronically recorded date"
+        field11 = "Electronic Recording Date"
+        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
+        queryA0 = "who is the Borrower?"
+        fieldA0 = "Borrower"
+        queryA1 = "what is the Property Address?"
+        fieldA1 = "Property Address"
+        queryA2 = "who is the Co-Borrower?"
+        fieldA2 = "Co-Borrower"
+        queryA3 = "what is the loan term?"
+        fieldA3 = "Loan Term"
+        queryA4 = "What is the base income?"
+        fieldA4 = "Base Income"
+        queryA5 = "what is the original loan amount?"
+        fieldA5 = "Original Loan Amount"
+        queryA6 = "what is the Initial P&I Payment?"
+        fieldA6 = "Initial P&I Payment"
+        queryA7 = "what is the borrower's SSN?"
+        fieldA7 = "Borrower SSN"
+        queryA8 = "what is the co-borrower's SSN?"
+        fieldA8 = "C0-Borrower SSN"
+        queryA9 = "Number of units?"
+        fieldA9 = "Number of units"
+        queryA10 = "who is the seller?"
+        fieldA10 = "Seller"
+        queryA11 = "Document signed date?"
+        fieldA11 = "Singed Date"
+        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]
+        field = fieldList[i]
+        response += "Field Name:", field "; Question sent to gpt-4: ", question, "; Response from gpt-4:",pdf_qa.run(question)
+    return response
+        
+
     
 def answer_query(query):
     question = query
@@ -72,17 +138,21 @@ with gr.Blocks(css=css,theme=gr.themes.Monochrome()) as demo:
         with gr.Row():
             status = gr.Textbox(label="Status", placeholder="", interactive=False)
             load_pdf = gr.Button("Load PDF")
-        
-            
             
+        with gr.Row():
+            document_type = gr.Radio(['Deed of Trust', 'Transmittal Summary'], label="Select the Document Type")
+            answers = gr.Textbox(label="Answers to Predefined Question set")
+            answers_for_predefined_question_set = gr.Button("Get Answers to Pre-defined Question set")
+              
         with gr.Row():
             input = gr.Textbox(label="Type in your question")
             output = gr.Textbox(label="Answer")
-        submit_query = gr.Button("Submit")
+            submit_query = gr.Button("Submit your question")
         
  
     load_pdf.click(load_pdf_and_generate_embeddings, inputs=[pdf_doc, openai_key], outputs=status)
-    
+
+    answers_for_predefined_question_set.click(answer_predefined_questions, document_type, answers)
         
     submit_query.click(answer_query,input,output)