38 lines
1.9 KiB
Python
38 lines
1.9 KiB
Python
from .polymath import Polymath, MediaScraper
|
|
from .helper import SaveAbs
|
|
from .concept_eraser import ConceptEraserNode
|
|
from .textmask import TextMaskGeneratorNode
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"polymath_chat": Polymath,
|
|
"polymath_scraper": MediaScraper,
|
|
"polymath_helper": SaveAbs,
|
|
"polymath_concept_eraser": ConceptEraserNode,
|
|
"polymath_text_mask": TextMaskGeneratorNode
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"polymath_chat": "LLM Polymath Chat with Advanced Web and Link Search",
|
|
"polymath_scraper": "LLM Polymath Scraper for various sites",
|
|
"polymath_helper": "Save Image to Absolute Path",
|
|
"polymath_concept_eraser": "Erase Concept from Model",
|
|
"polymath_text_mask": "Generate a mask from text"
|
|
}
|
|
|
|
ascii_art = """
|
|
Polymath is brought to you by
|
|
|
|
_ _ _ _ _ _
|
|
(▒)(▒) _(▒)(▒)(▒)(▒)_
|
|
(▒) _ _ _ _ _ _ (▒) (▒) _ _ _ _ _ _ _
|
|
(▒) (▒) (▒) (▒)(▒)_(▒)(▒) _ _(▒) _ (▒)(▒)(▒) _ (▒)(▒)(▒)(▒)_
|
|
(▒) (▒) (▒) (▒) (▒) (▒) (▒)(▒)_ (▒) (▒)(▒) (▒)
|
|
(▒) (▒) (▒) (▒) (▒) (▒) _ (▒)(▒) (▒)(▒) (▒)
|
|
_ (▒) _(▒)_ _ _(▒)_ (▒) (▒) (▒)(▒)_ _ _ _(▒)(▒) _ _ _ (▒)(▒) (▒)
|
|
(▒)(▒)(▒) (▒)(▒)(▒) (▒)(▒) (▒) (▒) (▒)(▒)(▒)(▒) (▒)(▒)(▒) (▒) (▒)
|
|
"""
|
|
print(f"\033[92m{ascii_art}\033[0m")
|
|
|
|
WEB_DIRECTORY = "./web"
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
|