Skip to content

Instantly share code, notes, and snippets.

@zhanwenchen
Created January 9, 2019 00:35
Show Gist options
  • Save zhanwenchen/677f909681f535c1364eb2b97d2829fa to your computer and use it in GitHub Desktop.
Save zhanwenchen/677f909681f535c1364eb2b97d2829fa to your computer and use it in GitHub Desktop.
A Flask web app that uses the Server class defined in translate_client.py
# -*- coding: utf-8 -*-
import os
import sys
import logging
from flask import Flask, request, jsonify
from flask_cors import CORS, cross_origin
from translate_client import Server
# IMPORTANT: DO NOT USE 'localhost' because AWS uses localhost/127.0.0.1
# for private internal addressing so you might have access problems
HOST='0.0.0.0'
# define the app
app = Flask(__name__)
CORS(app) # needed for cross-domain requests, allow everything by default
# logging for heroku
if 'DYNO' in os.environ:
app.logger.addHandler(logging.StreamHandler(sys.stdout))
app.logger.setLevel(logging.INFO)
# load the model
server = Server(HOST, 9000)
# API route
@app.route('/api', methods=['POST'])
@cross_origin()
def api():
"""API function
All model-specific logic to be defined in the get_model_api()
function
"""
input_data = request.json
app.logger.info("api_input: " + str(input_data))
print(input_data)
output_data = server.translate(input_data['word'])
app.logger.info("api_output: " + str(output_data))
response = jsonify(output_data)
return response
@app.route('/')
def index():
return "Index API"
# HTTP Errors handlers
@app.errorhandler(404)
def url_error(e):
return """
Wrong URL!
<pre>{}</pre>""".format(e), 404
@app.errorhandler(500)
def server_error(e):
return """
An internal error occurred: <pre>{}</pre>
See logs for full stacktrace.
""".format(e), 500
if __name__ == '__main__':
# This is used when running locally.
app.run(host=HOST, debug=True)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment