-
Notifications
You must be signed in to change notification settings - Fork 0
/
Chatbot_Telegram.py
60 lines (43 loc) · 1.49 KB
/
Chatbot_Telegram.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
# 'É PRECISO IMPORTAR AS BIBLIOTECAS:
# pip install python-dotenv
# pip install openai
# pip install pyTelegramBotAPI
# INFORMAÇÕES DETALHADA DE UM VALIDADOR
import telebot
import openai
import os
from dotenv import load_dotenv
# CARREGANDO AS VARIÁVEIS QUE ESTÃO NO ARQUIVO .env
load_dotenv()
# CHAVE BOT - TELEGRAM
chave_api = os.environ.get("API_KEY_TELEGRAM_PROD")
# Ambiente de Teste - TELEGRAM
# chave_api = os.environ.get("API_KEY_TELEGRAM_TEST")
# CRIANDO A INTEGRAÇÃO DO BOT - TELEGRAM
bot = telebot.TeleBot(chave_api)
# CHAVE CONTA - OPENAI
openai.api_key = os.environ.get("API_KEY_OPENAI")
# VARIÁVEIS DE APOIO
texto = ""
resposta = ""
resultado = ""
# TODAS AS MENSAGENS CONSIDERAR - TELEGRAM
def verificar(mensagem):
if mensagem.text != "/start":
if mensagem.text != "\start":
return True
# INFORMA QUANDO ESTA FUNÇÃO SERÁ EXECUTADA
@bot.message_handler(func=verificar)
# FUNÇÃO RESPONDENDO A RESPOSTA
def responder(mensagem):
# MENSAGEM ESCRITA TELEGRAM
texto = mensagem.text
# RESPOSTA DO CHAT DE INTELIGENCIA ARTIFICIAL - OPENAI
resposta = openai.Completion.create(engine="text-davinci-003", prompt=texto, max_tokens=2048, n=1, stop=None,
temperature=0, )
resultado = resposta["choices"][0]["text"]
# ENVIANDO MENSAGEM COM UMA DAS OPÇÕES ACIMA PARA INTERAGIR COM O USUÁRIO
bot.reply_to(mensagem, resultado)
print("Preparado!")
# LOOP INFINITO PARA OUVIR AS MENSAGENS
bot.polling()