update

console_play.py

@@ -30,16 +30,16 @@ def play_unconstrained():
 
             action = action[0].upper() + action[1:]
 
-            action = "\n> " + action + ".\n"
+            action = "\n> " + action + "\n"
             action = remove_profanity(action)
             #action = first_to_second_person(action)
         
         result = story_manager.act(action)
-        if "you die" in result or "you are dead" in result:
+        if player_died(result):
             print(result + "\nGAME OVER")
             break
         else:
-            print("\n" + result)
+            print(result)
 
 
 if __name__ == '__main__':

data/scraper/scraper.py

@@ -35,7 +35,28 @@ class Scraper:
                             "You have died.",
                             "You have died",
                             "Epilogue",
-                            "Save Game"}
+                            "Save Game",
+                            "Your quest might have been more successful...",
+                            "5 - not the best, certainly not the worst",
+                            "The End! (leave comments on game)",
+                            "6 - it's worth every cent",
+                            "Quit the game.",
+                            "7 - even better than Reeses' Cups®",
+                            "8 - it will bring you enlightenment",
+                            "End of game! Leave a comment!",
+                            "Better luck next time",
+                            "click here to continue",
+                            "Rating And Leaving Comments",
+                            "Your Outlaw Career has come to an end",
+                            "Thank you for taking the time to read my story",
+                            "You have no further part in the story, End Game and Leave Comments",
+                            '',
+                            "You play no further part in this story. End Game and Leave Comments",
+                            "drivers",
+                            "Alas, poor Yorick, they slew you well",
+                            "My heart bleeds for you",
+                            "To End the Game and Leave Comments click here",
+                            "Call it a day"}
         self.texts = set()
 
     def GoToURL(self, url):
@@ -146,10 +167,59 @@ urls = ["http://chooseyourstory.com/story/viewer/default.aspx?StoryId=10638",
         "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=10359",
         "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=5466",
         "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=28030",
-        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=56515"
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=56515",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=7480",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=7566",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=53134",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=11413",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=470",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=8041",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=22",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=47011",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=45866",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=55043",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=6376",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=36791",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=34193",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=10885",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=7567",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=51926",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=8035",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=8038",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=56742",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=14013",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=53356",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=6382",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=12165",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=49200",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=48393",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=51934",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=42380",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=38774",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=13349",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=8129",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=7426",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=34838",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=54011",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=52961",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=50303",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=60128",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=60232",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=10183",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=23928",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=25548",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=20983",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=17306",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=9918",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=11274",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=58502",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=39055",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=16577",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=52499",
+        "http://chooseyourstory.com/story/viewer/default.aspx?StoryId=29597"
         ]
 
-for i in range(19, len(urls)):
+for i in range(67, len(urls)):
     print("****** Extracting Adventure ", urls[i], " ***********")
     tree = scraper.BuildStoryTree(urls[i])
     save_tree(tree, "stories/story" + str(i) + ".json")

data/scraper/stories/build_training_data.py

@@ -33,16 +33,19 @@ def get_stories(filename):
 
 output_file_path = "text_adventures.txt"
 with open(output_file_path, 'w') as output_file:
-    filenames = ["story" + str(i) + ".json" for i in range(1,20)]
+    filenames = ["story" + str(i) + ".json" for i in range(0,69)]
+    filenames += ["apoc_seed" + str(i) + ".json" for i in range(1,5)]
     stories = []
     for filename in filenames:
         stories += get_stories(filename)
+        print(len(stories))
 
     raw_text = ""
     start_token = "<|startoftext|>"
     end_token = "<|endoftext|>"
     for story in stories:
         raw_text += start_token + story + end_token + "\n"
+        print(len(raw_text))
 
     output_file.write(raw_text)
 

story/utils.py

@@ -33,6 +33,14 @@ def get_action_verbs(key):
     return data_loaded["action_verbs"][key]
 
 
+def player_died(text):
+
+    dead_phrases = ["you died", "you are dead", "You died", "You are dead", "You're dead", "you're dead"]
+    for phrase in dead_phrases:
+        if phrase in text:
+            return True
+
+
 def get_ctrl_verbs(key):
     with open(YAML_FILE, 'r') as stream:
         data_loaded = yaml.safe_load(stream)