from aiogram.fsm.state import State, StatesGroup

class ChooseModelState(StatesGroup):
    choosing_model = State()

class SendMessageState(StatesGroup):
    waiting_for_message = State()