File size: 2,747 Bytes
9fa4f9e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 |
# MTUOC_typtNMT
# Copyright (C) 2023 Antoni Oliver
# v. 07/06/2023
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <https://www.gnu.org/licenses/>.
import sys
from flask import Flask, jsonify, request
from MTUOC_misc import get_IP_info
from MTUOC_misc import printLOG
import config
from MTUOC_translate import translate_para
def start_OpenNMT_server():
cli = sys.modules['flask.cli']
cli.show_server_banner = lambda *x: None
STATUS_OK = "ok"
STATUS_ERROR = "error"
printLOG(1,"MTUOC server started as OpenNMT server")
STATUS_ERROR = "error"
out={}
def start(url_root="./translator",
host="0.0.0.0",
port=5000,
debug=True):
def prefix_route(route_function, prefix='', mask='{0}{1}'):
def newroute(route, *args, **kwargs):
return route_function(mask.format(prefix, route), *args, **kwargs)
return newroute
app = Flask(__name__)
app.route = prefix_route(app.route, url_root)
@app.route('/translate', methods=['POST'])
def translateONMT():
inputs = request.get_json(force=True)
inputs0=inputs[0]
out = {}
try:
out = [[]]
ss=inputs0['src']
ts=translate_para(ss)
response = {"src": ss, "tgt": ts,
"n_best": 0, "pred_score": 0}
out[0].append(response)
except:
out['error'] = "Error"
out['status'] = STATUS_ERROR
return jsonify(out)
from waitress import serve
serve(app, host=host, port=port,threads= 1)
#app.run(debug=debug, host=host, port=port, use_reloader=False,threaded=True)
url_root=config.MTUOCServer_ONMT_url_root
ip="0.0.0.0"
ip=get_IP_info()
debug="store_true"
print("MTUOC server IP: ", ip)
print("MTUOC server port: ", config.MTUOCServer_port)
print("MTUOC server type: OpenNMT")
start(url_root=url_root, host=ip, port=config.MTUOCServer_port,debug=debug) |