def find(self):
db = DB()
db.init()
if db.find_one(self.collection, {"intent": self.intent}):
return True
else:
return False
def find_response(self):
db = DB()
db.init()
if self.find:
get = db.find_one(self.collection, {"intent": self.intent,
"entity": self.entity})
return get['response']
else:
return {"message": "Object not found."}
def find_response(self):
db = DB()
db.init()
if self.intent == "admissibilite_congé":
if self.find:
get = db.find_one(self.collection, {"intent": self.intent})
return random.choice(get['response'])
else:
return {"message": "Object not found."}
else:
if self.find:
get = db.find_one(self.collection, {"intent": self.intent})
return get['response']
else:
return {"message": "Object not found."}
def post():
global entitiz, nlu2, nlu
# -- NLP treatment --
# Input body composition
data = request.get_json()
body = dict({
'query': data['msg'],
'nlp-address': data['nlp-url'],
'collection-name': data['collection']
})
# NLU post request
intent = requests.post(url=body['nlp-address'],
data=json.dumps(dict({'query': body['query']})))
entity = requests.post(url=body['nlp-address'],
data=json.dumps(dict({'query': body['query']})))
text = requests.post(url=body['nlp-address'],
data=json.dumps(dict({'query': body['query']})))
date = intent.headers['Date']
# Database class using
db = DB()
db.Collection = body['collection-name']
db.init()
# Extracting Intent and Entity
intent = intent.json()['intent']['name']
intent_confidence = entity.json()['intent']['confidence']
entity_array = entity.json()['entities']
# Implement entity array into list
length = len(entity_array)
i = 0
entitiz = list()
while i != length:
entitiz.append(entity_array[i]['value'])
i = i + 1
text = text.json()['text']
# Checking if the entity is none
if not is_empty(entitiz):
nlu2 = NluObjectWithEntity(body['collection-name'],
intent,
entitiz[0],
response=None)
nlu = NluObjectSimple(body['collection-name'], intent, response=None)
# Saving request informations
saves = dict(date=date,
text=text,
intent=intent,
intent_confidence=intent_confidence,
entities=entity_array)
save(saves)
# Sending Json Object with Headers & Response
if nlu.find:
json_object = dict({
'success':
'Message sent with success',
'text':
text,
'msg-type':
'text',
'bot':
dict(response=nlu.find_response
if is_empty(entitiz) else nlu2.find_response,
quickReplies=True,
quicks=["congé", "contrats", "obligation patronale"])
})
return json_object, 200, {
'Access-Control-Allow-Origin': '*',
'Content-Type': 'application/json'
}
else:
json_object = {
'response': 'Pouvez-vous mieux préciser ce vous voulez dire ??'
}
return json_object, 400, {'Content-Type': 'application/json'}