mirror of
https://github.com/wassname/Clover-Edition.git
synced 2026-09-20 12:20:53 +08:00
91 lines
2.7 KiB
Python
91 lines
2.7 KiB
Python
# Copyright 2018 Google LLC
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
# [START gae_python37_render_template]
|
|
import datetime
|
|
from generator import StoryGenerator
|
|
import tensorflow as tf
|
|
from flask import g
|
|
import pdb
|
|
|
|
import json
|
|
from flask import Flask, render_template, request
|
|
|
|
app = Flask(__name__)
|
|
|
|
session = None
|
|
generator = None
|
|
|
|
@app.route('/')
|
|
def root():
|
|
# For the sake of example, use static information to inflate the template.
|
|
# This will be replaced with real information in later steps.
|
|
return render_template('index.html')
|
|
|
|
|
|
@app.route('/generate', methods=['POST'])
|
|
def story_request():
|
|
print("****Generating Story****")
|
|
prompt = request.form["prompt"] # given prompt
|
|
phrase = request.form["phrase"]
|
|
gen_actions = request.form["actions"] # given prompt
|
|
|
|
print("\n Given prompt is: \n",prompt," ", phrase,"\n")
|
|
|
|
generator = get_generator()
|
|
|
|
if gen_actions == "true":
|
|
action, result = generator.generate_action_result(prompt, phrase)
|
|
response = json.dumps([action, result])
|
|
else:
|
|
response = generator.generate_story_block(prompt)
|
|
|
|
print("\nGenerated response is: \n", response)
|
|
print("")
|
|
|
|
return response
|
|
|
|
|
|
def get_generator():
|
|
if "gen" not in g:
|
|
if "sess" not in g:
|
|
g.sess = tf.Session()
|
|
g.gen = StoryGenerator(g.sess)
|
|
|
|
return g.gen
|
|
|
|
@app.teardown_appcontext
|
|
def teardown_sess(_):
|
|
sess = g.pop("sess",None)
|
|
|
|
if sess is not None:
|
|
sess.close()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
# This is used when running locally only. When deploying to Google App
|
|
# Engine, a webserver process such as Gunicorn will serve the app. This
|
|
# can be configured by adding an `entrypoint` to app.yaml.
|
|
# Flask's development server will automatically serve static files in
|
|
# the "static" directory. See:
|
|
# http://flask.pocoo.org/docs/1.0/quickstart/#static-files. Once deployed,
|
|
# App Engine itself will serve those files as configured in app.yaml.
|
|
with tf.Session(graph=tf.Graph()) as sess:
|
|
app.run(host='127.0.0.1', port=8090, debug=False)
|
|
|
|
|
|
|
|
|
|
# [START gae_python37_render_template]
|