wehub-resource-sync
52 lines
1.8 KiB
Python
52 lines
1.8 KiB
Python
"""
|
|
ComfyUI Agnes AI Extension
|
|
===========================
|
|
Provides 7 custom nodes to call Agnes AI models within ComfyUI:
|
|
|
|
- AgnesAPIKeyConfig : Persist API key to plugin config file
|
|
- AgnesLLMChat : LLM dialogue (agnes-2.0-flash)
|
|
- AgnesImageReverse : Image reverse prompt analysis
|
|
- AgnesImageToImage : Image-to-image / image editing (multi-image)
|
|
- AgnesTextToImage : Text-to-image generation
|
|
- AgnesImageToVideo : Image-to-video generation (multi-image)
|
|
- AgnesTextToVideo : Text-to-video generation
|
|
|
|
Author: ComfyUI Agnes AI Plugin
|
|
Version: 1.1.0
|
|
"""
|
|
|
|
import os
|
|
|
|
# Web directory for frontend extensions
|
|
WEB_DIRECTORY = os.path.join(os.path.dirname(__file__), "web", "js")
|
|
|
|
from .nodes.llm_chat import AgnesLLMChat
|
|
from .nodes.image_reverse import AgnesImageReverse
|
|
from .nodes.img2img import AgnesImageToImage
|
|
from .nodes.text2img import AgnesTextToImage
|
|
from .nodes.img2video import AgnesImageToVideo
|
|
from .nodes.text2video import AgnesTextToVideo
|
|
from .nodes.api_key_config import AgnesAPIKeyConfig
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"AgnesAPIKeyConfig": AgnesAPIKeyConfig,
|
|
"AgnesLLMChat": AgnesLLMChat,
|
|
"AgnesImageReverse": AgnesImageReverse,
|
|
"AgnesImageToImage": AgnesImageToImage,
|
|
"AgnesTextToImage": AgnesTextToImage,
|
|
"AgnesImageToVideo": AgnesImageToVideo,
|
|
"AgnesTextToVideo": AgnesTextToVideo,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"AgnesAPIKeyConfig": "Agnes API Key Config",
|
|
"AgnesLLMChat": "Agnes LLM Chat",
|
|
"AgnesImageReverse": "Agnes Image Reverse Prompt",
|
|
"AgnesImageToImage": "Agnes Image-to-Image",
|
|
"AgnesTextToImage": "Agnes Text-to-Image",
|
|
"AgnesImageToVideo": "Agnes Image-to-Video",
|
|
"AgnesTextToVideo": "Agnes Text-to-Video",
|
|
}
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
|