sugoi-v4 / server /app.py
playmak3r's picture
feat: add route to the model
5f9624d
raw
history blame
869 Bytes
from flask import Flask, send_from_directory, json, Response, request
from waitress import serve
from flask_cors import CORS, cross_origin
from server.model import SugoiTranslator
app = Flask(__name__)
CORS(app)
@app.route('/')
def index():
return send_from_directory('dist', "index.html")
@app.route('/<path:filename>')
def download_file(filename):
return send_from_directory('.', filename)
@app.route('/assets/<path:filename>')
def download_assets(filename):
return send_from_directory('dist/assets', filename)
@app.route('api/translate', methods= ['POST'])
@cross_origin()
def translate_api():
text = request.args.get('text')
sugoiTranslator = SugoiTranslator()
if isinstance(text, str) and len(text) > 0: return json.dump(sugoiTranslator.translate(text))
return Response(status= 400)
if __name__ == '__main__':
serve(app)