Anush
47050a7f33
ComfyUI custom nodes for Google Gemini image generation and editing using the Gemini web interface. Features: - Text-to-Image generation - Image-to-Image editing - Vision Chat - Auto Authentication with browser cookies - Self-contained with bundled dependencies
38 lines
958 B
Python
38 lines
958 B
Python
import sys
|
|
from loguru import logger as _logger
|
|
|
|
_handler_id = None
|
|
|
|
|
|
def set_log_level(level: str | int) -> None:
|
|
"""
|
|
Set the log level for gemini_webapi. The default log level is "INFO".
|
|
|
|
Note: calling this function for the first time will globally remove all existing loguru
|
|
handlers. To avoid this, you may want to set logging behaviors directly with loguru.
|
|
|
|
Parameters
|
|
----------
|
|
level : `str | int`
|
|
Log level: "TRACE", "DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"
|
|
|
|
Examples
|
|
--------
|
|
>>> from gemini_webapi import set_log_level
|
|
>>> set_log_level("DEBUG") # Show debug messages
|
|
>>> set_log_level("ERROR") # Only show errors
|
|
"""
|
|
|
|
global _handler_id
|
|
|
|
_logger.remove(_handler_id)
|
|
|
|
_handler_id = _logger.add(
|
|
sys.stderr,
|
|
level=level,
|
|
filter=lambda record: record["extra"].get("name") == "gemini_webapi",
|
|
)
|
|
|
|
|
|
logger = _logger.bind(name="gemini_webapi")
|