mirror of
https://github.com/wassname/Clover-Edition.git
synced 2026-09-19 12:10:28 +08:00
134 lines
4.2 KiB
Python
134 lines
4.2 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 flask import g
|
|
import os
|
|
|
|
os.environ['GOOGLE_APPLICATION_CREDENTIALS']="./AI-Adventure-2bb65e3a4e2f.json"
|
|
|
|
from google.cloud import storage
|
|
from google import cloud
|
|
import json
|
|
from flask import Flask, render_template, request
|
|
storage_client = storage.Client()
|
|
bucket = storage_client.get_bucket("dungeon-cache")
|
|
from flask import Response
|
|
import requests
|
|
app = Flask(__name__)
|
|
|
|
|
|
# App Info
|
|
gen_ip = "http://35.192.97.36:8010"
|
|
phrases = [" You attack", " You use", " You tell", " You go"]
|
|
prompts = ["You enter a dungeon with your trusty sword and shield. You are searching for the evil necromancer who killed your family. You've heard that he resides at the bottom of the dungeon, guarded by legions of the undead. You enter the first door and see"]
|
|
requested_map = {}
|
|
|
|
@app.route('/')
|
|
def root():
|
|
return render_template('index.html')
|
|
|
|
@app.route('/index.html')
|
|
def index():
|
|
return render_template('index.html')
|
|
|
|
@app.route('/about.html')
|
|
def about():
|
|
return render_template('about.html')
|
|
|
|
|
|
def cache_file(seed, prompt_num, choices, response, tag):
|
|
|
|
blob_file_name = "p" + str(prompt_num) + "/seed" + str(seed) + "/" + tag
|
|
for action in choices:
|
|
blob_file_name = blob_file_name + str(action)
|
|
blob = bucket.blob(blob_file_name)
|
|
|
|
blob.upload_from_string(response)
|
|
|
|
print("File ", blob_file_name, " cached")
|
|
|
|
|
|
def retrieve_from_cache(seed, prompt_num, choices, tag):
|
|
blob_file_name = "p" + str(prompt_num) + "/seed" + str(seed) + "/" + tag
|
|
|
|
for action in choices:
|
|
blob_file_name = blob_file_name + str(action)
|
|
|
|
blob = bucket.blob(blob_file_name)
|
|
|
|
if blob.exists(storage_client):
|
|
result = blob.download_as_string().decode("utf-8")
|
|
print(blob_file_name, " found in cache")
|
|
else:
|
|
result = None
|
|
print(blob_file_name, " not found in cache")
|
|
|
|
return result
|
|
|
|
|
|
@app.route('/generate', methods=['POST'])
|
|
def story_request():
|
|
print("****Generating Story****")
|
|
seed = request.form["seed"]
|
|
prompt_num = int(request.form["prompt_num"])
|
|
gen_actions = request.form["actions"]
|
|
|
|
if gen_actions == "true":
|
|
|
|
prompt = request.form["prompt"]
|
|
choices = json.loads(request.form["choices"])
|
|
print("Getting response for seed ", seed, " prompt_num ", prompt_num, " and choices ", choices)
|
|
|
|
action_results = retrieve_from_cache(seed, prompt_num, choices, "choices")
|
|
|
|
if action_results is not None:
|
|
response = action_results
|
|
else:
|
|
response = requests.post(gen_ip + "/generate",
|
|
data={"actions":"true","seed":seed, "prompt_num":prompt_num, "prompt": prompt, "choices": json.dumps(choices)})
|
|
response = response.text
|
|
cache_file(seed, prompt_num, choices, response, "choices")
|
|
else:
|
|
|
|
print("Getting response for seed ", seed, " prompt_num ", prompt_num)
|
|
result = retrieve_from_cache(seed, prompt_num, [], "story")
|
|
|
|
if result is not None:
|
|
response = result
|
|
else:
|
|
response = requests.post(gen_ip + "/generate", data={"actions":"false","seed":seed, "prompt_num":prompt_num})
|
|
response=response.text
|
|
cache_file(seed, prompt_num, [], response, "story")
|
|
|
|
print("\nGenerated response is: \n", response)
|
|
print("")
|
|
|
|
return response
|
|
|
|
@app.teardown_appcontext
|
|
def teardown_sess(_):
|
|
sess = g.pop("sess",None)
|
|
|
|
if sess is not None:
|
|
sess.close()
|
|
|
|
if __name__ == '__main__':
|
|
app.run(host='0.0.0.0', port=8080)
|
|
|
|
|
|
|
|
# [START gae_python37_render_template]
|