Develop (#10)

* adds saving and loading

* fixed some cmd issues

generator/gpt2/gpt2_generator.py

@@ -84,15 +84,7 @@ class GPT2Generator:
 
         return result
 
-    def generate(self, prompt, options=None, seed=1):
-
-        debug_print = False
-        prefix = self.prompt_replace(prompt)
-
-        if debug_print:
-            print("******DEBUG******")
-            print("Prompt is: ", repr(prefix))
-
+    def generate_raw(self, prompt):
         context_tokens = self.enc.encode(prompt)
         generated = 0
         for _ in range(self.samples // self.batch_size):
@@ -102,6 +94,19 @@ class GPT2Generator:
             for i in range(self.batch_size):
                 generated += 1
                 text = self.enc.decode(out[i])
+        return text
+
+
+    def generate(self, prompt, options=None, seed=1):
+
+        debug_print = False
+        prompt = self.prompt_replace(prompt)
+
+        if debug_print:
+            print("******DEBUG******")
+            print("Prompt is: ", repr(prompt))
+
+        text = self.generate_raw(prompt)
 
         if debug_print:
             print("Generated result is: ", repr(text))

generator/human_dm.py

@@ -0,0 +1,6 @@
+from story.utils import *
+
+class HumanDM:
+
+    def generate(self, prompt, options=None, seed=None):
+        return input()

play.py

@@ -18,7 +18,7 @@ def select_game():
         else:
             print_str += " (experimental)"
         console_print(print_str)
-    console_print(str(len(settings)) + ") custom expiermental")
+    console_print(str(len(settings)) + ") custom experimental")
     choice = get_num_options(len(settings)+1)
 
     if choice == len(settings):

play_dm.py

@@ -0,0 +1,44 @@
+from story.story_manager import *
+from generator.human_dm import *
+from generator.gpt2.gpt2_generator import *
+from story.utils import *
+from termios import tcflush, TCIFLUSH
+from play import *
+import time, sys, os
+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():
+    generator = GPT2Generator()
+    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]
+        shown_action = "> You" + action
+        console_print(second_to_first_person(shown_action))
+        story_manager.act(action)
+
+
+
+
+if __name__ == '__main__':
+    play_dm()
+
+

story/story_manager.py

@@ -72,7 +72,6 @@ class Story():
         if len(self.results) < 2:
             latest_result += self.story_start
 
-
         while mem_ind > 0:
 
             if len(self.results) >= mem_ind:
@@ -138,22 +137,19 @@ class Story():
     def load_from_storage(self, story_id):
 
         file_name = "story" + story_id + ".json"
-        cmd = "gsutil cp gs://aidungeonstories/" + file_name + " . >/dev/null 2>&1"
+        cmd = "gsutil cp gs://aidungeonstories/" + file_name + " ."
         os.system(cmd)
         exists = os.path.isfile(file_name)
 
-        with open(file_name, 'r') as fp:
-            game = json.load(fp)
-        self.init_from_dict(game)
-
         if exists:
+            with open(file_name, 'r') as fp:
+                game = json.load(fp)
+            self.init_from_dict(game)
             return str(self)
         else:
             return "Error save not found."
 
 
-
-
 class StoryManager():
 
     def __init__(self, generator):
@@ -163,7 +159,7 @@ class StoryManager():
     def start_new_story(self, story_prompt, context="", game_state=None, upload_story=False):
         block = self.generator.generate(context + story_prompt)
         block = cut_trailing_sentence(block)
-        self.story = Story(story_prompt + block, context=context, game_state=game_state, upload_story=upload_story)
+        self.story = Story(context + story_prompt + block, context=context, game_state=game_state, upload_story=upload_story)
         return self.story
     
     def load_story(self, story, from_json=False):