from enums.enums import Steps
from ui_manager.ui_manager import UIManager

user_msg = "Please describe what you need to do. To get the best results try to answer all the following questions:"


def use_streamlit(um: UIManager):
    if um.get_current_step() == Steps.INITIAL_STATE.value:
        um.build_ui_for_initial_state(user_msg)
    elif um.get_current_step() == Steps.PARSING_ANSWERS.value:
        um.build_ui_for_parsing_answers()
    elif um.get_current_step() == Steps.PARSING_ERROR.value:
        um.build_ui_for_validate_data_after_correction()
    elif um.get_current_step() == Steps.ASK_AGAIN.value:
        um.build_ui_for_ask_again()
    elif um.get_current_step() == Steps.FIND_CATEGORIES.value:
        um.build_ui_for_check_category()
    elif um.get_current_step() == Steps.VALIDATE_DATA.value:
        um.build_ui_to_confirm_form_data()
    elif um.get_current_step() == Steps.FORM_CREATED.value:
        um.build_ui_for_form_created()


use_streamlit(UIManager())