from .base_node import BaseNode
from crewai import Crew, Process


class CrewNode(BaseNode):
    def __init__(self):
        pass

    @classmethod
    def INPUT_TYPES(cls):
        return {
            "required": {
                "tasks": ("CREWAI_TASK", {"default": []}),
            },
            "optional": {
                "agents": ("CREWAI_AGENT", {"forceInput": False, "default": []}),
                "topic": ("STRING", {
                    "forceInput": False, "multiline": True, "default": ""
                }),
                "process": ("STRING", {
                    "default": "sequential",
                    "choices": ["sequential", "hierarchical"]
                }),
            },
        }
    INPUT_IS_LIST = (True, True, False, False)
    RETURN_TYPES = ("STRING", "STRING")
    FUNCTION = "create_crew"

    def create_crew(self, tasks, process, topic="", agents=[]):
        # Determine the process type
        isSequential: bool = process[0] == "sequential"
        process_type = Process.sequential if isSequential else Process.hierarchical
        result = ""
        # Create the Crew instance
        crew = Crew(
            agents=agents,
            tasks=tasks,
            process=process_type,
            topic=topic,
        )
        print("\n\n📎", crew, "\n\n")
        # Kick off the process and capture the result
        try:
            result = crew.kickoff(inputs={'topic': topic})
            print("\n\n📎Crew AI - result", result)
        except IndexError as e:
            print("Error during processing:", e)
        return (result)