import logging
from flask import Flask
from werkzeug.middleware.proxy_fix import ProxyFix

logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)
app = Flask(__name__)
app.wsgi_app = ProxyFix(app.wsgi_app)

# 创建自定义的日志过滤器
class StaticFilter(logging.Filter):
    def filter(self, record):
        # 如果是静态资源的请求并且状态码是304，则不记录
        return not (
            ('/static/' in record.getMessage() and '304' in record.getMessage()) or
            ('.css' in record.getMessage() and '304' in record.getMessage()) or
            ('.js' in record.getMessage() and '304' in record.getMessage())
        )

# 配置werkzeug的日志
logging.getLogger('werkzeug').addFilter(StaticFilter())

# 设置基本的日志格式
logging.basicConfig(
    level=logging.INFO,
    format='%(name)s:%(message)s'
)

def log_prompt(messages, preset_name):
    logger.info(f"Preset: {preset_name}")
    for message in messages:
        logger.info(message)