#@title flows.py
GENERAL_STATE_FLOWS = [
        "homework",
        "homework_with_question",
        "recommend",
        "recommend_with_topic",
        "assess_knowledge",
        "assess_knowledge_with_topic",
        "explain_drill",
        "exercise",
        "exercise_with_topic",
        "personal",
        "finish_learning",
        "provide_topic",
        "provide_subject",
        "provide_subtopic",
        "provide_concept",
        "general_intent",
        "explain_answer",
        "explain_answer_with_topic",
        "back_to_study",
        "summarize",
        "unknown"
    ]

UNDERSTAND_STATE_FLOWS = [
        "homework",
        "homework_with_question",
        "provide_topic",
        "provide_subject",
        "explain_answer",
        "explain_answer_with_topic",
        "core_explanation_repeat",
        "core_explanation_exercise",
        "core_explanation_exercise_with_topic",
        "core_explanation_video",
        "core_explanation_video_with_topic",
        "core_explanation_protips",
        "negative_respond",
        "positive_respond",
        "assess_knowledge",
        "assess_knowledge_with_topic",
        "recommend",
        "recommend_with_topic",
        "exercise",
        "exercise_with_topic",
        "personal",
        "unknown",
        "general_intent"
    ]

HOMEWORK_STATE_FLOWS = ["homework_with_question",
        "explain_answer",
        "explain_answer_with_topic",
        "provide_topic",
        "provide_subject",
        "homework",
        "recommend",
        "recommend_with_topic",
        "assess_knowledge",
        "assess_knowledge_with_topic",
        "exercise",
        "exercise_with_topic",
        "exercise_multiple_question",
        "exercise_multiple_question_with_topic",
        "similar_question",
        "personal",
        "finish_learning",
        "unknown",
        "back_to_study",
]

RECOMMEND_MATERIAL_FLOWS = [
        "homework",
        "homework_with_question",
        "understand",
        "understand_with_topic",
        "recommend",
        "recommend_with_topic",
        "provide_topic",
        "provide_subject",
        "explain_answer",
        "assess_knowledge",
        "assess_knowledge_with_topic",
        "exercise",
        "exercise_with_topic",
        "exercise_multiple_question",
        "exercise_multiple_question_with_topic",
        "personal",
        "finish_learning",
        "go_back_to_general",
        "unknown"
]

PERSONAL_STATE_FLOWS = [
        "homework",
        "homework_with_question",
        "recommend",
        "recommend_with_topic",
        "assess_knowledge",
        "assess_knowledge_with_topic",
        "personal",
        "exercise",
        "exercise_with_topic",
        "explain_answer",
        "explain_answer_with_topic",
        "provide_topic",
        "provide_subject",
        "back_to_study",
        "finish_learning",
        "unknown"
]

EXERCISE_STATE_FLOWS = [
        "exercise",
        "exercise_with_image",
        "exercise_with_topic",
        "exercise_multiple_question",
        "exercise_multiple_question_with_topic",
        "provide_topic",
        "provide_subject",
        "explain_answer",
        "explain_answer_with_topic",
        "provide_answer",
        "similar_question",
        "assess_knowledge",
        "assess_knowledge_with_topic",
        "homework",
        "homework_with_question",
        "recommend",
        "recommend_with_topic",
        "finish_learning",
        "personal",
        "back_to_study",
        "unknown"
]

ASSESS_KNOWLEDGE_STATE_FLOWS = [
        "assess_knowledge",
        "assess_knowledge_with_topic",
        "assess_knowledge_answer",
        "explain_drill",
        "provide_topic",
        "provide_subject",
        "diagnosis_result",
        "explain_answer",
        "explain_answer_with_topic",
        "homework",
        "homework_with_question",
        "recommend",
        "recommend_with_topic",
        "exercise",
        "exercise_with_topic",
        "exercise_multiple_question",
        "exercise_multiple_question_with_topic",
        "general_intent",
        "finish_learning",
        "personal",
        "back_to_study",
        "unknown"
]

STATE_FLOWS_MAP = {
    "GeneralState":GENERAL_STATE_FLOWS,
    "HomeworkState":HOMEWORK_STATE_FLOWS,
    "ExerciseState":EXERCISE_STATE_FLOWS,
    "UnderstandState":UNDERSTAND_STATE_FLOWS,
    "RecommendMaterialState":RECOMMEND_MATERIAL_FLOWS,
    "PersonalState":PERSONAL_STATE_FLOWS,
    "AssessKnowledgeState":ASSESS_KNOWLEDGE_STATE_FLOWS,
}