diff --git a/handlers/davinci.py b/handlers/davinci.py index 0bf4283..146ab7e 100644 --- a/handlers/davinci.py +++ b/handlers/davinci.py @@ -1,4 +1,4 @@ -import openai +from openai import AsyncOpenAI from aiogram import types from create_bot import dp, bot import logging @@ -6,69 +6,59 @@ from global_conf import CONFIG admins = CONFIG['telegram_admins_ids'] -openai.api_key = "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" + ban_comands = ['/dick','/topdick'] # Image_promt =['тупица сгенерируй','тупица сгенерируй картинку',\ # 'тупица отправь картинку', 'тупица отправь картинку'] +client = AsyncOpenAI(api_key="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") - -@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() +# @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: +# 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", +# 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('у меня не получилось') +# ) +# image_url = response['data'][0]['url'] +# await message.answer('Генерирую изображение...') +# await message.reply_photo(image_url) +# except Exception as e: +# await message.reply('у меня не получилось') + + +async def gpt_text(req): + completion = await client.chat.completions.create( + messages=[ + { + "role": "user", + "content": req, + } + ], + model="gpt-4", + ) + return completion @dp.message_handler() async def send(message: types.Message): if message.from_user.id in admins and message.chat.type == 'private': + response = await gpt_text(message.text) + await message.answer(response.choices[0].message.content) - ## 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'])