Update src/main.py

src/main.py

@@ -1,12 +1,12 @@
 import display_gloss as dg
 import synonyms_preprocess as sp
 from NLP_Spacy_base_translator import NlpSpacyBaseTranslator 
-from flask import Flask, render_template, Response, request
+from flask import Flask, render_template, Response, request, jsonify
 import requests
 
-app = Flask(__name__)
+app = Flask(__name__, static_folder='static')
+app.config['TITLE'] = 'Sign Language Translate'
 
-# 데이터 초기화
 nlp, dict_docs_spacy = sp.load_spacy_values()
 dataset, list_2000_tokens = dg.load_data()
 
@@ -20,33 +20,36 @@ def translate_korean_to_english(text):
         "q": text
     }
     response = requests.get(url, params=params)
-    result = response.json()
-    return result[0][0][0]
+    return response.json()[0][0][0]
 
 @app.route('/')
 def index():
-    return render_template('index.html')
+    return render_template('index.html', title=app.config['TITLE'])
 
 @app.route('/translate/', methods=['POST'])
 def result():
     if request.method == 'POST':
         korean_sentence = request.form['inputSentence']
-        english_translation = translate_korean_to_english(korean_sentence)
-        
-        eng_to_asl_translator = NlpSpacyBaseTranslator(sentence=english_translation)
-        generated_gloss = eng_to_asl_translator.translate_to_gloss()
-        gloss_list_lower = [gloss.lower() for gloss in generated_gloss.split() if gloss.isalnum()]
-        gloss_sentence_before_synonym = " ".join(gloss_list_lower)
-        
-        gloss_list = [sp.find_synonyms(gloss, nlp, dict_docs_spacy, list_2000_tokens) 
-                     for gloss in gloss_list_lower]
-        gloss_sentence_after_synonym = " ".join(gloss_list)
-        
-        return render_template('translate.html',
-                            original_sentence=korean_sentence,
-                            english_translation=english_translation,
-                            gloss_sentence_before_synonym=gloss_sentence_before_synonym,
-                            gloss_sentence_after_synonym=gloss_sentence_after_synonym)
+        try:
+            english_translation = translate_korean_to_english(korean_sentence)
+            eng_to_asl_translator = NlpSpacyBaseTranslator(sentence=english_translation)
+            generated_gloss = eng_to_asl_translator.translate_to_gloss()
+            
+            gloss_list_lower = [gloss.lower() for gloss in generated_gloss.split() if gloss.isalnum()]
+            gloss_sentence_before_synonym = " ".join(gloss_list_lower)
+            
+            gloss_list = [sp.find_synonyms(gloss, nlp, dict_docs_spacy, list_2000_tokens) 
+                         for gloss in gloss_list_lower]
+            gloss_sentence_after_synonym = " ".join(gloss_list)
+            
+            return render_template('result.html',
+                                title=app.config['TITLE'],
+                                original_sentence=korean_sentence,
+                                english_translation=english_translation,
+                                gloss_sentence_before_synonym=gloss_sentence_before_synonym,
+                                gloss_sentence_after_synonym=gloss_sentence_after_synonym)
+        except Exception as e:
+            return render_template('error.html', error=str(e))
 
 @app.route('/video_feed')
 def video_feed():
@@ -56,5 +59,4 @@ def video_feed():
                    mimetype='multipart/x-mixed-replace; boundary=frame')
 
 if __name__ == "__main__":
-    app.debug = True
     app.run(host="0.0.0.0", port=5000, debug=True)