Properly indented play.py with the set temperature command - VBPXKSMI/Wiki-Test GitHub Wiki
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()