Update app.py

app.py

@@ -69,11 +69,7 @@ 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
@@ -124,7 +120,12 @@ async def on_chat_start():
         | RunnablePassthrough.assign(context=itemgetter("context"))
         | chat_prompt | chat_model
     )
-
+    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")
+    
     # Let the user know that the system is ready
     msg.content = measure_embedding_time(docs)
     await msg.update()