import openai from aiogram import types from create_bot import dp, bot import logging from global_conf import CONFIG admins = CONFIG['telegram_admins_ids'] token = CONFIG['openai']['token'] openai.api_key = CONFIG['openai']['api_key'] ban_comands = ['/dick','/topdick'] # Image_promt =['тупица сгенерируй','тупица сгенерируй картинку',\ # 'тупица отправь картинку', 'тупица отправь картинку'] @dp.message_handler(commands=['image']) async def send_image(message: types.Message): if message.from_user.id in admins and message.chat.type == 'private': try: description = message.text.replace('/image', '').strip() if not description: await message.reply('Ты даун, описание запроса напиши после пробела') return except Exception as e: logging.error(f'Error in send_image: {e}') try: response = openai.Image.create( prompt=message.text, n=1, size="1024x1024", response_format="url", ) image_url = response['data'][0]['url'] await message.answer('Генерирую изображение...') await message.reply_photo(image_url) except Exception as e: await message.reply('у меня не получилось') @dp.message_handler() async def send(message: types.Message): if message.from_user.id in admins and message.chat.type == 'private': ## TODO уточнить остатки токенов и денег response = openai.Completion.create( model="gpt-4-turbo", prompt=message.text[7:], temperature=0.7, max_tokens=1000, top_p=1.0, frequency_penalty=0.0, presence_penalty=0.6, stop=["сброс"] ) await message.reply(response['choices'][0]['text']) elif message.reply_to_message.from_user.is_bot: response = openai.Completion.create( model="gpt-4-turbo", prompt=message.text, temperature=0.7, max_tokens=1000, top_p=1.0, frequency_penalty=0.0, presence_penalty=0.6, stop=["сброс"] ) await message.reply(response['choices'][0]['text'])