Update app.py

app.py

@@ -46,73 +46,51 @@ 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"
-        
-        queryList = [query0, query1]
-        fieldList= [field0, field1]
-
-
-        
+        query2 = "who is the Borrower?"
+        field2 = "Borrower"
+        query3 = "who is the Co-Borrower?"
+        field3 = "Co-Borrower"
+        query4 = "what is the Mortgage Identification number?"
+        field4 = "MIN Number"
+        query5 = "On what date was the Deed of Trust signed?"
+        field5 = "Signed Date"        
         
     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]
-        fieldList = [fieldA0, fieldA1]
+        query1 = "who is the Borrower?"
+        field1 = "Borrower"
+        query2 = "what is the Property Address?"
+        field2 = "Property Address"
+        query3 = "what is the loan term?"
+        field3 = "Loan Term"
+        query4 = "What is the base income?"
+        field4 = "Base Income"
+        query5 = "what is the borrower's SSN?"
+        field5 = "Borrower SSN"
 
     else:
         return "Please choose your Document Type"
 
-        
-    response=""
+    queryList = [query1, query2, query3, query4, query5]
+    fieldList = [field1, field2, field3, field4, field5]
+    responseList =[]
+
     i = 0
     while i < len(queryList):
         question = queryList[i]
-        field = fieldList[i]
-        
-        fieldInfo = "Field Name:"+ field
-        response += fieldInfo
-        questionInfo = "; Question sent to gpt-4: "+ question
-        response += questionInfo
-        answer = pdf_qa.run(question)
-        gptResponse = "; Response from gpt-4:"+ answer
-        response += gptResponse
-        
-    return response
+        responseList[i] = 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]]})
         
 
     
 def answer_query(query):
     question = query
-    response = "Field Name: Location; Question sent to gpt-4: ", question, "Response from gpt-4:",pdf_qa.run(question)
-    #return response
-    return pd.DataFrame({"Field": ['Location'], "Question": ['what is the location'], "Answer": ['Coppell,TX']})
+    return pdf_qa.run(question)
+    
 
 css="""
 #col-container {max-width: 700px; margin-left: auto; margin-right: auto;}
@@ -137,17 +115,17 @@ 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").style(full_width=False)
+            load_pdf = gr.Button(label="Load PDF").style(full_width=False)
             
         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 = gr.Dataframe(label="Answers to Predefined Question set")
             answers_for_predefined_question_set = gr.Button("Get Answers to Pre-defined Question set").style(full_width=False)
               
         with gr.Row():
             input = gr.Textbox(label="Type in your question")
-            output = gr.Dataframe(label="Answer")
-            submit_query = gr.Button("Submit your own question").style(full_width=False)
+            output = gr.Textbox(label="Answer")
+            submit_query = gr.Button(label="Submit your own question").style(full_width=False)
         
  
     load_pdf.click(load_pdf_and_generate_embeddings, inputs=[pdf_doc, openai_key], outputs=status)