main.py

import telebot
import openai
from deep_translator import GoogleTranslator
import APIs
import mes

openai.api_key = APIs.API_KEY_OPENAI  # Api ключ OpenAI
bot = telebot.TeleBot(APIs.API_KEY_TELEGRAM_BOT)  # Api ключ Telegram бота


@bot.message_handler(commands=['start', 'clear', 'image'])
def handle_message(message):
    print('Команда обнаружена')
    if message.text == '/clear':
        f = open(f'Logs/{message.from_user.id}.txt', 'w')
        f.write(mes.mes)
        f.close()
        bot.send_message(chat_id=message.from_user.id, text='Память бота очищена')
    elif message.text == '/start':
        f = open(f'Logs/{message.from_user.id}.txt', 'w')
        f.write(mes.mes)
        f.close()
        bot.send_message(chat_id=message.from_user.id, text='Задавайте любые вопрысы или генерируйте картинки по описанию через /image ОПИСАНИЕ')
    elif message.text.find('/image') == 0:
        bot.send_message(chat_id=message.from_user.id, text='Генерация...')
        image = openai.Image.create(
            prompt=message.text.removeprefix('/image '),
            n=2,
            size="1024x1024"
        )
        bot.send_photo(chat_id=message.from_user.id, photo=image.data[0].url)
        bot.send_document(chat_id=message.from_user.id, document=image.data[0].url)
        bot.send_photo(chat_id=message.from_user.id, photo=image.data[1].url)
        bot.send_document(chat_id=message.from_user.id, document=image.data[1].url)


@bot.message_handler(func=lambda message: True)
def handle_message(message):
    f = open(f'Logs/{message.from_user.id}.txt', 'r')
    zapros = f.read() + '\nYou: ' + message.text + '\nComputer:'
    f.close()
    try:
        response = openai.Completion.create(
            model="text-davinci-003",
            prompt=zapros,
            temperature=0.5,
            max_tokens=1000,
            top_p=1,
            frequency_penalty=0.5,
            presence_penalty=0.0,
            stop=['You:']
        )
        bot.send_message(chat_id=message.from_user.id, text=response['choices'][0]['text'])
        bot.send_message(chat_id=message.from_user.id,
                         text=GoogleTranslator(source='auto', target='ru').translate(response['choices'][0]['text']))
        zapros = zapros + response['choices'][0]['text']
        f = open(f'Logs/{message.from_user.id}.txt', 'w')
        f.write(zapros)
        f.close()
    except:
        bot.send_message(chat_id=message.from_user.id, text='Ошибка')


bot.infinity_polling()