openfree commited on
Commit
8ed001f
·
verified ·
1 Parent(s): c11c758

Update src/main.py

Browse files
Files changed (1) hide show
  1. src/main.py +16 -7
src/main.py CHANGED
@@ -2,14 +2,27 @@ import display_gloss as dg
2
  import synonyms_preprocess as sp
3
  from NLP_Spacy_base_translator import NlpSpacyBaseTranslator
4
  from flask import Flask, render_template, Response, request
5
- from translate import Translator
6
 
7
  app = Flask(__name__)
8
 
9
- # Initialize translators and data
10
  nlp, dict_docs_spacy = sp.load_spacy_values()
11
  dataset, list_2000_tokens = dg.load_data()
12
 
 
 
 
 
 
 
 
 
 
 
 
 
 
13
  @app.route('/')
14
  def index():
15
  return render_template('index.html')
@@ -17,18 +30,14 @@ def index():
17
  @app.route('/translate/', methods=['POST'])
18
  def result():
19
  if request.method == 'POST':
20
- # Get Korean input and translate to English
21
  korean_sentence = request.form['inputSentence']
22
- translator = Translator(from_lang="ko", to_lang="en")
23
- english_translation = translator.translate(korean_sentence)
24
 
25
- # Translate English to ASL gloss
26
  eng_to_asl_translator = NlpSpacyBaseTranslator(sentence=english_translation)
27
  generated_gloss = eng_to_asl_translator.translate_to_gloss()
28
  gloss_list_lower = [gloss.lower() for gloss in generated_gloss.split() if gloss.isalnum()]
29
  gloss_sentence_before_synonym = " ".join(gloss_list_lower)
30
 
31
- # Process synonyms
32
  gloss_list = [sp.find_synonyms(gloss, nlp, dict_docs_spacy, list_2000_tokens)
33
  for gloss in gloss_list_lower]
34
  gloss_sentence_after_synonym = " ".join(gloss_list)
 
2
  import synonyms_preprocess as sp
3
  from NLP_Spacy_base_translator import NlpSpacyBaseTranslator
4
  from flask import Flask, render_template, Response, request
5
+ import requests
6
 
7
  app = Flask(__name__)
8
 
9
+ # 데이터 초기화
10
  nlp, dict_docs_spacy = sp.load_spacy_values()
11
  dataset, list_2000_tokens = dg.load_data()
12
 
13
+ def translate_korean_to_english(text):
14
+ url = "https://translate.googleapis.com/translate_a/single"
15
+ params = {
16
+ "client": "gtx",
17
+ "sl": "ko",
18
+ "tl": "en",
19
+ "dt": "t",
20
+ "q": text
21
+ }
22
+ response = requests.get(url, params=params)
23
+ result = response.json()
24
+ return result[0][0][0]
25
+
26
  @app.route('/')
27
  def index():
28
  return render_template('index.html')
 
30
  @app.route('/translate/', methods=['POST'])
31
  def result():
32
  if request.method == 'POST':
 
33
  korean_sentence = request.form['inputSentence']
34
+ english_translation = translate_korean_to_english(korean_sentence)
 
35
 
 
36
  eng_to_asl_translator = NlpSpacyBaseTranslator(sentence=english_translation)
37
  generated_gloss = eng_to_asl_translator.translate_to_gloss()
38
  gloss_list_lower = [gloss.lower() for gloss in generated_gloss.split() if gloss.isalnum()]
39
  gloss_sentence_before_synonym = " ".join(gloss_list_lower)
40
 
 
41
  gloss_list = [sp.find_synonyms(gloss, nlp, dict_docs_spacy, list_2000_tokens)
42
  for gloss in gloss_list_lower]
43
  gloss_sentence_after_synonym = " ".join(gloss_list)