Adding Time to logs

app.py

@@ -1,3 +1,4 @@
+import logging
 import chainlit as cl
 import os
 from datasets import load_dataset
@@ -14,7 +15,7 @@ from langchain_core.prompts import ChatPromptTemplate
 from langchain_openai import ChatOpenAI
 import asyncio
 
-print("""
+logging.info("""
 =================================================================================
  STARTING
 =================================================================================
@@ -40,17 +41,17 @@ cached_embedder = CacheBackedEmbeddings.from_bytes_store(embedding_model, store,
 index_path = "faiss_index"
 if os.path.exists(index_path):
     vector_store = FAISS.load_local(index_path, cached_embedder, allow_dangerous_deserialization=True)
-    print("Vector store loaded from saved index.")
+    logging.info("Vector store loaded from saved index.")
 else:
     vector_store = FAISS.from_documents(chunked_documents, cached_embedder)
-    print("Vector store created from documents.")
+    logging.info("Vector store created from documents.")
     vector_store.save_local(index_path)
-    print("Vector store saved locally.")
+    logging.info("Vector store saved locally.")
 
 
 @cl.on_chat_start
 async def on_chat_start():
-    print("""
+    logging.info("""
     =================================================================================
      ON START CHAT
     =================================================================================
@@ -72,7 +73,7 @@ async def on_chat_start():
 
 @cl.on_message
 async def on_message(message: cl.Message):
-    print(f"""
+    logging.info(f"""
     =================================================================================
      ON MESSAGE: {message.content}
     =================================================================================
@@ -85,9 +86,9 @@ async def on_message(message: cl.Message):
             message.content,
             config=RunnableConfig(callbacks=[cl.LangchainCallbackHandler()]),
     ):
-        print(f"Received chunk  <{chunk}>")
+        logging.info(f"Received chunk  <{chunk}>")
         await msg.stream_token(chunk)
 
-    print(f"Sending message")
+    logging.info(f"Sending message")
     await msg.send()
-    print(f"Done with <{message.content}>")
+    logging.info(f"Done with <{message.content}>")