import telebot
import Handlers
import config
import UserStates
import Data


Handlers.LoadData()
bot = telebot.TeleBot(config.token) #подключение бота к программе


@bot.message_handler(commands=['admin']) #Admin checker
def AdminComand(message):
    UserStates.SetState(message.chat.id, config.States.S_Admin.value)
    bot.send_message(message.chat.id,'Вы перешли в режим администратора, введите вопрос')

@bot.message_handler(func= lambda message: UserStates.GetState(message.chat.id)==config.States.S_Admin.value)
def QuestionInput(message):
    Data.question = message.text
    if message.text == '/exit':
        UserStates.SetState(message.chat.id, config.States.S_Chat.value)
        bot.send_message(message.chat.id, "Вы вышли из режима администратора")
    else:
        UserStates.SetState(message.chat.id, config.States.S_Admin_Q.value)
        bot.send_message(message.chat.id,'Введите ответ')


@bot.message_handler(func=lambda message: UserStates.GetState(message.chat.id)==config.States.S_Admin_Q.value)
def AnswerInput(message):
    Data.answer = message.text
    if message.text == '/exit':
        UserStates.SetState(message.chat.id, config.States.S_Chat.value)
        bot.send_message(message.chat.id, "Вы вышли из режима администратора")
    else:
        Handlers.AddQuestionToBase(Data.question,Data.answer)
        bot.send_message(message.chat.id, "Вопрос и ответ успешнно добавлены")
        UserStates.SetState(message.chat.id, config.States.S_Chat.value)
        Handlers.LoadData()



@bot.message_handler(content_types=['text']) #получение сообщения от пользователя
def SearchInBase(message):
    file = open('Messages.txt', 'a')
    file.write(message.text + '\n')
    file.close()
    print('начальная форма: ' + message.text )
    MessageBaseForm = Handlers.SentenceToNormal(message.text)
    print('словарная форма: '+MessageBaseForm)
    AnswerIndex, Value = Handlers.bestOfSimmularity(MessageBaseForm, Data.QuestionDataBase, Data.QuestionDict)

    if(Value * 100):
        bot.send_message(message.chat.id, Data.AnswerDataBase[AnswerIndex])
    else:
        bot.send_message(message.chat.id, 'Ответа на ваш вопрос нет')
    print()

bot.polling(none_stop=True)