Update app.py

app.py

@@ -19,6 +19,7 @@ from operator import itemgetter
 from langchain_core.runnables.passthrough import RunnablePassthrough
 import chainlit as cl
 from langchain_core.runnables.config import RunnableConfig
+import time
 
 load_dotenv()
 
@@ -68,6 +69,11 @@ def process_file(file: AskFileResponse):
         doc.metadata["source"] = f"source_{i}"
     return docs
 
+def measure_embedding_time(docs):
+    start_time = time.time()
+    vectorstore.add_documents(docs)
+    elapsed_time = time.time() - start_time
+    print(f"{str(docs)} took {elapsed_time:.2f} seconds")
 
 
 @cl.on_chat_start
@@ -76,8 +82,6 @@ async def on_chat_start():
     files = None
 
     while files == None:
-        # Async method: This allows the function to pause execution while waiting for the user to upload a file,
-        # without blocking the entire application. It improves responsiveness and scalability.
         files = await cl.AskFileMessage(
             content="Please upload a PDF file to begin!",
             accept=["application/pdf"],
@@ -122,7 +126,7 @@ async def on_chat_start():
     )
 
     # Let the user know that the system is ready
-    msg.content = f"Processing `{file.name}` done. You can now ask questions!"
+    msg.content = measure_embedding_time(docs)
     await msg.update()
 
     cl.user_session.set("chain", retrieval_augmented_qa_chain)