from ai_bot.handlers.FluxHandler import FluxHandler
from ai_bot.handlers.MidjourneyHandler import MidjourneyHandler
from piapi_ai.PiapiAiApi import PiapiAiApi


states_instance_arr = [
    FluxHandler,
    MidjourneyHandler,
]

states = {}

for state in states_instance_arr:
    states.update(state.get_states())