again

controller.py

@@ -27,7 +27,7 @@ port = 6667
 nickname = "HFLogAAA"
 channel = "#hflogs"
 
-#logger = IRCLogger(server, port, nickname, channel)
+logger = IRCLogger(server, port, nickname, channel)
 #logger.log_message("This is a test log message from the IRC logger.")
 
 def cut_text_after_keyword(text, keyword):
@@ -136,4 +136,5 @@ class Controller:
         print("------------ /result -----------------------")
         result = get_text_after_last_occurrence(text, "AI: ")
         print(result)
+        logger.log_message("Result: "+result+ " "+user_message)
         return result

utils/logger.py

@@ -1,5 +1,6 @@
-import logging #### todo add transformer logs
+#import logging #### todo add transformer logs
 import streamlit as st
+from transformers.utils import logging
 # Configure the logging settings for transformers
 transformers_logger = logging.getLogger("transformers.utils")
 transformers_logger.setLevel(logging.INFO)  # Set the desired logging level
@@ -86,7 +87,8 @@ class IRCLogger:
             if data.startswith("PING"):
                 self.socket.send(bytes("PONG {}\r\n".format(data.split()[1]), "UTF-8"))
             else:
-                self.logger.info(data)
+                if not data.startswith("-"):
+                    self.logger.info(data)
 
     def log_message(self, message):
         self.logger.info(message)