diff --git a/console_play.py b/console_play.py index 07eb74c..78aec3d 100644 --- a/console_play.py +++ b/console_play.py @@ -1,7 +1,7 @@ from story.story_manager import * from generator.gpt2.gpt2_generator import * -import os -os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3' + + def console_print(text, width=75): last_newline = 0 @@ -17,12 +17,20 @@ def console_print(text, width=75): def play_unconstrained(): + + print("Initializing AI Dungeon! (This might take a few minutes)") generator = GPT2Generator() prompt = get_story_start("knight") context = get_context("knight") story_manager = UnconstrainedStoryManager(generator) + print("Generating initial story.") story_manager.start_new_story(prompt, context=context) + with open('opening.txt', 'r') as file: + starter = file.read() + + print(starter) + print("\n") console_print(context) console_print(str(story_manager.story)) diff --git a/opening.txt b/opening.txt new file mode 100644 index 0000000..c6fc636 --- /dev/null +++ b/opening.txt @@ -0,0 +1,11 @@ + + ▄▄▄ ██▓ ▓█████▄ █ ██ ███▄ █ ▄████ ▓█████ ▒█████ ███▄ █ +▒████▄ ▓██▒ ▒██▀ ██▌ ██ ▓██▒ ██ ▀█ █ ██▒ ▀█▒▓█ ▀ ▒██▒ ██▒ ██ ▀█ █ +▒██ ▀█▄ ▒██▒ ░██ █▌▓██ ▒██░▓██ ▀█ ██▒▒██░▄▄▄░▒███ ▒██░ ██▒▓██ ▀█ ██▒ +░██▄▄▄▄██ ░██░ ░▓█▄ ▌▓▓█ ░██░▓██▒ ▐▌██▒░▓█ ██▓▒▓█ ▄ ▒██ ██░▓██▒ ▐▌██▒ + ▓█ ▓██▒░██░ ░▒████▓ ▒▒█████▓ ▒██░ ▓██░░▒▓███▀▒░▒████▒░ ████▓▒░▒██░ ▓██░ + ▒▒ ▓▒█░░▓ ▒▒▓ ▒ ░▒▓▒ ▒ ▒ ░ ▒░ ▒ ▒ ░▒ ▒ ░░ ▒░ ░░ ▒░▒░▒░ ░ ▒░ ▒ ▒ + ▒ ▒▒ ░ ▒ ░ ░ ▒ ▒ ░░▒░ ░ ░ ░ ░░ ░ ▒░ ░ ░ ░ ░ ░ ░ ▒ ▒░ ░ ░░ ░ ▒░ + ░ ▒ ▒ ░ ░ ░ ░ ░░░ ░ ░ ░ ░ ░ ░ ░ ░ ░ ░ ░ ░ ▒ ░ ░ ░ + ░ ░ ░ ░ ░ ░ ░ ░ ░ ░ ░ ░ + ░