From 2a98952b23a86b575f6ed886ccc19daa6929b227 Mon Sep 17 00:00:00 2001
From: Nick Walton <nickwalton00@gmail.com>
Date: Mon, 18 Nov 2019 13:28:24 -0700
Subject: [PATCH] updated main.py

---
 Dockerfile       | 14 +++++++++
 install_model.sh |  1 +
 main.py          | 74 +++++++++++++++++++++++++++++++-----------------
 start.sh         |  1 +
 4 files changed, 64 insertions(+), 26 deletions(-)
 create mode 100644 Dockerfile
 create mode 100755 install_model.sh
 create mode 100644 start.sh

diff --git a/Dockerfile b/Dockerfile
new file mode 100644
index 0000000..1426756
--- /dev/null
+++ b/Dockerfile
@@ -0,0 +1,14 @@
+FROM nvcr.io/nvidia/tensorflow:19.08-py3
+
+ENV APP_HOME /app
+WORKDIR $APP_HOME
+COPY . ./
+
+RUN pip install -r requirements.txt
+WORKDIR generator/ctrl
+RUN ls
+RUN sh install_ctrl_py3.sh
+RUN sh download_model.sh
+WORKDIR $APP_HOME
+
+CMD exec guincorn --bind :%PORT --workers 1 --threads 8 app:app
diff --git a/install_model.sh b/install_model.sh
new file mode 100755
index 0000000..4922b61
--- /dev/null
+++ b/install_model.sh
@@ -0,0 +1 @@
+gsutil -m cp -r gs://aidungeon2model/model_v1 ./generator/gpt2/models
diff --git a/main.py b/main.py
index b3ac428..2d97d45 100644
--- a/main.py
+++ b/main.py
@@ -5,24 +5,43 @@ from story.utils import *
 import json
 from flask import Flask, render_template, request, abort
 from story.story_manager import *
-from generator.web.web_generator import *
-from other.cacher import *
 import numpy as np
+from story.story_manager import *
+from generator.gpt2.gpt2_generator import *
+
+
+"""
+generator = GPT2Generator()
+    prompt = get_story_start("knight")
+    context = get_context("knight")
+    story_manager = UnconstrainedStoryManager(generator)
+    story_manager.start_new_story(prompt, context=context)
+
+    print("\n")
+    print(context)
+    print(str(story_manager.story))
+    while True:
+        action = input("> ")
+
+        if action != "":
+            action = action.strip()
+
+            action = action[0].upper() + action[1:]
+
+            action = "\n> " + action + "\n"
+            action = remove_profanity(action)
+            #action = first_to_second_person(action)
+        
+        result = story_manager.act(action)
+        if player_died(result):
+            print(result + "\nGAME OVER")
+            break
+        else:
+            print(result)"""
 
 app = Flask(__name__)
-app.secret_key = '#d\xe0\xd1\xfb\xee\xa4\xbb\xd0\xf0/e)\xb5g\xdd<`\xc7\xa5\xb0-\xb8d0S'
-CRED_FILE = "./AI-Adventure-2bb65e3a4e2f.json"
-generator = WebGenerator(CRED_FILE)
-story_manager = ConstrainedStoryManager(generator)
-
-def get_response_string(story_text, possible_actions):
-    string_list = ["\n\n", story_text, "\n\nOptions:" + "\n"]
-    for i, action in enumerate(possible_actions):
-        string_list.append(str(i) + ") " + action + "\n")
-    string_list.append("\nWhich action do you choose? ")
-
-    response = "".join(string_list)
-    return response
+generator = GPT2Generator()
+story_manager = UnconstrainedStoryManager(generator)
 
 # Shows about. (Should also link to paper when published)
 @app.route('/about.html')
@@ -37,12 +56,10 @@ def generate():
     # If there is no story in session, make a new one
     if "story" not in session or session["story"] is None:
         print("Starting new story")
-        seed = np.random.randint(100)
-        story_manager.enable_caching(credentials_file=CRED_FILE, seed=seed, bucket_name="dungeon-cache")
-        prompt = get_story_start("classic")
-        story_manager.start_new_story(prompt, seed)
-        possible_actions = story_manager.get_possible_actions()
-        response = get_response_string(str(story_manager.story), possible_actions)
+        prompt = get_story_start("knight")
+        context = get_context("knight")
+        story_manager.start_new_story(prompt, context=context)
+        response = context + str(story_manager.story)
 
     # If there is a story in session continue from it.
     else:
@@ -50,11 +67,15 @@ def generate():
         story = session["story"]
         story_manager.load_story(story, from_json=True)
 
-        result, possible_actions = story_manager.act(action)
-        if result is None:
-            response = "\nInvalid choice. Must be a number from 0 to 3. \n" + "\nWhich action do you choose? "
-        else:
-            response = get_response_string(result, possible_actions)
+        if action != "":
+            action = action.strip()
+
+            action = action[0].upper() + action[1:]
+
+            action = "\n> " + action + "\n"
+            #action = remove_profanity(action)
+
+        response = story_manager.act(action)
 
     session["story"] = story_manager.json_story()
     print("Returning response")
@@ -66,5 +87,6 @@ def root():
     session["story"] = None
     return render_template('index.html')
 
+
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=8080)
\ No newline at end of file
diff --git a/start.sh b/start.sh
new file mode 100644
index 0000000..e4e6e3d
--- /dev/null
+++ b/start.sh
@@ -0,0 +1 @@
+./gunicorn --timeout=300 --graceful-timeout=300 -b 0.0.0.0:8010 main:app