mirror of
https://github.com/LibreTranslate/LibreTranslate.git
synced 2024-06-02 21:49:33 +00:00
Swagger docs, fix links, character limits
This commit is contained in:
Piero Toffanin
parent
e9609ddd64
commit
c5863fcc84
180
main.py
180
main.py
|
|
@ -2,12 +2,16 @@ from app.init import boot
|
|||
import argparse
|
||||
from flask import Flask, render_template, jsonify, request, abort, send_from_directory
|
||||
from app.language import languages
|
||||
from flask_swagger import swagger
|
||||
from flask_swagger_ui import get_swaggerui_blueprint
|
||||
|
||||
parser = argparse.ArgumentParser(description='LibreTranslate - Free and Open Source Translation API')
|
||||
parser.add_argument('host', type=str,
|
||||
help='Hostname', default="127.0.0.1")
|
||||
parser.add_argument('port', type=int,
|
||||
help='Port', default=5000)
|
||||
parser.add_argument('--char-limit', default=-1,
|
||||
help='Character limit')
|
||||
args = parser.parse_args()
|
||||
|
||||
boot()
|
||||
|
|
@ -27,46 +31,166 @@ def server_error(e):
|
|||
def index():
|
||||
return send_from_directory('static', 'index.html')
|
||||
|
||||
|
||||
@app.route("/languages")
|
||||
def langs():
|
||||
return jsonify([{'code': l.code, 'name': l.name } for l in languages])
|
||||
"""
|
||||
Retrieve list of supported languages
|
||||
---
|
||||
tags:
|
||||
- translate
|
||||
responses:
|
||||
200:
|
||||
description: List of languages
|
||||
content:
|
||||
application/json:
|
||||
schema:
|
||||
type: array
|
||||
items:
|
||||
type: object
|
||||
properties:
|
||||
code:
|
||||
type: string
|
||||
description: Language code
|
||||
name:
|
||||
type: string
|
||||
description: Human-readable language name (in English)
|
||||
charLimit:
|
||||
type: string
|
||||
description: Character input limit for this language (-1 indicates no limit)
|
||||
"""
|
||||
return jsonify([{'code': l.code, 'name': l.name, 'charLimit': args.char_limit } for l in languages])
|
||||
|
||||
# Add cors
|
||||
@app.after_request
|
||||
def after_request(response):
|
||||
response.headers.add('Access-Control-Allow-Origin','*')
|
||||
response.headers.add('Access-Control-Allow-Headers', "Authorization, Content-Type")
|
||||
response.headers.add('Access-Control-Expose-Headers', "Authorization")
|
||||
response.headers.add('Access-Control-Allow-Methods', "GET, POST")
|
||||
response.headers.add('Access-Control-Allow-Credentials', "true")
|
||||
response.headers.add('Access-Control-Max-Age', 60 * 60 * 24 * 20)
|
||||
return response
|
||||
|
||||
|
||||
@app.route("/translate", methods=['POST'])
|
||||
def translate():
|
||||
"""
|
||||
Translate text from a language to another
|
||||
---
|
||||
tags:
|
||||
- translate
|
||||
parameters:
|
||||
- in: formData
|
||||
name: q
|
||||
schema:
|
||||
type: string
|
||||
example: Hello world!
|
||||
required: true
|
||||
description: Text to translate
|
||||
- in: formData
|
||||
name: source
|
||||
schema:
|
||||
type: string
|
||||
example: en
|
||||
required: true
|
||||
description: Source language code
|
||||
- in: formData
|
||||
name: target
|
||||
schema:
|
||||
type: string
|
||||
example: es
|
||||
required: true
|
||||
description: Target language code
|
||||
responses:
|
||||
200:
|
||||
description: Translated text
|
||||
content:
|
||||
application/json:
|
||||
schema:
|
||||
type: object
|
||||
properties:
|
||||
translatedText:
|
||||
type: string
|
||||
description: Translated text
|
||||
400:
|
||||
description: Invalid request
|
||||
content:
|
||||
application/json:
|
||||
schema:
|
||||
type: object
|
||||
properties:
|
||||
error:
|
||||
type: string
|
||||
description: Error message
|
||||
500:
|
||||
description: Translation error
|
||||
content:
|
||||
application/json:
|
||||
schema:
|
||||
type: object
|
||||
properties:
|
||||
error:
|
||||
type: string
|
||||
description: Error message
|
||||
"""
|
||||
|
||||
if request.is_json:
|
||||
json = request.get_json()
|
||||
q = json.get('q')
|
||||
source_lang = json.get('source')
|
||||
target_lang = json.get('target')
|
||||
else:
|
||||
q = request.values.get("q")
|
||||
source_lang = request.values.get("source")
|
||||
target_lang = request.values.get("target")
|
||||
if request.is_json:
|
||||
json = request.get_json()
|
||||
q = json.get('q')
|
||||
source_lang = json.get('source')
|
||||
target_lang = json.get('target')
|
||||
else:
|
||||
q = request.values.get("q")
|
||||
source_lang = request.values.get("source")
|
||||
target_lang = request.values.get("target")
|
||||
|
||||
if not q:
|
||||
abort(400, description="Invalid request: missing q parameter")
|
||||
if not source_lang:
|
||||
abort(400, description="Invalid request: missing source parameter")
|
||||
if not target_lang:
|
||||
abort(400, description="Invalid request: missing target parameter")
|
||||
if not q:
|
||||
abort(400, description="Invalid request: missing q parameter")
|
||||
if not source_lang:
|
||||
abort(400, description="Invalid request: missing source parameter")
|
||||
if not target_lang:
|
||||
abort(400, description="Invalid request: missing target parameter")
|
||||
|
||||
if args.char_limit != -1:
|
||||
q = q[:args.char_limit]
|
||||
|
||||
src_lang = next(iter([l for l in languages if l.code == source_lang]), None)
|
||||
tgt_lang = next(iter([l for l in languages if l.code == target_lang]), None)
|
||||
|
||||
if src_lang is None:
|
||||
abort(400, description="%s is not supported" % source_lang)
|
||||
if tgt_lang is None:
|
||||
abort(400, description="%s is not supported" % target_lang)
|
||||
|
||||
translator = src_lang.get_translation(tgt_lang)
|
||||
try:
|
||||
return jsonify({"translatedText": translator.translate(q) })
|
||||
except Exception as e:
|
||||
abort(500, description="Cannot translate text: %s" % str(e))
|
||||
|
||||
|
||||
src_lang = next(iter([l for l in languages if l.code == source_lang]), None)
|
||||
tgt_lang = next(iter([l for l in languages if l.code == target_lang]), None)
|
||||
swag = swagger(app)
|
||||
swag['info']['version'] = "1.0"
|
||||
swag['info']['title'] = "LibreTranslate"
|
||||
|
||||
if src_lang is None:
|
||||
abort(400, description="%s is not supported" % source_lang)
|
||||
if tgt_lang is None:
|
||||
abort(400, description="%s is not supported" % target_lang)
|
||||
@app.route("/spec")
|
||||
def spec():
|
||||
return jsonify(swag)
|
||||
|
||||
translator = src_lang.get_translation(tgt_lang)
|
||||
try:
|
||||
return jsonify({"translatedText": translator.translate(q) })
|
||||
except Exception as e:
|
||||
abort(500, description="Cannot translate text: %s" % str(e))
|
||||
SWAGGER_URL = '/docs' # URL for exposing Swagger UI (without trailing '/')
|
||||
API_URL = 'http://petstore.swagger.io/v2/swagger.json' # Our API url (can of course be a local resource)
|
||||
|
||||
# Call factory function to create our blueprint
|
||||
swaggerui_blueprint = get_swaggerui_blueprint(
|
||||
SWAGGER_URL,
|
||||
"",
|
||||
config={ # Swagger UI config overrides
|
||||
'app_name': "LibreTranslate",
|
||||
"spec": swag
|
||||
}
|
||||
)
|
||||
|
||||
app.register_blueprint(swaggerui_blueprint)
|
||||
|
||||
if __name__ == "__main__":
|
||||
app.run(host=args.host)
|
||||
|
|
|
|||
|
|
@ -1,2 +1,4 @@
|
|||
argostranslate==1.0.3
|
||||
Flask==1.1.2
|
||||
flask-swagger==0.2.14
|
||||
flask-swagger-ui==3.36.0
|
||||
|
|
|
|||
|
|
@ -21,22 +21,27 @@
|
|||
overflow: auto;
|
||||
font-family: monospace;
|
||||
min-height: 280px;
|
||||
width: 100%;
|
||||
overflow: auto;
|
||||
}
|
||||
.progress.translate{
|
||||
position: absolute;
|
||||
}
|
||||
.card.horizontal .card-stacked{
|
||||
overflow: auto;
|
||||
}
|
||||
</style>
|
||||
</head>
|
||||
<body>
|
||||
<nav class="blue lighten-1" role="navigation">
|
||||
<div class="nav-wrapper container"><a id="logo-container" href="#" class="brand-logo"><i class="material-icons">translate</i> LibreTranslate</a>
|
||||
<div class="nav-wrapper container"><a id="logo-container" href="/" class="brand-logo"><i class="material-icons">translate</i> LibreTranslate</a>
|
||||
<ul class="right hide-on-med-and-down">
|
||||
<li><a href="https://github.com/uav4geo/LibreTranslate">Documentation</a></li>
|
||||
<li><a href="/docs">API Docs</a></li>
|
||||
<li><a href="https://github.com/uav4geo/LibreTranslate">GitHub</a></li>
|
||||
</ul>
|
||||
|
||||
<ul id="nav-mobile" class="sidenav">
|
||||
<li><a href="https://github.com/uav4geo/LibreTranslate">Documentation</a></li>
|
||||
<li><a href="/docs">API Docs</a></li>
|
||||
<li><a href="https://github.com/uav4geo/LibreTranslate">GitHub</a></li>
|
||||
</ul>
|
||||
<a href="#" data-target="nav-mobile" class="sidenav-trigger"><i class="material-icons">menu</i></a>
|
||||
|
|
@ -61,7 +66,24 @@
|
|||
</div>
|
||||
</div>
|
||||
<div v-else-if="error">
|
||||
{{ error }}: TODO
|
||||
<div class="section no-pad-bot">
|
||||
<div class="container">
|
||||
<div class="row">
|
||||
<div class="col s12 m7">
|
||||
<div class="card horizontal">
|
||||
<div class="card-stacked">
|
||||
<div class="card-content">
|
||||
<i class="material-icons">warning</i><p> {{ error }}</p>
|
||||
</div>
|
||||
<div class="card-action">
|
||||
<a href="#" @click="dismissError">Dismiss</a>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
<div v-else>
|
||||
<div class="section no-pad-bot">
|
||||
|
|
@ -95,6 +117,9 @@
|
|||
<div class="input-field col s6">
|
||||
<textarea id="textarea1" class="materialize-textarea" v-model="inputText" @input="handleInput" ref="inputTextarea"></textarea>
|
||||
<label for="textarea1">Input Text</label>
|
||||
<div v-if="charactersLimit !== -1">
|
||||
<label>{{ inputText.length }} / {{ charactersLimit }}</label>
|
||||
</div>
|
||||
</div>
|
||||
<div class="input-field col s6">
|
||||
<div>
|
||||
|
|
@ -121,13 +146,13 @@
|
|||
<div class="card-stacked">
|
||||
<div class="card-content">
|
||||
<div class="row center">
|
||||
<div class="col s12 m6 left-align">
|
||||
<div class="col s12 m12 l6 left-align">
|
||||
<p>Request</p>
|
||||
<p>
|
||||
<pre class="code"><code class="language-javascript" v-html="$options.filters.highlight(requestCode)">
|
||||
</code></pre></p>
|
||||
</div>
|
||||
<div class="col s12 m6 left-align">
|
||||
<div class="col s12 m12 l6 left-align">
|
||||
<p>Response</p>
|
||||
<pre class="code"><code class="language-javascript" v-html="$options.filters.highlight(output)">
|
||||
</code></pre>
|
||||
|
|
@ -200,7 +225,8 @@ document.addEventListener('DOMContentLoaded', function(){
|
|||
loadingTranslation: false,
|
||||
inputText: "",
|
||||
translatedText: "",
|
||||
output: ""
|
||||
output: "",
|
||||
charactersLimit: -1,
|
||||
},
|
||||
mounted: function(){
|
||||
var self = this;
|
||||
|
|
@ -219,6 +245,8 @@ document.addEventListener('DOMContentLoaded', function(){
|
|||
|
||||
self.sourceLang = self.langs[0].code;
|
||||
self.targetLang = self.langs[1].code;
|
||||
self.charactersLimit = self.langs[0].charLimit;
|
||||
// TODO: update this when switching languages
|
||||
|
||||
// Set Spanish
|
||||
for (var i = 1; i < self.langs.length; i++){
|
||||
|
|
@ -251,6 +279,10 @@ document.addEventListener('DOMContentLoaded', function(){
|
|||
this.$refs.inputTextarea.style.height = Math.max(150, this.$refs.inputTextarea.scrollHeight) + "px";
|
||||
this.$refs.translatedTextarea.style.height = Math.max(150, this.$refs.translatedTextarea.scrollHeight) + "px";
|
||||
}
|
||||
|
||||
if (this.charactersLimit !== -1 && this.inputText.length >= this.charactersLimit){
|
||||
this.inputText = this.inputText.substring(0, this.charactersLimit);
|
||||
}
|
||||
},
|
||||
computed: {
|
||||
requestCode: function(){
|
||||
|
|
@ -291,6 +323,9 @@ document.addEventListener('DOMContentLoaded', function(){
|
|||
this.translatedText = "";
|
||||
this.handleInput();
|
||||
},
|
||||
dismissError: function(){
|
||||
this.error = '';
|
||||
},
|
||||
handleInput: function(e){
|
||||
if (this.timeout) clearTimeout(this.timeout);
|
||||
this.timeout = null;
|
||||
|
|
|
|||
Loading…
Reference in a new issue