# -*- codding: utf-8 -*-
import os
import telebot
from chatterbot import ChatBot
from chatterbot.trainers import ListTrainer

#Token do Telegram
bot = telebot.TeleBot("835354383:AAHrKka7RGg4iLHweYejPgrf6oBqgB5Z86Y")

#Esse é o chatbot que eu criei e funciona fora do telegram
def Futaba(message):
    chatbot = ChatBot("Futaba")
    trainer = ListTrainer(chatbot)

    #Pega os corpus na pasta arquivos
    for arquivos in os.listdir('arquivos'):
        chats = open('arquivos/' + arquivos, 'r', encoding="utf8").readlines()
        trainer.train(chats)

    print(
        "Hey, meu nome é Futaba e você pode se sentir confortável para conversar comigo mesmo que conversar não seja lá a coisa mais confortável pra você.")
    response = chatbot.get_response("Hey!")
    print(response)

    while True:
        resq = input('Você: ')
        resp = chatbot.get_response(resq)
        if float(response.confidence) > 0.5:
            print('Futaba: ' + str(resp))
        else:
            print('Como você tem se sentido?')

#Receber e enviar resposta
@bot.message_handler(commands=['start', 'help'])
def send_welcome(message):
    bot.reply_to(message, u"Olá, bem-vindo ao bot!")

#Receber qualquer outra mensagem
@bot.message_handler(func=lambda message:True)
def mensaje(message):
    Futaba(message.text)
    bot.reply_to(message.text)

bot.polling()