def chat(): # variables used in response should be held here but can't be quickly changed sentence = request.get_data().decode('utf-8') answer = chatbot.response(sentence) #if answer is None: # return 'Did not get an answer. return answer
def addchatbot_image(): global i image = Label(window, image=chatbot_image) if i < 350: image.place(x=15, y=i) res = chatbot.response(input_box.get("1.0", END)) bot_text = Label(window, anchor=W, text=res, wraplength=270, justify=LEFT) bot_text.place(x=75, y=i + 21) i += 60
def response(): input = request.args.get("q") return chatbot.response(input)
def get_bot_response(): userText = request.args.get('msg') return str(response(userText))
shuffle = False retriever = rr.raw_retriever(saved_checkpoint, tf_responses_file, rnn_dim, raw_responses_file, get_response, encoder, retrieve_batch_size, drop_remainder, shuffle, use_gpu) #准备bert bert_config_file = 'bert/pretrained/chinese_L-12_H-768_A-12/bert_config.json' bert_init_checkpoint = 'bert/ckpts/dnf/o1/model.ckpt-41093' bert_max_seq_length = 256 bert_vocab_file = 'bert/pretrained/chinese_L-12_H-768_A-12/vocab.txt' num_labels = 2 bf = BERTfunc.BERTFunction(bert_config_file, bert_init_checkpoint, bert_max_seq_length, bert_vocab_file, num_labels, use_gpu) judge = chatbot.BERTJudge(bf) default_num_of_candidates = 100 chatbot = chatbot.BERT_chatbot(retriever, default_num_of_candidates, judge) whatYouSay = "" while (True): whatYouSay = input("say something:") if whatYouSay == "goodbye!": break elif whatYouSay == "forget it!": chatbot.forget() continue print("A:" + str(chatbot.response(whatYouSay, 100)))