import openai
from aiogram import types
from create_bot import dp, bot
import logging
from global_conf import CONFIG

admins = CONFIG['telegram_admins_ids']

openai.api_key = "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"

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.ChatCompletion.create(
            model="gpt-4",
            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'])

    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'])