def process_input(user_input):
简单的词干提取和匹配
user_input = user_input.lower()
words = nltk.word_tokenize(user_input)
stemmed_words = [stemmer.stem(word) for word in words]
for key in knowledge_base:
if any(word in stemmer.stem(key) for word in stemmed_words):
return knowledge_base[key]
return 对不起,我不懂你的意思。
def chat():
print(欢迎!你想和我聊什么?(输入‘退出’以离开))
while True:
user_input = input(你: )
if user_input.lower() == '退出':
break
print(AI: , process_input(user_input))