# Copyright (c) 2025 Patrick Motsch
# AI node definitions - map to methodAi actions.

from modules.shared.i18nRegistry import t

AI_NODES = [
    {
        "id": "ai.prompt",
        "category": "ai",
        "label": t("Prompt"),
        "description": t("Prompt eingeben und KI führt aus"),
        "parameters": [
            {"name": "aiPrompt", "type": "string", "required": True, "frontendType": "textarea",
             "description": t("KI-Prompt")},
            {"name": "resultType", "type": "string", "required": False, "frontendType": "select",
             "frontendOptions": {"options": ["txt", "json", "md", "csv", "xml", "html", "pdf", "docx", "xlsx", "pptx", "png", "jpg"]},
             "description": t("Ausgabeformat"), "default": "txt"},
            {"name": "documentList", "type": "string", "required": False, "frontendType": "hidden",
             "description": t("Dokumentenliste (via Wire oder DataRef)"), "default": ""},
            {"name": "simpleMode", "type": "boolean", "required": False, "frontendType": "checkbox",
             "description": t("Einfacher Modus"), "default": True},
        ],
        "inputs": 1,
        "outputs": 1,
        "inputPorts": {0: {"accepts": [
            "DocumentList", "AiResult", "TextResult", "Transit", "LoopItem", "ActionResult",
        ]}},
        "outputPorts": {0: {"schema": "AiResult"}},
        "meta": {"icon": "mdi-robot", "color": "#9C27B0", "usesAi": True},
        "_method": "ai",
        "_action": "process",
    },
    {
        "id": "ai.webResearch",
        "category": "ai",
        "label": t("Web-Recherche"),
        "description": t("Recherche im Web"),
        "parameters": [
            {"name": "prompt", "type": "string", "required": True, "frontendType": "textarea",
             "description": t("Recherche-Anfrage")},
        ],
        "inputs": 1,
        "outputs": 1,
        "inputPorts": {0: {"accepts": ["Transit"]}},
        "outputPorts": {0: {"schema": "AiResult"}},
        "meta": {"icon": "mdi-magnify", "color": "#9C27B0", "usesAi": True},
        "_method": "ai",
        "_action": "webResearch",
    },
    {
        "id": "ai.summarizeDocument",
        "category": "ai",
        "label": t("Dokument zusammenfassen"),
        "description": t("Dokumentinhalt zusammenfassen"),
        "parameters": [
            {"name": "documentList", "type": "string", "required": True, "frontendType": "hidden",
             "description": t("Dokumentenliste (via Wire oder DataRef)"), "default": ""},
            {"name": "summaryLength", "type": "string", "required": False, "frontendType": "select",
             "frontendOptions": {"options": ["brief", "medium", "detailed"]},
             "description": t("Kurz, mittel oder ausführlich"), "default": "medium"},
        ],
        "inputs": 1,
        "outputs": 1,
        "inputPorts": {0: {"accepts": ["DocumentList", "Transit"]}},
        "outputPorts": {0: {"schema": "AiResult"}},
        "meta": {"icon": "mdi-file-document-outline", "color": "#9C27B0", "usesAi": True},
        "_method": "ai",
        "_action": "summarizeDocument",
    },
    {
        "id": "ai.translateDocument",
        "category": "ai",
        "label": t("Dokument übersetzen"),
        "description": t("Dokument in Zielsprache übersetzen"),
        "parameters": [
            {"name": "documentList", "type": "string", "required": True, "frontendType": "hidden",
             "description": t("Dokumentenliste (via Wire oder DataRef)"), "default": ""},
            {"name": "targetLanguage", "type": "string", "required": True, "frontendType": "text",
             "description": t("Zielsprache (z.B. de, en, French)")},
        ],
        "inputs": 1,
        "outputs": 1,
        "inputPorts": {0: {"accepts": ["DocumentList", "Transit"]}},
        "outputPorts": {0: {"schema": "AiResult"}},
        "meta": {"icon": "mdi-translate", "color": "#9C27B0", "usesAi": True},
        "_method": "ai",
        "_action": "translateDocument",
    },
    {
        "id": "ai.convertDocument",
        "category": "ai",
        "label": t("Dokument konvertieren"),
        "description": t("Dokument in anderes Format konvertieren"),
        "parameters": [
            {"name": "documentList", "type": "string", "required": True, "frontendType": "hidden",
             "description": t("Dokumentenliste (via Wire oder DataRef)"), "default": ""},
            {"name": "targetFormat", "type": "string", "required": True, "frontendType": "select",
             "frontendOptions": {"options": ["docx", "pdf", "xlsx", "csv", "txt", "html", "json", "md"]},
             "description": t("Zielformat")},
        ],
        "inputs": 1,
        "outputs": 1,
        "inputPorts": {0: {"accepts": ["DocumentList", "Transit"]}},
        "outputPorts": {0: {"schema": "DocumentList"}},
        "meta": {"icon": "mdi-file-convert", "color": "#9C27B0", "usesAi": True},
        "_method": "ai",
        "_action": "convertDocument",
    },
    {
        "id": "ai.generateDocument",
        "category": "ai",
        "label": t("Dokument generieren"),
        "description": t("Dokument aus Prompt generieren"),
        "parameters": [
            {"name": "prompt", "type": "string", "required": True, "frontendType": "textarea",
             "description": t("Generierungs-Prompt")},
        ],
        "inputs": 1,
        "outputs": 1,
        "inputPorts": {0: {"accepts": ["Transit"]}},
        "outputPorts": {0: {"schema": "DocumentList"}},
        "meta": {"icon": "mdi-file-plus", "color": "#9C27B0", "usesAi": True},
        "_method": "ai",
        "_action": "generateDocument",
    },
    {
        "id": "ai.generateCode",
        "category": "ai",
        "label": t("Code generieren"),
        "description": t("Code aus Beschreibung generieren"),
        "parameters": [
            {"name": "prompt", "type": "string", "required": True, "frontendType": "textarea",
             "description": t("Code-Generierungs-Prompt")},
            {"name": "resultType", "type": "string", "required": False, "frontendType": "select",
             "frontendOptions": {"options": ["py", "js", "ts", "html", "java", "cpp", "txt", "json", "csv", "xml"]},
             "description": t("Datei-Endung der erzeugten Code-Datei"), "default": "py"},
        ],
        "inputs": 1,
        "outputs": 1,
        "inputPorts": {0: {"accepts": ["Transit"]}},
        "outputPorts": {0: {"schema": "AiResult"}},
        "meta": {"icon": "mdi-code-tags", "color": "#9C27B0", "usesAi": True},
        "_method": "ai",
        "_action": "generateCode",
    },
    {
        "id": "ai.consolidate",
        "category": "ai",
        "label": t("KI-Konsolidierung"),
        "description": t("Gesammelte Ergebnisse mit KI zusammenfassen, klassifizieren oder semantisch zusammenführen"),
        "parameters": [
            {"name": "mode", "type": "string", "required": False, "frontendType": "select",
             "frontendOptions": {"options": ["summarize", "classify", "semanticMerge"]},
             "description": t("Konsolidierungsmodus"), "default": "summarize"},
            {"name": "prompt", "type": "string", "required": False, "frontendType": "textarea",
             "description": t("Optionaler Prompt für die Konsolidierung"), "default": ""},
        ],
        "inputs": 1,
        "outputs": 1,
        "inputPorts": {0: {"accepts": ["AggregateResult", "Transit"]}},
        "outputPorts": {0: {"schema": "ConsolidateResult"}},
        "meta": {"icon": "mdi-table-merge-cells", "color": "#9C27B0", "usesAi": True},
        "_method": "ai",
        "_action": "consolidate",
    },
]