from machine_learning import model, words, labels, ans_data, npl import datetime import random import numpy def bag_of_words(s, words): bag = [0 for _ in range(len(words))] s_words = npl(s.lower()) s_words = [word.lemma_ for word in s_words] for se in s_words: for i, w in enumerate(words): if w == se: bag[i] = 1 return numpy.array(bag) unknown = ["I'm afraid I don't follow; could you perhaps give more detail?", "I'm sorry, but I need you to elaborate a little bit more.", "I don't understand, can you try another question?"] def chat(): print("Start talking with the bot (type quit to stop)!") print("\n---------------------------------------------\nCellanet: Hello there!") while True: inp = input("You: ") if inp.lower() == "quit": break results = model.predict([bag_of_words(inp, words)]) # print(f"Predict: {results}\n") results_index = numpy.argmax(results) max_result = numpy.max(results) # print(f"max: {max_result}") if max_result < 0.65: tag = 'unknown' else: tag = labels[results_index] responses = [] print(f"({tag})") if tag in ans_data: if tag == 'what time': responses.append(f"Now is {datetime.datetime.now()}.") else: for x in ans_data[tag]: for z in x: responses.append(z) else: for x in unknown: responses.append(x) print('Cellanet:', random.choice(responses)) # return f"Cellanet: {random.choice(responses)}." def onlineChat(inp): print("\n---------------------------------------------\n") while True: # inp = input("You: ") if inp.lower() == "quit": break results = model.predict([bag_of_words(inp, words)]) # print(f"Predict: {results}\n") results_index = numpy.argmax(results) max_result = numpy.max(results) # print(f"max: {max_result}") if max_result < 0.45: tag = 'unknown' else: tag = labels[results_index] responses = [] print(f"({tag})") if tag in ans_data: if tag == 'what time': responses.append(f"Now is {datetime.datetime.now()}.") else: for x in ans_data[tag]: for z in x: responses.append(z) else: for x in unknown: responses.append(f"*********** {x} ***********") f = open('question and answer.txt', 'a') f.write(f"- - {inp}\n - {random.choice(responses)}.\n\n\n") f.close() return f"Cellanet: {random.choice(responses)}" # chat()