diff --git a/console_play.py b/console_play.py index 4deccf5..8d53d6f 100644 --- a/console_play.py +++ b/console_play.py @@ -23,7 +23,7 @@ def console_print(str, pycharm=False): def play_unconstrained(): generator = CTRLGenerator() #generator = WebGenerator(CRED_FILE) - prompt = get_story_start("hospital") + prompt = get_story_start("peasant") story_manager = UnconstrainedStoryManager(generator) story_manager.start_new_story(prompt) diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py index aa8cbbf..33dd72f 100644 --- a/generator/ctrl/ctrl_generator.py +++ b/generator/ctrl/ctrl_generator.py @@ -217,7 +217,7 @@ class CTRLGenerator(): # disallow some tokens forbidden_tokens = ['', 'Sco@@', "&@@", "1]@@", "2]@@", "3]@@", "4]@@", "https://www.@@", "[@@", ":@@", - "Edit", "&@@", "2:","1:", ":", "Edit@@", "EDI@@", "EDIT@@", "edit", "TL@@"] + "Edit", "&@@", "2:","1:", ":", "Edit@@", "EDI@@", "EDIT@@", "edit", "TL@@", "tl@@", ";@@", '**'] if num_new_lines > self.max_new_lines: forbidden_tokens.append("\n") diff --git a/story/story_data.yaml b/story/story_data.yaml index c697616..d49c052 100644 --- a/story/story_data.yaml +++ b/story/story_data.yaml @@ -3,6 +3,8 @@ prompts: hospital: "You wake up in an old rundown hospital with no memory of how you got there. You look around and see " + peasant: "You are a medieval peasant. You just finished the days work in the wheat fields. You get home to your thatched hut and see " + alien: "You've crash landed on an alien planet. You don't know if you can get your ship working. You get out to explore your surroundings and see " action_verbs: