__init__.py

import os
import sys

# Add parent directory to path
current_dir = os.path.dirname(os.path.abspath(__file__))
parent_dir = os.path.dirname(current_dir)
if parent_dir not in sys.path:
    sys.path.insert(0, parent_dir)

# Now import folder_paths
import folder_paths

# Then import your other modules
from .IFLLMNode import IFLLM
from .IFDisplayTextWildcardNode import IFDisplayTextWildcard
from .IFLLMSaveTextNode import IFSaveText
from .IFLLMDisplayTextNode import IFDisplayText
from .IFLLMDisplayOmniNode import IFDisplayOmni
from .IFLLMTextTyperNode import IFTextTyper
from .IFLLMJoinTextNode import IFJoinText
from .IFLLMLoadImagesNodeS import IFLoadImagess
from .ListModelsNode import ListModelsNode
from .send_request import *


'''# Unified omost import handling
try:
    if "omost" not in sys.modules:  # Check if already imported
        try:
            from .omost import omost_function  # Try relative import first
            print("Successfully imported omost_function from current directory")
        except ImportError:
            # If relative import fails, try from parent directory
            parent_dir = os.path.dirname(current_dir)
            parent_dir_name = os.path.basename(parent_dir)
            if parent_dir_name == 'ComfyUI-IF_LLM':
                sys.path.insert(0, parent_dir)
                from omost import omost_function
                print(f"Successfully imported omost_function from {parent_dir}/omost.py")
    else:
        omost_function = sys.modules["omost"].omost_function
        print("Using already imported omost_function")
except ImportError as e:
    print(f"Error importing omost_function: {e}")
    print(f"Current sys.path: {sys.path}")
    raise'''

class OmniType(str):
    """A special string type that acts as a wildcard for universal input/output. 
       It always evaluates as equal in comparisons."""
    def __ne__(self, __value: object) -> bool:
        return False
    
OMNI = OmniType("*")
                       
NODE_CLASS_MAPPINGS = {
    "IF_LLM": IFLLM,
    "IF_LLM_SaveText": IFSaveText,
    "IF_LLM_DisplayText": IFDisplayText,
    "IF_LLM_DisplayTextWildcard": IFDisplayTextWildcard,
    "IF_LLM_DisplayOmni": IFDisplayOmni,
    "IF_LLM_TextTyper": IFTextTyper,
    "IF_LLM_JoinText": IFJoinText,
    "IF_LLM_LoadImagesS": IFLoadImagess,
    "IF_LLM_ListModels": ListModelsNode,
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "IF_LLM": "IF LLM🎨",
    "IF_LLM_SaveText": "IF Save Text📝",
    "IF_LLM_DisplayText": "IF Display Text📟",
    "IF_LLM_DisplayTextWildcard": "IF Display Text Wildcard📟",
    "IF_LLM_DisplayOmni": "IF Display Omni🔍",
    "IF_LLM_TextTyper": "IF Text Typer✍️",
    "IF_LLM_JoinText": "IF Join Text 📝",
    "IF_LLM_LoadImagesS": "IF LLM Load Images S 🖼️",
    "IF_LLM_ListModels": "IF LLM List Models 📚",
}

WEB_DIRECTORY = "./web"
__all__ = [
    "NODE_CLASS_MAPPINGS", 
    "NODE_DISPLAY_NAME_MAPPINGS", 
    "WEB_DIRECTORY", 
    "omost_function"
    ]