"""
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)