From 27c6e2e1ac75c25c52a467d2e4e5c40eed699393 Mon Sep 17 00:00:00 2001 From: Nick Date: Thu, 19 Sep 2019 11:28:13 -0600 Subject: [PATCH] update --- console_play.py | 1 + generator/ctrl/ctrl_generator.py | 2 -- story/story_manager.py | 3 ++- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/console_play.py b/console_play.py index dcd50da..609e665 100644 --- a/console_play.py +++ b/console_play.py @@ -20,6 +20,7 @@ def console_print(str): def play_unconstrained(): generator = CTRLGenerator() + #generator = WebGenerator(CRED_FILE) story_manager = UnconstrainedStoryManager(generator, prompt) console_print(str(story_manager.story)) diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py index 96e8e91..eda8a25 100644 --- a/generator/ctrl/ctrl_generator.py +++ b/generator/ctrl/ctrl_generator.py @@ -254,8 +254,6 @@ class CTRLGenerator(): tokens_generated_so_far = re.sub('(@@ )', '', string=tokens_generated_so_far) tokens_generated_so_far = re.sub('(@@ ?$)', '', string=tokens_generated_so_far) - print(tokens_generated_so_far) - result = tokens_generated_so_far[prompt_length:] return result diff --git a/story/story_manager.py b/story/story_manager.py index 9684bc0..301d3aa 100644 --- a/story/story_manager.py +++ b/story/story_manager.py @@ -29,7 +29,7 @@ class Story(): if len(self.results) > 0: return self.results[-1] else: - return "" + return self.story_start def __str__(self): story_list = [self.story_start] @@ -71,6 +71,7 @@ class UnconstrainedStoryManager(StoryManager): return result def generate_result(self, action): + print("Story context is ", self.story_context()) block = self.generator.generate(self.story_context() + action) block = cut_trailing_sentence(block) block = story_replace(block)