diff --git a/doc/core-analyze.PNG b/doc/core-analyze.PNG index c53341b..fdb8e8e 100644 Binary files a/doc/core-analyze.PNG and b/doc/core-analyze.PNG differ diff --git a/doc/core-chat.PNG b/doc/core-chat.PNG index 2963e3c..50fade3 100644 Binary files a/doc/core-chat.PNG and b/doc/core-chat.PNG differ diff --git a/doc/core-conversations.PNG b/doc/core-conversations.PNG index 8555998..2c52e0b 100644 Binary files a/doc/core-conversations.PNG and b/doc/core-conversations.PNG differ diff --git a/doc/core-domain_file.PNG b/doc/core-domain_file.PNG index b2533fc..7e01cc7 100644 Binary files a/doc/core-domain_file.PNG and b/doc/core-domain_file.PNG differ diff --git a/doc/core-memoization_policy.PNG b/doc/core-memoization_policy.PNG new file mode 100644 index 0000000..54830e9 Binary files /dev/null and b/doc/core-memoization_policy.PNG differ diff --git a/doc/core-stories_files.PNG b/doc/core-stories_files.PNG index a4baf14..050b680 100644 Binary files a/doc/core-stories_files.PNG and b/doc/core-stories_files.PNG differ diff --git a/doc/nlu-md_files.PNG b/doc/nlu-md_files.PNG index 2c5d27e..959dcdb 100644 Binary files a/doc/nlu-md_files.PNG and b/doc/nlu-md_files.PNG differ diff --git a/src/RasaHost/RasaHost/__init__.py b/src/RasaHost/RasaHost/__init__.py index 3e85673..f592cb4 100644 --- a/src/RasaHost/RasaHost/__init__.py +++ b/src/RasaHost/RasaHost/__init__.py @@ -17,6 +17,7 @@ class RasaHost: nlu_path = os.path.join(current_dir, "data/nlu/") stories_path = os.path.join(current_dir, "data/stories/") domain_path = os.path.join(current_dir, "data/domain.yml") + memoization_policy_path = os.path.join(current_dir, "models/current/dialogue/policy_1_MemoizationPolicy") host = environ.get('SERVER_HOST', '0.0.0.0') port = 5555 agent = None diff --git a/src/RasaHost/RasaHost/controllers/models_controller.py b/src/RasaHost/RasaHost/controllers/models_controller.py index c3dc3da..c7e351c 100644 --- a/src/RasaHost/RasaHost/controllers/models_controller.py +++ b/src/RasaHost/RasaHost/controllers/models_controller.py @@ -14,6 +14,7 @@ @app.route('/models/MemoizationPolicy') def models_memoization_policy(): + MemoizationPolicyService(host.memoization_policy_path).decode() return render_template( 'models/memoization_policy.html', title='MemoizationPolicy', @@ -21,7 +22,7 @@ def models_memoization_policy(): @app.route('/api/models/MemoizationPolicy/lookups/decode', methods=['POST']) def api_models_memoization_policy_lookups_decode(): - MemoizationPolicyService('C:\\Projects\\GSA\\Investec.Bot\\Investec.Bot\\models\\current\\dialogue\\policy_1_MemoizationPolicy').decode() + MemoizationPolicyService(host.memoization_policy_path).decode() return jsonify({"result":"ok"}) @@ -29,5 +30,5 @@ def api_models_memoization_policy_lookups_decode(): def api_models_memoization_policy_lookups(): query = request.args.get('q') last_index = int(request.args.get('i', 0)) - result = MemoizationPolicyService('C:\\Projects\\GSA\\Investec.Bot\\Investec.Bot\\models\\current\\dialogue\\policy_1_MemoizationPolicy').find(query, last_index) + result = MemoizationPolicyService(host.memoization_policy_path).find(query, last_index) return jsonify({"result": result}) diff --git a/src/RasaHost/RasaHost/services/memoization_policy_service.py b/src/RasaHost/RasaHost/services/memoization_policy_service.py index c631f68..5c58cef 100644 --- a/src/RasaHost/RasaHost/services/memoization_policy_service.py +++ b/src/RasaHost/RasaHost/services/memoization_policy_service.py @@ -11,12 +11,16 @@ class MemoizationPolicyService(object): def __init__(self, path = None, *args, **kwargs): self.path = path + self.memorized_turns_path = os.path.join(self.path, 'memorized_turns.json') + self.memorized_turns_lookups_path = os.path.join(self.path, 'memorized_turns_lookups.txt') pass def decode(self): - with open(os.path.join(self.path, 'memorized_turns.json')) as f_encoded: + if not os.path.exists(self.memorized_turns_path): + return + with open(self.memorized_turns_path) as f_encoded: memorized_turns = json.load(f_encoded) - with open(os.path.join(self.path, 'memorized_turns_lookups.txt'), 'a+') as f_decoded: + with open(self.memorized_turns_lookups_path, 'a+') as f_decoded: for lookup in memorized_turns["lookup"]: b64decoded = base64.b64decode(lookup) decompressed = zlib.decompress(bytes(b64decoded)) @@ -24,9 +28,11 @@ def decode(self): f_decoded.write(text + "\n") def find(self, query, last_index): + if not os.path.exists(self.memorized_turns_lookups_path): + return [] lookups = [] current_last_index = 0 - with open(os.path.join(self.path, 'memorized_turns_lookups.txt'), 'r') as f_decoded: + with open(self.memorized_turns_lookups_path, 'r') as f_decoded: for index, line in enumerate(f_decoded): current_last_index = index if current_last_index > last_index: @@ -44,7 +50,7 @@ def parse_lookup(self, lookup): def parse(self): lookups = [] - with open(os.path.join(self.path, 'memorized_turns.json')) as f_encoded: + with open(self.memorized_turns_path) as f_encoded: memorized_turns = json.load(f_encoded) with Pool() as p: lookups = p.map(parse_lookup, memorized_turns["lookup"]) diff --git a/src/RasaHost/RasaHost/templates/layout.html b/src/RasaHost/RasaHost/templates/layout.html index 8ccbe4c..2a7e8c9 100644 --- a/src/RasaHost/RasaHost/templates/layout.html +++ b/src/RasaHost/RasaHost/templates/layout.html @@ -93,6 +93,11 @@