@LidarInfo

import asyncio

import os

import re

import telebot

from EdgeGPT import Chatbot, ConversationStyle

from telebot.util import quick_markup

TOKEN = "token"

COOKIE_PATH = './cookie.json'

bot = telebot.TeleBot(TOKEN)

EDGES = {}

my_conversation_style = ConversationStyle.balanced

@bot.message_handler(commands=['start', 'help'])

def send_welcome(message):

  bot.reply_to(

    message, "Введи /help для показа информации\nДля смены стилей вводи /switch и следующую приписку: \ncreative (Креативный)\nbalanced (Баланс)\nprecise (Строгий)")

@bot.message_handler(commands=['reset'])

def send_reset(message):

  try:

    if message.from_user.id not in EDGES:

      EDGES[message.from_user.id] = Chatbot(cookie_path=COOKIE_PATH)

    asyncio.run(EDGES[message.from_user.id].reset())

  except Exception as e:

    bot.reply_to(message, "Ошибка: " + str(e), parse_mode='Markdown')

  else:

    bot.reply_to(message, "Очищено успешно!")

@bot.message_handler(commands=['switch'])

def switch_style(message):

  message_list = message.text.split(" ")

  if len(message_list) > 1:

    styles = {

      "creative": ConversationStyle.creative,

      "balanced": ConversationStyle.balanced,

      "precise": ConversationStyle.precise

    }

    if message_list[1] in styles:

      global my_conversation_style

      my_conversation_style = styles[message_list[1]]

      bot.reply_to(

        message, f"Текущий стиль: {message_list[1].capitalize()}")

    else:

      bot.reply_to(

        message, "Выберите один из параметров! (/help)")

  else:

    bot.reply_to(

      message, "Выберите один из параметров! (/help)")

@bot.message_handler(func=lambda msg: True)

def response_all(message):

  message_text = ''

  if message.chat.type == "private":

    message_text = message.text

    bot.reply_to(message, "Обработка запроса, ожидайте!")

    response_list = asyncio.run(bing_chat(message_text, message))

    if len(response_list[0]) > 4095:

      for x in range(0, len(response_list[0]), 4095):

        bot.reply_to(

          message, response_list[0][x:x + 4095], parse_mode='Markdown', reply_markup=response_list[1])

    else:

      bot.reply_to(

        message, response_list[0], parse_mode='Markdown', reply_markup=response_list[1])

@bot.callback_query_handler(func=lambda msg: True)

def callback_all(callback_query):

  try:

    response_list = asyncio.run(

      bing_chat(callback_query.data, callback_query))

  except Exception as e:

    bot.reply_to(callback_query.message, "Ошибка: " +

           str(e), parse_mode='Markdown')

  else:

    if len(response_list[0]) > 4095:

      for x in range(0, len(response_list[0]), 4095):

        bot.reply_to(

          callback_query.message, response_list[0][x:x +

                               4095], parse_mode='Markdown',

          reply_markup=response_list[1])

    else:

      bot.reply_to(

        callback_query.message, response_list[0], parse_mode='Markdown', reply_markup=response_list[1])

async def bing_chat(message_text, message):

  if message.from_user.id not in EDGES:

    EDGES[message.from_user.id] = Chatbot(cookie_path=COOKIE_PATH)

  response_dict = await EDGES[message.from_user.id].ask(prompt=message_text,

                             conversation_style=my_conversation_style)

  if 'text' in response_dict['item']['messages'][1]:

    response = re.sub(r'\[\^\d\^]', '',

             response_dict['item']['messages'][1]['text'])

  else:

    response = "Что-то не так. Пожалуйста, перезагрузите чат"

  if 'suggestedResponses' in response_dict['item']['messages'][1]:

    suggested_responses = response_dict['item']['messages'][1]['suggestedResponses']

    markup = quick_markup({

      re.sub(r'\[\^\d\^]', '', suggested_responses[i]['text']): {

        'callback_data': suggested_responses[i]['text'].encode('utf-8')[:64].decode('utf-8', 'ignore')}

      for i in range(min(len(suggested_responses), 3))

    }, row_width=1)

  else:

    markup = quick_markup({

      'Предложенных ответов нет': {'url': 'https://bing.com/chat'}

    }, row_width=1)

  throttling = response_dict['item']['throttling']

  if 'maxNumUserMessagesInConversation' in throttling and 'numUserMessagesInConversation' in throttling:

    max_num_user_messages_in_conversation = throttling['maxNumUserMessagesInConversation']

    num_user_messages_in_conversation = throttling['numUserMessagesInConversation']

    response += "\n———————\n"

    response += f"Контекст: {num_user_messages_in_conversation} / {max_num_user_messages_in_conversation}"

  if num_user_messages_in_conversation >= max_num_user_messages_in_conversation:

    await EDGES[message.from_user.id].reset()

    response += "\nКонтекст был автоматически очищен"

  attributions = response_dict['item']['messages'][1]['sourceAttributions']

  if len(attributions) >= 3:

    response += "\n———————\nИсточники:\n"

    for i in range(3):

      provider_display_name = re.sub(

        r'\[\^\d\^]', '', attributions[i]['providerDisplayName'])

      see_more_url = re.sub(

        r'\[\^\d\^]', '', attributions[i]['seeMoreUrl'])

      response += f"{i + 1}.[{provider_display_name}]({see_more_url})\n"

  response_list = [response, markup]

  return response_list

bot.polling()