from enum import Enum
from typing import List, Dict

class ConfigurationType(Enum):
    CHATBOT = "Basic Chatbot"
    DOCUMENT_EXTRACTION = "Basic Document Extraction"
    CHATBOT_AND_DOC = "Basic Chatbot and Document Extraction"

CONFIGURATION_TYPE_FUNCTIONS = {
    ConfigurationType.CHATBOT: [
        "generate_intents_csv",
        "generate_engage_plan_test_mandays",
        "generate_dev_mandays",
    ],
    ConfigurationType.DOCUMENT_EXTRACTION: [
        "generate_page_plan_test_mandays",
        "generate_dev_mandays",
    ],
    ConfigurationType.CHATBOT_AND_DOC: [
        "generate_intents_csv",
        "generate_hybrid_plan_test_mandays",
        "generate_dev_mandays",
    ]
}

ANALYZE_COMPONENTS_FUNCTIONS = {
    ConfigurationType.CHATBOT: [
        "analyze_planning_testing_mandays",
        "analyze_development_mandays",
        "analyze_MVP_intents"
    ],
    ConfigurationType.DOCUMENT_EXTRACTION: [
        "analyze_planning_testing_mandays",
        "analyze_development_mandays",
    ],
    ConfigurationType.CHATBOT_AND_DOC: [
        "analyze_planning_testing_mandays",
        "analyze_development_mandays",
        "analyze_MVP_intents"
    ]
}

RECALCULATE_MVP_MANDAYS_FUNCTIONS = {
    ConfigurationType.CHATBOT: [
        "recalculate_engage_MVP_mandays"
    ],
    ConfigurationType.DOCUMENT_EXTRACTION: [
        "recalculate_page_MVP_mandays"
    ],
    ConfigurationType.CHATBOT_AND_DOC: [
        "recalculate_engage_MVP_mandays"
    ]
}

GENERATE_MVP_MANDAYS_FUNCTIONS = {
    ConfigurationType.CHATBOT: [
        "generate_engage_MVP_mandays"
    ],
    ConfigurationType.DOCUMENT_EXTRACTION: [
        "generate_page_MVP_mandays"
    ],
    ConfigurationType.CHATBOT_AND_DOC: [
        "generate_engage_MVP_mandays"
    ]
}

GENERATE_FINAL_DOCUMENT_FUNCTIONS = {
    ConfigurationType.CHATBOT: [
        "generate_engage_MVP_prd",
        "generate_engage_BD_SOW",
        "generate_Tech_SOW"
    ],
    ConfigurationType.DOCUMENT_EXTRACTION: [
        "generate_page_MVP_prd",
        "generate_page_BD_SOW",
        "generate_Tech_SOW"
    ],
    ConfigurationType.CHATBOT_AND_DOC: [
        "generate_engage_MVP_prd",
        "generate_engage_BD_SOW",
        "generate_Tech_SOW"
    ]
}