import sys
from loguru import logger
from ui.core.state import app_state

class GuiLogger:
    def write(self, message):
        if app_state.log_element:
            app_state.log_element.push(message.strip())

def setup_logger():
    logger.remove()
    # 控制台输出
    logger.add(sys.stderr, format="{time:HH:mm:ss} | {level} | {message}")
    # GUI 输出
    logger.add(GuiLogger(), format="{time:HH:mm:ss} | {level} | {message}", level="INFO")