Проблема aiogram python opeanai

Делаю бота ChatGPT в телеграмм, всё отлично работает. Решил добавить команду /image, по которой пользователь сначала выберет размер, количество и потом запрос к изображению, и Dall'e сгенерирует картинку.

Команду я-то сделал, но проблема в том, что одновременно с чатгпт работать не хотят. Я сделал переменные, чтобы gpt_answer не реагировал на сообщение, пока /image не завершит работу, но не получилось вот код:

@dp.message_handler()
async def gpt_answer(message: types.Message):
    global is_image_processing
    if message.text.startswith('/image'):
        await images(message)
        return
    if not is_image_processing:
        #Весь код функции в нём ошибок точно нет

#image
@dp.message_handler(commands=['image'])
async def images(message: types.Message):
    is_image_processing = True
    image_zapiska()
    zapiska()
    ins1 = InlineKeyboardButton('256x256', callback_data='size_256x256')
    ins2 = InlineKeyboardButton('512x512', callback_data='size_512x512')
    ins3 = InlineKeyboardButton('1024x1024', callback_data='size_1024x1024')
    keyboard = InlineKeyboardMarkup().add(ins1, ins2, ins3)
    await message.reply("Выберите размеры изображения:", reply_markup=keyboard)

@dp.callback_query_handler(lambda c: c.data.startswith('size_'))
async def process_callback_size(callback_query: types.CallbackQuery):
    size = callback_query.data.split('_')[1]
    await bot.answer_callback_query(callback_query.id)
    inn1 = InlineKeyboardButton('1', callback_data='num_1')
    inn2 = InlineKeyboardButton('2', callback_data='num_2')
    inn3 = InlineKeyboardButton('3', callback_data='num_3')
    inn4 = InlineKeyboardButton('4', callback_data='num_4')
    keyboard = InlineKeyboardMarkup().add(inn1, inn2, inn3, inn4)
    await bot.edit_message_text(chat_id=callback_query.message.chat.id, message_id=callback_query.message.message_id, text=f'Размер изображения: {size}\nВыберите количество изображений:', reply_markup=keyboard)

@dp.callback_query_handler(lambda c: c.data.startswith('num_'))
async def process_callback_num(callback_query: types.CallbackQuery):
    global is_image_processing
    count = callback_query.data.split('_')[1]
    await bot.answer_callback_query(callback_query.id)
    # Добавляем обработку выбранного количества изображений
    await bot.send_message(chat_id=callback_query.message.chat.id, text=f'Выбрано количество изображений: {count} \nвведите запрос:')
    is_image_processing = True

@dp.message_handler()
async def process_user_input(message: types.Message):
    global is_image_processing
    if is_image_processing:
        # Обработка запроса пользователя
        promt = message.text
        print(promt)
        is_image_processing = False
        return
    else:
        print(is_image_processing)
        return