class UserStateManager:
    def __init__(self):
        # Словарь для хранения состояния каждого пользователя
        self.user_states = {}

    def set_state(self, chat_id, state):
        """Устанавливает состояние для пользователя."""
        self.user_states[chat_id] = state

    def get_state(self, chat_id):
        """Получает текущее состояние пользователя."""
        return self.user_states.get(chat_id, "MAIN_MENU")

    def reset_state(self, chat_id):
        """Сбрасывает состояние пользователя в главное меню."""
        self.user_states[chat_id] = "MAIN_MENU"
