File size: 2,681 Bytes
171e985 4f52ef2 171e985 89b9dc1 171e985 b409e9d 171e985 b409e9d 171e985 b409e9d 171e985 fa54c46 da41b19 fa54c46 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 |
import logging #### todo add transformer logs
import streamlit as st
# Configure the logging settings for transformers
transformers_logger = logging.getLogger("transformers.file_utils")
transformers_logger.setLevel(logging.INFO) # Set the desired logging level
log_enabled = False
##################################### logger #################################
def log_response(response):
if log_enabled:
with st.chat_message("ai"):
st.markdown("Agent Response\n {}".format(response))
print(response)
# Custom logging handler to append log messages to the chat
class ChatHandler(logging.Handler):
def __init__(self):
super().__init__()
def emit(self, record):
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()
transformers_logger.addHandler(chat_handler)
import socket
import threading
import logging
class IRCLogger:
def __init__(self, server, port, nickname, channel):
self.server = server
self.port = port
self.nickname = nickname
self.channel = channel
self.socket = socket.socket()
self.socket.connect((self.server, self.port))
# Configure logging
self.logger = logging.getLogger(__name__)
self.logger.setLevel(logging.INFO)
# Create a stream handler and set its level to INFO
handler = logging.StreamHandler()
handler.setLevel(logging.INFO)
# Create a formatter and set the format
formatter = logging.Formatter('%(asctime)s - %(message)s')
handler.setFormatter(formatter)
# Add the handler to the logger
self.logger.addHandler(handler)
# Join the IRC channel
self.join_channel()
def send_message(self, message):
self.socket.send(bytes("PRIVMSG {} :{}\r\n".format(self.channel, message), "UTF-8"))
def join_channel(self):
self.socket.send(bytes("NICK {}\r\n".format(self.nickname), "UTF-8"))
self.socket.send(bytes("USER {} 0 * :{}\r\n".format(self.nickname, self.nickname), "UTF-8"))
self.socket.send(bytes("JOIN {}\r\n".format(self.channel), "UTF-8"))
def log_message(self, message):
self.logger.info(message)
self.send_message(message)
#if __name__ == "__main__":
# # Configure logging
# logging.basicConfig(level=logging.INFO)#
#
# server = "irc.efnet.org"
# port = 6667
# nickname = "HFLogger"
# channel = "#hflogs"
|