16nic--comfyui-agnes-ai
wehub-resource-sync
65 行
2.1 KiB
Python
65 行
2.1 KiB
Python
"""
|
|
Agnes Image Reverse Prompt Node
|
|
===============================
|
|
Analyzes an input image and generates a detailed AI image generation prompt
|
|
that describes the image. Uses Agnes-2.0-Flash vision capability.
|
|
"""
|
|
|
|
from typing import Tuple
|
|
|
|
from ..api import AgnesClient, get_api_key, CHAT_MODEL, AVAILABLE_CHAT_MODELS, tensor_to_pil
|
|
|
|
|
|
class AgnesImageReverse:
|
|
"""Agnes AI Image Reverse Prompt node - generates prompts from images."""
|
|
|
|
CATEGORY = "Agnes AI"
|
|
RETURN_TYPES = ("STRING", "STRING",)
|
|
RETURN_NAMES = ("prompt", "brief_prompt",)
|
|
FUNCTION = "reverse"
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"api_key": ("STRING", {
|
|
"multiline": False,
|
|
"default": "",
|
|
"placeholder": "sk-...",
|
|
"tooltip": "Your Agnes AI API key",
|
|
}),
|
|
"image": ("IMAGE", {
|
|
"tooltip": "Input image to analyze for prompt generation",
|
|
}),
|
|
"model": (AVAILABLE_CHAT_MODELS, {
|
|
"default": CHAT_MODEL,
|
|
"tooltip": "Vision-capable model for image analysis",
|
|
}),
|
|
},
|
|
}
|
|
|
|
def reverse(self, api_key: str, image, model: str = CHAT_MODEL) -> Tuple[str, str]:
|
|
# Runtime fallback: try config file if widget value is empty
|
|
if not api_key.strip():
|
|
api_key = get_api_key()
|
|
if not api_key.strip():
|
|
err = "[Error: API key is required]"
|
|
return (err, err)
|
|
|
|
try:
|
|
# Convert ComfyUI tensor to PIL
|
|
pil_img = tensor_to_pil(image)
|
|
|
|
client = AgnesClient(api_key)
|
|
|
|
# Generate detailed prompt
|
|
detailed = client.reverse_prompt(pil_img, model=model, detail="detailed")
|
|
|
|
# Generate brief prompt
|
|
brief = client.reverse_prompt(pil_img, model=model, detail="brief")
|
|
|
|
return (detailed, brief)
|
|
except Exception as e:
|
|
err = f"[Error] {str(e)}"
|
|
return (err, err)
|