exec time

app.py

@@ -8,6 +8,9 @@ from main import chatgpt_prompt, get_chatgpt_resp, generate_kyc_output, gsearch
 # Function to perform the search
 # This is a placeholder function, replace it with your actual search implementation
 def perform_search(pname, keywords, num_results):
+    # record current timestamp
+    start_time = datetime.now()
+    
     # Google search for the person name and get the first 20 query links 
     query = pname + " " + keywords
     search_links = gsearch(query, num_results)
@@ -17,7 +20,7 @@ def perform_search(pname, keywords, num_results):
     #get ChatGPT response 
     resp = get_chatgpt_resp(prompt_text)
     # Create PDF with links and summary 
-    rep_txt= generate_kyc_output(pname, search_links, resp)
+    rep_txt= generate_kyc_output(query, search_links, resp, start_time)
     return (rep_txt)
 
 # Streamlit app

main.py

@@ -39,10 +39,10 @@ def chatgpt_prompt(pname, search_links):
            news (Yes or No)".format(pname, all_links)
     return(prompt_text)
 
-def generate_kyc_output(pname, search_links, chat_response):
+def generate_kyc_output(query, search_links, chat_response, start_time):
     rep_txt = ''
 
-    rep_txt += 'Summary of Google Search for {} \n'.format(pname)
+    rep_txt += 'Summary of Google Search for {} \n'.format(query)
     rep_txt += '\n'
     rep_txt += "Report generated on {} \n".format(datetime.now())
     #rep_txt += "----------------------------------------------------- \n"
@@ -55,6 +55,12 @@ def generate_kyc_output(pname, search_links, chat_response):
     rep_txt+= "\n Summary of searches and adverse news findings \n"
     #rep_txt += "----------------------------------------------------- \n"
     rep_txt += chat_response 
+    rep_txt += '\n'
+    
+    end_time = datetime.now()
+    exec_time = (end_time - start_time).total_seconds() 
+    rep_txt += "Execution runtime {} seconds \n".format(exec_time)
+    rep_txt += '\n'
 
     return(rep_txt)
 
@@ -80,9 +86,9 @@ def main(argv):
     resp = get_chatgpt_resp(prompt_text)
     
     # Create PDF with links and summary 
-    rep_txt= generate_kyc_output(pname, search_links, resp)
+    #rep_txt= generate_kyc_output(pname, search_links, resp)
 
-    print(rep_txt)
+    #print(rep_txt)
 
 
 if __name__ == "__main__":