From 3965b1164388368fc4ff42a9371d00dd1f7699ae Mon Sep 17 00:00:00 2001
From: Pavel_Durov <pavel_durov@telegram.com>
Date: Thu, 20 Jul 2023 23:10:49 +0300
Subject: [PATCH] add generate images

---
 handlers/davinci.py | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)

diff --git a/handlers/davinci.py b/handlers/davinci.py
index c664f56..7a90c67 100644
--- a/handlers/davinci.py
+++ b/handlers/davinci.py
@@ -1,6 +1,7 @@
 import openai
 from aiogram import types, Dispatcher
 from create_bot import dp, bot
+import logging
 
 
 
@@ -9,6 +10,35 @@ token = '5947963644:AAF_GKgMmU5ovqMpc1KXIpcf4aN0JMyKPqc'
 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):
+    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.text.lower().startswith('тупица'):
@@ -38,6 +68,10 @@ async def send(message: types.Message):
             stop=["сброс"]
         )
         await message.reply(response['choices'][0]['text'])
+
+
+
+