-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathmain.py
More file actions
77 lines (54 loc) · 1.62 KB
/
main.py
File metadata and controls
77 lines (54 loc) · 1.62 KB
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
from nltk.sentiment.vader import SentimentIntensityAnalyzer
from nltk.corpus import wordnet as wn
from nltk import tokenize
from googletrans import Translator
from flask import Flask, request, jsonify
from celery import Celery
translator = Translator()
app = Flask(__name__)
app.config['CELERY_BROKER_URL'] = 'amqp://jdv:jdv@localhost:5672/v_host'
celery = Celery(app.name, broker=app.config['CELERY_BROKER_URL'])
celery.conf.update(app.config)
@app.route('/')
def health_check():
return 'Translation Service is up.'
@app.route('/detect-sentiment', methods=['POST'])
def sentiment():
form = request.form
text = form['text']
return _sentiment(text)
@celery.task
def _sentiment(text):
sid = SentimentIntensityAnalyzer()
sentiment_score = sid.polarity_scores(text)
return sentiment_score
@app.route('/synonyms', methods=['POST'])
def synonyms():
form = request.form
text = form['text']
return _synonyms(text)
@celery.task
def _synonyms(text):
synset = wn.synsets(text)
return str(synset[0].hyponyms())
@app.route('/detect-lang', methods=['POST'])
def detect():
form = request.form
text = form['text']
return _detect(text)
@celery.task
def _detect(text):
lang_detected = translator.detect(text)
return lang_detected.lang
@app.route('/translate', methods=['POST'])
def translate():
form = request.form
text = form['text']
target = form['target']
return _translate(text, target)
@celery.task
def _translate(text, target):
translated_text = translator.translate(text, target)
return translated_text.text
if __name__ == '__main__':
app.run(debug=True)