From 9c1c1c5b3df058bae69b6e0bc25e6c41cc6c81ea Mon Sep 17 00:00:00 2001 From: nickwalton Date: Wed, 30 Oct 2019 17:53:48 -0600 Subject: [PATCH] update --- generator/ctrl/ctrl_generator.py | 3 ++- story/story_manager.py | 2 -- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py index 142b495..973dbe0 100644 --- a/generator/ctrl/ctrl_generator.py +++ b/generator/ctrl/ctrl_generator.py @@ -169,6 +169,7 @@ class CTRLGenerator(): # print("\n\nBEFORE RESULT_REPLACE:") # print(repr(result)) + result = cut_trailing_sentence(result) first_letter_capitalized = result[0].isupper() result = result.replace('."', '".') result = result.replace("#", "") @@ -274,7 +275,7 @@ class CTRLGenerator(): def generate(self, prompt, options=None): prompt = self.prompt_replace(prompt) - debug_print = False + debug_print = True if debug_print: print("\n\n*****DEBUG*****") diff --git a/story/story_manager.py b/story/story_manager.py index 29cdce6..72f5bf0 100644 --- a/story/story_manager.py +++ b/story/story_manager.py @@ -102,7 +102,6 @@ class UnconstrainedStoryManager(StoryManager): def generate_result(self, action): block = self.generator.generate(self.story_context() + action) - block = cut_trailing_sentence(block) return block @@ -199,7 +198,6 @@ class ConstrainedStoryManager(StoryManager): def generate_action_result(self, prompt, phrase, options=None): action = phrase + " " + self.generator.generate(prompt + " " + phrase, options) - action_result = cut_trailing_sentence(action) action, result = split_first_sentence(action_result) return action, result