from bot.common import dp
from .txt2img import generate_command
from .set_model import set_model_command
from .status import get_status
from .current import get_current
from .set_settings import (
    set_height_command, set_negative_prompt_command, set_size_command, set_steps_command, set_width_command,
    set_prompt_command, set_sampler_command, set_cfg_scale_command, set_restore_faces_command
)


def register():
    dp.register_message_handler(txt2img.generate_command, commands='generate')
    dp.register_message_handler(set_settings.set_prompt_command, commands='setprompt')
    dp.register_message_handler(set_settings.set_height_command, commands='setheight')
    dp.register_message_handler(set_settings.set_width_command, commands='setwidth')
    dp.register_message_handler(set_settings.set_negative_prompt_command, commands='setnegative')
    dp.register_message_handler(set_settings.set_size_command, commands='setsize')
    dp.register_message_handler(set_settings.set_steps_command, commands='setsteps')
    dp.register_message_handler(set_settings.set_sampler_command, commands='setsampler')
    dp.register_message_handler(set_settings.set_cfg_scale_command, commands='setscale')
    dp.register_message_handler(set_settings.set_restore_faces_command, commands='setfaces')
    dp.register_message_handler(set_model.set_model_command, commands='setmodel')
    dp.register_message_handler(status.get_status, commands='status')
    dp.register_message_handler(current.get_current, commands='current')