Создайте своего первого Telegram-бота на базе ChatGPT: Пошаговое руководство 2023

Вступление

Краткая информация о ChatGPT

Приступаем к работе

mkdir chatgptbot cd chatgptbot

python3 -m venv env source env/bin/activate

pip install openai pip install python-dotenv

import os import json from dotenv import load_dotenv import openai

Создание класса и функций

class Copilot: def clear_text(self, text): a = text.replace("\n", " ") b = a.split() c = " ".join(b) return c def get_answer(self, question): prompt = question load_dotenv() openai.api_key = os.getenv("CHAT_GPT3_API_KEY") response = openai.Completion.create( engine="text-davinci-003", prompt=prompt, max_tokens=512, temperature=0.5, ) json_object = response # Convert the JSON object to a JSON string json_string = json.dumps(json_object) # Parse the JSON string using json.loads() parsed_json = json.loads(json_string) text = parsed_json['choices'][0]['text'] cleared_text = self.clear_text(text) return cleared_text

Получение API-ключа ChatGPT3

touch .env

Вы можете получить его здесь: OpenAI

CHAT_GPT3_API_KEY=paste here your api key

copilot = Copilot() a = copilot.get_answer("Hello, there!") print(a)

Telegram-бот

pip install python-telegram-bot --pre pip install requests

import os import json import requests import time from copilot import Copilot from dotenv import load_dotenv from telegram import ( ReplyKeyboardMarkup, Update, KeyboardButton, ) from telegram.ext import ( Application, CommandHandler, ContextTypes, ConversationHandler, MessageHandler, filters, )

(ENTRY_STATE, QUESTION_STATE, ) = range(2)

Его можно получить здесь: BotFather

TELEGRAM_BOT_TOKEN=paste your Telegram bot token

if __name__ == '__main__': load_dotenv() application = Application.builder().token(os.getenv("TELEGRAM_BOT_TOKEN")).read_timeout(100).get_updates_read_timeout(100).build() conv_handler = ConversationHandler( entry_points=[CommandHandler('start', start)], states={ ENTRY_STATE: [ MessageHandler(filters.Regex('^Back$'), start), MessageHandler(filters.Regex('^Question-Answering$'), pre_query_handler), ], QUESTION_STATE: [ MessageHandler(filters.Regex('^Back$'), start), MessageHandler(filters.TEXT, pre_query_answer_handler), ], }, fallbacks=[], ) application.add_handler(conv_handler) print("Bot is running ...") application.run_polling()

def _generate_copilot(prompt: str): """Gets answer from copilot""" copilot = Copilot() c = copilot.get_answer(prompt) return c

async def start(update: Update, context: ContextTypes): """Start the conversation and ask user for an option.""" button = [[KeyboardButton(text="Question-Answering")]] reply_markup = ReplyKeyboardMarkup( button, resize_keyboard=True ) await update.message.reply_text( "Choose an option: 👇🏻", reply_markup=reply_markup, ) return ENTRY_STATE

#Handling the question async def pre_query_handler(update: Update, context: ContextTypes): """Ask the user for a query.""" button = [[KeyboardButton(text="Back")]] reply_markup = ReplyKeyboardMarkup( button, resize_keyboard=True ) await update.message.reply_text( "Enter your text: 👇🏻", reply_markup=reply_markup, ) return QUESTION_STATE

#Handling the answer async def pre_query_answer_handler(update: Update, context: ContextTypes): """Display the answer to the user.""" button = [[KeyboardButton(text="Back")]] reply_markup = ReplyKeyboardMarkup( button, resize_keyboard=True ) question = update.message.text answer = _generate_copilot(question) context.user_data['answer'] = answer await update.message.reply_text( answer, reply_markup=reply_markup, ) return QUESTION_STATE

Полезные ссылки

Создайте своего первого Telegram-бота на базе…

Вступление Хотели бы вы создать чат-бота, который мог бы вести интеллектуальные беседы с вашими…

uproger.com