View difference between Paste ID: AMYbspLN and CA8WZZuR

 from story.story_manager import *
 from generator.gpt2.gpt2_generator import *
 from story.utils import *
 import time, sys, os
 os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
 
 def flush_input():
     import msvcrt
     while msvcrt.kbhit():
         msvcrt.getch()
 
 def select_game():
     with open(YAML_FILE, 'r') as stream:
         data = yaml.safe_load(stream)
 
     print("Pick a setting.")
     settings = data["settings"].keys()
     for i, setting in enumerate(settings):
         print_str = str(i) + ") " + setting
         if setting == "fantasy":
             print_str += " (recommended)"
 
         console_print(print_str)
     console_print(str(len(settings)) + ") custom")
     choice = get_num_options(len(settings)+1)
 
     if choice == len(settings):
 
         context = ""
         console_print("\nEnter a prompt that describes who you are and the first couple sentences of where you start "
                       "out ex:\n 'You are a knight in the kingdom of Larion. You are hunting the evil dragon who has been " +
                       "terrorizing the kingdom. You enter the forest searching for the dragon and see' ")
         prompt = input("Starting Prompt: ")
         return context, prompt
 
     setting_key = list(settings)[choice]
 
     print("\nPick a character")
     characters = data["settings"][setting_key]["characters"]
     for i, character in enumerate(characters):
         console_print(str(i) + ") " + character)
     character_key = list(characters)[get_num_options(len(characters))]
 
     name = input("\nWhat is your name? ")
     setting_description = data["settings"][setting_key]["description"]
     character = data["settings"][setting_key]["characters"][character_key]
 
     context = "You are " + name + ", a " + character_key + " " + setting_description + \
               "You have a " + character["item1"] + " and a " + character["item2"] + ". "
     prompt_num = np.random.randint(0, len(character["prompts"]))
     prompt = character["prompts"][prompt_num]
 
     return context, prompt
 
 
 def instructions():
     text = "\nAI Dungeon 2 Instructions:"
     text += '\n Enter actions starting with a verb ex. "go to the tavern" or "attack the orc."'
     text += '\n To speak enter \'say "(thing you want to say)"\' or just "(thing you want to say)" '
     text += '\n\nThe following commands can be entered for any action: '
     text += '\n  "revert"   Reverts the last action allowing you to pick a different action.'
     text += '\n  "quit"     Quits the game and saves'
     text += '\n  "restart"  Starts a new game and saves your current one'
     text += '\n  "save"     Makes a new save of your game and gives you the save ID'
     text += '\n  "load"     Asks for a save ID and loads the game if the ID is valid'
     text += '\n  "print"    Prints a transcript of your adventure (without extra newline formatting)'
     text += '\n  "settemp" Changes the AI temperature, default is now 0.15, previously 0.4. Going lower will make AI less random.'
     text += '\n  "help"     Prints these instructions again'
     return text
 
 def play_aidungeon_2():
 
     console_print("AI Dungeon 2 will save and use your actions and game to continually improve AI Dungeon."
                   + " If you would like to disable this enter 'nosaving' for any action. This will also turn off the "
                   + "ability to save games.")
 
     upload_story = True
 
     print("\nInitializing AI Dungeon! (This might take a few minutes)\n")
     generator = GPT2Generator()
     story_manager = UnconstrainedStoryManager(generator)
     print("\n")
 
     #with open('opening.txt', 'r') as file:
     #    starter = file.read()
     #print(starter)
 
     while True:
         if story_manager.story != None:
             del story_manager.story
 
         print("\n\n")
         context, prompt = select_game()
         console_print(instructions())
         print("\nGenerating story...")
 
         story_manager.start_new_story(prompt, context=context, upload_story=upload_story)
 
         print("\n")
         console_print(str(story_manager.story))
         while True:
             flush_input()
             action = input("> ")
             if action == "restart":
                 rating = input("Please rate the story quality from 1-10: ")
                 rating_float = float(rating)
                 story_manager.story.rating = rating_float
                 break
 
             elif action == "quit":
                 rating = input("Please rate the story quality from 1-10: ")
                 rating_float = float(rating)
                 story_manager.story.rating = rating_float
                 exit()
 
             elif action == "nosaving":
                 upload_story = False
                 story_manager.story.upload_story = False
                 console_print("Saving turned off.")
 
             elif action == "help":
                 console_print(instructions())
 
             elif action == "save":
                 if upload_story:
                     id = story_manager.story.save_to_storage()
                     console_print("Game saved.")
                     console_print("To load the game, type 'load' and enter the following ID: " + id)
                 else:
                     console_print("Saving has been turned off. Cannot save.")
 
             elif action =="load":
                 load_ID = input("What is the ID of the saved game?")
                 result = story_manager.story.load_from_storage(load_ID)
                 console_print("\nLoading Game...\n")
                 console_print(result)
 
             elif len(action.split(" ")) == 2 and action.split(" ")[0] == "load":
                 load_ID = action.split(" ")[1]
                 result = story_manager.story.load_from_storage(load_ID)
                 console_print("\nLoading Game...\n")
                 console_print(result)
 
             elif action == "print":
                 print("\nPRINTING\n")
                 if line_break == "n":
                     print(str(story_manager.story))
                 elif line_break == "y":
                     console_print(str(story_manager.story))
                 elif line_break.isdigit():
                     console_print(str(story_manager.story), int(line_break))
 
             elif action == "settemp":
                 temp = float(input("Set a new temperature>"))
                 if type(temp)==int or float:
                     story_manager.generator = GPT2Generator(temperature=temp)
                 else:
                     print("\nEnter a number, please")
                     print(str(story_manager.story))
 
             elif action == "revert":
 
                 if len(story_manager.story.actions) is 0:
                     console_print("You can't go back any farther. ")
                     continue
 
                 story_manager.story.actions = story_manager.story.actions[:-1]
                 story_manager.story.results = story_manager.story.results[:-1]
                 console_print("Last action reverted. ")
                 if len(story_manager.story.results) > 0:
                     console_print(story_manager.story.results[-1])
                 else:
                     console_print(story_manager.story.story_start)
                 continue
 
             else:
                 if action == "":
                     action = ""
                     result = story_manager.act(action)
                     console_print(result)
 
                 elif action[0] == '"':
                     action = "You say " + action
 
                 else:
                     action = action.strip()
                     action = action[0].lower() + action[1:]
 
                     if "You" not in action[:6] and "I" not in action[:6]:
                         action = "You " + action
 
                     if action[-1] not in [".", "?", "!"]:
                         action = action + "."
 
                     action = first_to_second_person(action)
 
                     action = "\n> " + action + "\n"
 
                 result = "\n" + story_manager.act(action)
                 if len(story_manager.story.results) >= 2:
                     similarity = get_similarity(story_manager.story.results[-1], story_manager.story.results[-2])
                     if similarity > 0.9:
                         story_manager.story.actions = story_manager.story.actions[:-1]
                         story_manager.story.results = story_manager.story.results[:-1]
                         console_print("Woops that action caused the model to start looping. Try a different action to prevent that.")
                         continue
 
                 if player_won(result):
                     console_print(result + "\n CONGRATS YOU WIN")
                     break
                 elif player_died(result):
                     console_print(result)
                     console_print("YOU DIED. GAME OVER")
                     console_print("\nOptions:")
                     console_print('0) Start a new game')
                     console_print('1) "I\'m not dead yet!" (If you didn\'t actually die) ')
                     console_print('Which do you choose? ')
                     choice = get_num_options(2)
                     if choice == 0:
                         break
                     else:
                         console_print("Sorry about that...where were we?")
                         console_print(result)
 
                 else:
                     console_print(result)
 
 
 if __name__ == '__main__':
     play_aidungeon_2()