52 lines
1.3 KiB
Python
52 lines
1.3 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
import time
|
||
|
|
|
||
|
|
from generator.gpt2.gpt2_generator import *
|
||
|
|
from generator.human_dm import *
|
||
|
|
from play import *
|
||
|
|
from story.story_manager import *
|
||
|
|
from story.utils import *
|
||
|
|
|
||
|
|
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
|
||
|
|
|
||
|
|
|
||
|
|
class AIPlayer:
|
||
|
|
def __init__(self, generator):
|
||
|
|
self.generator = generator
|
||
|
|
|
||
|
|
def get_action(self, prompt):
|
||
|
|
return self.generator.generate_raw(prompt)
|
||
|
|
|
||
|
|
|
||
|
|
def play_dm():
|
||
|
|
|
||
|
|
console_print("Initializing AI Dungeon DM Mode")
|
||
|
|
generator = GPT2Generator(temperature=0.9)
|
||
|
|
|
||
|
|
story_manager = UnconstrainedStoryManager(HumanDM())
|
||
|
|
context, prompt = select_game()
|
||
|
|
console_print(context + prompt)
|
||
|
|
story_manager.start_new_story(prompt, context=context, upload_story=False)
|
||
|
|
|
||
|
|
player = AIPlayer(generator)
|
||
|
|
|
||
|
|
while True:
|
||
|
|
action_prompt = story_manager.story_context() + "What do you do next? \n> You"
|
||
|
|
action = player.get_action(action_prompt)
|
||
|
|
print("\n******DEBUG FULL ACTION*******")
|
||
|
|
print(action)
|
||
|
|
print("******END DEBUG******\n")
|
||
|
|
action = action.split("\n")[0]
|
||
|
|
punc = action.rfind(".")
|
||
|
|
if punc > 0:
|
||
|
|
action = action[: punc + 1]
|
||
|
|
shown_action = "> You" + action
|
||
|
|
console_print(second_to_first_person(shown_action))
|
||
|
|
story_manager.act(action)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
play_dm()
|