mirror of
https://github.com/LibreTranslate/LibreTranslate.git
synced 2024-11-21 23:41:01 +00:00
Add /detect API
This commit is contained in:
parent
3156761037
commit
46fca61cc2
2 changed files with 124 additions and 0 deletions
80
app/app.py
80
app/app.py
|
@ -258,6 +258,86 @@ def create_app(char_limit=-1, req_limit=-1, batch_limit=-1, ga_id=None, debug=Fa
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
abort(500, description="Cannot translate text: %s" % str(e))
|
abort(500, description="Cannot translate text: %s" % str(e))
|
||||||
|
|
||||||
|
@app.route("/detect", methods=['POST'])
|
||||||
|
def detect():
|
||||||
|
"""
|
||||||
|
Detect the language of a single text
|
||||||
|
---
|
||||||
|
tags:
|
||||||
|
- translate
|
||||||
|
parameters:
|
||||||
|
- in: formData
|
||||||
|
name: q
|
||||||
|
schema:
|
||||||
|
type: string
|
||||||
|
example: Hello world!
|
||||||
|
required: true
|
||||||
|
description: Text to detect
|
||||||
|
responses:
|
||||||
|
200:
|
||||||
|
description: Detections
|
||||||
|
schema:
|
||||||
|
id: detections
|
||||||
|
type: array
|
||||||
|
items:
|
||||||
|
type: object
|
||||||
|
properties:
|
||||||
|
confidence:
|
||||||
|
type: number
|
||||||
|
format: float
|
||||||
|
minimum: 0
|
||||||
|
maximum: 1
|
||||||
|
description: Confidence value
|
||||||
|
example: 0.6
|
||||||
|
language:
|
||||||
|
type: string
|
||||||
|
description: Language code
|
||||||
|
example: en
|
||||||
|
400:
|
||||||
|
description: Invalid request
|
||||||
|
schema:
|
||||||
|
id: error-response
|
||||||
|
type: object
|
||||||
|
properties:
|
||||||
|
error:
|
||||||
|
type: string
|
||||||
|
description: Error message
|
||||||
|
500:
|
||||||
|
description: Detection error
|
||||||
|
schema:
|
||||||
|
id: error-response
|
||||||
|
type: object
|
||||||
|
properties:
|
||||||
|
error:
|
||||||
|
type: string
|
||||||
|
description: Error message
|
||||||
|
429:
|
||||||
|
description: Slow down
|
||||||
|
schema:
|
||||||
|
id: error-slow-down
|
||||||
|
type: object
|
||||||
|
properties:
|
||||||
|
error:
|
||||||
|
type: string
|
||||||
|
description: Reason for slow down
|
||||||
|
"""
|
||||||
|
if request.is_json:
|
||||||
|
json = request.get_json()
|
||||||
|
q = json.get('q')
|
||||||
|
else:
|
||||||
|
q = request.values.get("q")
|
||||||
|
|
||||||
|
if not q:
|
||||||
|
abort(400, description="Invalid request: missing q parameter")
|
||||||
|
|
||||||
|
candidate_langs = list(filter(lambda l: l.lang in language_map, detect_langs(q)))
|
||||||
|
candidate_langs.sort(key=lambda l: l.prob, reverse=True)
|
||||||
|
return jsonify([{
|
||||||
|
'confidence': l.prob,
|
||||||
|
'language': l.lang
|
||||||
|
} for l in candidate_langs])
|
||||||
|
|
||||||
|
|
||||||
@app.route("/frontend/settings")
|
@app.route("/frontend/settings")
|
||||||
def frontend_settings():
|
def frontend_settings():
|
||||||
"""
|
"""
|
||||||
|
|
44
main.py
Normal file
44
main.py
Normal file
|
@ -0,0 +1,44 @@
|
||||||
|
import argparse
|
||||||
|
from app.app import create_app
|
||||||
|
|
||||||
|
parser = argparse.ArgumentParser(description='LibreTranslate - Free and Open Source Translation API')
|
||||||
|
parser.add_argument('--host', type=str,
|
||||||
|
help='Hostname (%(default)s)', default="127.0.0.1")
|
||||||
|
parser.add_argument('--port', type=int,
|
||||||
|
help='Port (%(default)s)', default=5000)
|
||||||
|
parser.add_argument('--char-limit', default=-1, type=int, metavar="<number of characters>",
|
||||||
|
help='Set character limit (%(default)s)')
|
||||||
|
parser.add_argument('--req-limit', default=-1, type=int, metavar="<number>",
|
||||||
|
help='Set maximum number of requests per minute per client (%(default)s)')
|
||||||
|
parser.add_argument('--batch-limit', default=-1, type=int, metavar="<number of texts>",
|
||||||
|
help='Set maximum number of texts to translate in a batch request (%(default)s)')
|
||||||
|
parser.add_argument('--ga-id', type=str, default=None, metavar="<GA ID>",
|
||||||
|
help='Enable Google Analytics on the API client page by providing an ID (%(default)s)')
|
||||||
|
parser.add_argument('--debug', default=False, action="store_true",
|
||||||
|
help="Enable debug environment")
|
||||||
|
parser.add_argument('--ssl', default=None, action="store_true",
|
||||||
|
help="Whether to enable SSL")
|
||||||
|
parser.add_argument('--frontend-language-source', type=str, default="en", metavar="<language code>",
|
||||||
|
help='Set frontend default language - source (%(default)s)')
|
||||||
|
parser.add_argument('--frontend-language-target', type=str, default="es", metavar="<language code>",
|
||||||
|
help='Set frontend default language - target (%(default)s)')
|
||||||
|
parser.add_argument('--frontend-timeout', type=int, default=500, metavar="<milliseconds>",
|
||||||
|
help='Set frontend translation timeout (%(default)s)')
|
||||||
|
|
||||||
|
args = parser.parse_args()
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
app = create_app(char_limit=args.char_limit,
|
||||||
|
req_limit=args.req_limit,
|
||||||
|
batch_limit=args.batch_limit,
|
||||||
|
ga_id=args.ga_id,
|
||||||
|
debug=args.debug,
|
||||||
|
frontend_language_source=args.frontend_language_source,
|
||||||
|
frontend_language_target=args.frontend_language_target,
|
||||||
|
frontend_timeout=args.frontend_timeout)
|
||||||
|
if args.debug:
|
||||||
|
app.run(host=args.host, port=args.port)
|
||||||
|
else:
|
||||||
|
from waitress import serve
|
||||||
|
serve(app, host=args.host, port=args.port, url_scheme='https' if args.ssl else 'http')
|
Loading…
Reference in a new issue