added code to check time cost

main.py

@@ -12,6 +12,7 @@ from llama_index.embeddings import HuggingFaceEmbedding
 from trulens_eval import Tru
 
 from utils import get_prebuilt_trulens_recorder
+import time
 
 openai.api_key = utils.get_openai_api_key()
 
@@ -19,6 +20,7 @@ def main():
     
     if not os.path.exists("./default.sqlite"):
 
+        start_time = time.time()
         file_ls_str = ", ".join(os.listdir("./raw_documents"))
         print(f"File list: {file_ls_str}")
         print("")
@@ -37,6 +39,7 @@ def main():
         embed_model = HuggingFaceEmbedding(model_name="BAAI/bge-small-en-v1.5")
 
         print("Creating vector store ..")
+        print("time spent:", time.time() - start_time)
         service_context = ServiceContext.from_defaults(llm=llm, embed_model=embed_model)
         index = VectorStoreIndex.from_documents([document], service_context=service_context)
 
@@ -72,6 +75,12 @@ def main():
         records.to_csv("./results/records.csv", index=False)
 
         print(tru.db.engine.url.render_as_string(hide_password=False))
+        
+        end_time = time.time()
+        time_spent_mins = (end_time - start_time) / 60
+        with open("./results/time_cost.txt", "w") as fp:
+            fp.write(f"Takes {int(time_spent_mins)} mins to create llm evaluation.")
+
         # tru.run_dashboard()
 
 if __name__ == "__main__":

pages/3_app.py

@@ -4,8 +4,13 @@ import os
 try:
     raw_docs_files = ", ".join(os.listdir("./raw_documents"))
     curr_directory_files = ", ".join(os.listdir("."))
-    file_ls_str = raw_docs_files + "\n\n" + curr_directory_files
+
+    with open("./results/time_cost.txt", "r") as fp:
+        time_cost_str = fp.read()
+
+    system_update = raw_docs_files + "\n\n" + curr_directory_files + "\n\n" + time_cost_str
+
 except:
-    file_ls_str = "NA"
+    system_update = "NA"
 
-st.write(f"Hello World! File list: {file_ls_str}")
+st.write(f"Hello World! File list: {system_update}")