Update logger.py

logger.py

@@ -7,7 +7,9 @@ transformers_logger.setLevel(logging.INFO)  # Set the desired logging level
 log_enabled = False
 
 
+##################################### logger #################################
 
+ 
 
 
 def log_response(response):
@@ -26,6 +28,8 @@ class ChatHandler(logging.Handler):
         log_message = self.format(record)
         #with st.chat_message("ai"):
         st.markdown(f"Log: {log_message}")
+        with st.chat_message("ai"):
+            st.markdown("Agent Response\n {}".format(record))
 
 # Add the custom handler to the transformers_logger
 chat_handler = ChatHandler()