apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

feat(api): add endpoints to serve GUI bundle

This commit is contained in:
Sean Sube 2023-01-12 22:54:32 -06:00
parent 3c0d854834
commit db06754c90
1 changed files with 16 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
api/onnx_web/serve.py
View File

@ -48,9 +48,13 @@ max_height = 512
max_width = 512
# paths
bundle_path = environ.get('ONNX_WEB_BUNDLE_PATH', path.join('..', '..', 'gui', 'out'))
# paths used for Flask files must have ../..
# paths used for fopen only need ../
bundle_path = environ.get('ONNX_WEB_BUNDLE_PATH',
path.join('..', '..', 'gui', 'out'))
model_path = environ.get('ONNX_WEB_MODEL_PATH', path.join('..', 'models'))
output_path = environ.get('ONNX_WEB_OUTPUT_PATH', path.join('..', 'outputs'))
output_path = environ.get('ONNX_WEB_OUTPUT_PATH',
path.join('..', '..', 'outputs'))
params_path = environ.get('ONNX_WEB_PARAMS_PATH', 'params.json')
@ -150,6 +154,10 @@ def json_with_cors(data, origin='*'):
return res
def serve_bundle_file(filename='index.html'):
return send_from_directory(bundle_path, filename)
def make_output_path(mode: str, seed: int, params: Tuple[Union[str, int, float]]):
sha = sha256()
sha.update(mode.encode('utf-8'))
@ -315,20 +323,15 @@ executor = Executor(app)
# routes
def serve_file(filename = 'index.html'):
file = path.join(bundle_path, filename)
print('index', file)
return send_file(file)
@app.route('/')
def index():
return serve_file()
return serve_bundle_file()
@app.route('/<path:filename>')
def index_path(filename):
return serve_file(filename)
return serve_bundle_file(filename)
@app.route('/api')
@ -474,4 +477,4 @@ def ready():
@app.route('/api/output/<path:filename>')
def output(filename: str):
return send_from_directory(path.join('..', output_path), filename, as_attachment=False)
return send_from_directory(output_path, filename, as_attachment=False)