final 4

app.py

@@ -283,6 +283,7 @@ def read_file_get_prompts(file_name):
         for q in range(len(text_per_page[page])):
             #print(f"{''.join(map(str, text_per_page[page][q]))}")
             result = result + f"{''.join(map(str, text_per_page[page][q]))}"
+
     return result
 
 def create_dataframe_from_text(text):
@@ -368,19 +369,28 @@ if st.button('Get Loan Details',type="primary"):
     with st.spinner("🤖 Operation in progress. Please wait! 🤖 "):
         result = read_file_get_prompts(file_name)
         
-        st.write(result.lower())
+        #st.write(result.lower())
         response_1 = OpenAI().complete(prompt_template_1.format(loan_data=result.lower()))
         st.table(create_dataframe_from_text(response_1.text))
 
         st.balloons()
 
-if st.button('Get Loan Transactions', type="secondary"):
+async def get_completion(prompt_template, response, data):
+    # Other code...
+    # Wait for completion of OpenAI().complete()
+    completion_result = await OpenAI().complete(prompt_template.format(response = response, loan_data=data.lower()))
+    return completion_result
+
+
+if st.button('Get Loan Transactions', type="primary"):
     with st.spinner("🤖 Operation in progress. Please wait! 🤖 "):
         result = read_file_get_prompts(file_name)
         
-        st.write(result.lower())
-        response_1 = OpenAI().complete(prompt_template_1.format(loan_data=result.lower()))
-        response_2 = OpenAI().complete(prompt_template_2.format(response_1=response_1.text, loan_data=result.lower()))
+        #st.write(result.lower())
+        response_1 = get_completion(prompt_template_1, "",  result)
+        
+        #response_2 = OpenAI().complete(prompt_template_2.format(response_1=response_1.text, loan_data=result.lower()))
+        response_2 = get_completion(prompt_template_2, response_1.text,  result)
         #st.write(response_2)
         st.table(create_dataframe_from_text_2(response_2.text))