-
Notifications
You must be signed in to change notification settings - Fork 0
/
chat.py
89 lines (70 loc) · 2.69 KB
/
chat.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
import random
import json
import torch
from model import NeuralNet
from nltk_utils import bag_of_words, tokenize
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
with open('intents.json', 'r') as json_data:
intents = json.load(json_data)
FILE = "data.pth"
data = torch.load(FILE)
input_size = data["input_size"]
hidden_size = data["hidden_size"]
output_size = data["output_size"]
all_words = data['all_words']
tags = data['tags']
model_state = data["model_state"]
model = NeuralNet(input_size, hidden_size, output_size).to(device)
model.load_state_dict(model_state)
model.eval()
bot_name = "DebbyBot"
print("Welcome to my maternity chatbot (type 'quit' to exit)")
def getResponse(sentence):
query = sentence + " for maternity"
sentence = tokenize(sentence)
X = bag_of_words(sentence, all_words)
X = X.reshape(1, X.shape[0])
X = torch.from_numpy(X).to(device)
output = model(X)
_, predicted = torch.max(output, dim=1)
tag = tags[predicted.item()]
probs = torch.softmax(output, dim=1)
prob = probs[0][predicted.item()]
if prob.item() > 0.75:
for intent in intents['intents']:
if tag == intent["tag"]:
return(f"{random.choice(intent['responses'])}")
else:
from googlesearch import search
resp = search(query, advanced=True, num_results=5)
htmlResponse = "<p class=''>I do not understand. But these resources will be of help</p><ul>"
for i, result in enumerate(resp, start=1):
htmlResponse += """
<li class='card'>
<div class='card-body my-2'>
<p>{}</p>
<a target='_blank' href={}>{}</p>
</div>
</li>
""".format(result.description, result.url, result.title)
htmlResponse += "</ul>"
return(htmlResponse)
# while True:
# sentence = input("You: ")
# if sentence == "quit" or sentence == "exit":
# break
# sentence = tokenize(sentence)
# X = bag_of_words(sentence, all_words)
# X = X.reshape(1, X.shape[0])
# X = torch.from_numpy(X).to(device)
# output = model(X)
# _, predicted = torch.max(output, dim=1)
# tag = tags[predicted.item()]
# probs = torch.softmax(output, dim=1)
# prob = probs[0][predicted.item()]
# if prob.item() > 0.75:
# for intent in intents['intents']:
# if tag == intent["tag"]:
# print(f"{bot_name}: {random.choice(intent['responses'])}")
# else:
# print(f"{bot_name}: I do not understand...")