28 lines
703 B
Python
28 lines
703 B
Python
# __init__.py
|
|
import os
|
|
import sys
|
|
|
|
from .nodes.tools import HighPassFilter
|
|
from .nodes.face_wrapper import FaceWrapper
|
|
from .nodes.face_fit_and_restore import FaceFitAndRestore
|
|
|
|
# Add current directory to system path
|
|
sys.path.append(os.path.dirname(os.path.realpath(__file__)))
|
|
|
|
# Get the path to the current directory
|
|
NODE_PATH = os.path.dirname(os.path.realpath(__file__))
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"FaceFitAndRestore": FaceFitAndRestore,
|
|
"FaceWrapper": FaceWrapper,
|
|
"HighPassFilter": HighPassFilter
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"FaceFitAndRestore": "Face Fit or Restore",
|
|
"FaceWrapper": "Face Wrapper",
|
|
"HighPassFilter": "High Pass Filter (HPF)"
|
|
}
|
|
|
|
__version__ = "1.0.1"
|