Update src/main.py

src/main.py

@@ -128,6 +128,7 @@ def translate_korean_to_english(text: str) -> str:
         text = normalize_quotes(text)
         
         if is_english(text):
+            # 영어 문장 안의 작은따옴표 단어를 대문자로 바꿔주는 처리
             quoted_match = re.search(r"'([^']*)'", text)
             if quoted_match:
                 quoted_word = quoted_match.group(1).upper()
@@ -189,7 +190,6 @@ def generate_complete_video(gloss_list: List[str], dataset: Dict[str, Any], list
 
         height, width = frames[0].shape[:2]
         
-        # 임시 파일 처리 최적화
         with tempfile.NamedTemporaryFile(suffix='.mp4', delete=False) as temp_file:
             temp_path = temp_file.name
             
@@ -216,6 +216,7 @@ def result():
     if request.method == 'POST':
         input_text = request.form['inputSentence'].strip()
         if not input_text:
+            # 에러 페이지로 이동
             return render_template('error.html', error="Please enter text to translate")
         
         try:
@@ -234,7 +235,7 @@ def result():
                 
             generated_gloss = executor.submit(process_nlp).result()
             
-            # Gloss 처리 최적화
+            # 작은따옴표 처리(지정된 영단어는 fingerspell로 변환)
             processed_gloss = []
             words = generated_gloss.split()
             
@@ -248,41 +249,43 @@ def result():
             
             gloss_sentence_before_synonym = " ".join(processed_gloss)
             
-            # 동의어 처리 최적화
+            # 동의어 처리
             final_gloss = []
             i = 0
             while i < len(processed_gloss):
                 if processed_gloss[i] == 'FINGERSPELL-START':
-                    final_gloss.extend(processed_gloss[i:i+2])
+                    # fingerspell 구간은 그대로 넣는다
+                    final_gloss.extend(processed_gloss[i:i+2])  # FINGERSPELL-START + 첫 글자
                     i += 2
                     while i < len(processed_gloss) and processed_gloss[i] != 'FINGERSPELL-END':
                         final_gloss.append(processed_gloss[i])
                         i += 1
                     if i < len(processed_gloss):
-                        final_gloss.append(processed_gloss[i])
+                        final_gloss.append(processed_gloss[i])  # FINGERSPELL-END
                         i += 1
                 else:
                     word = processed_gloss[i]
-                    # 동의어 찾기를 스레드 풀에서 실행
-                    final_gloss.append(
-                        executor.submit(
-                            sp.find_synonyms,
-                            word,
-                            nlp,
-                            dict_docs_spacy,
-                            list_2000_tokens
-                        ).result()
-                    )
+                    # 동의어 찾기를 스레드 풀에서 비동기로 실행하여 결과 얻기
+                    synonym = executor.submit(
+                        sp.find_synonyms,
+                        word,
+                        nlp,
+                        dict_docs_spacy,
+                        list_2000_tokens
+                    ).result()
+                    final_gloss.append(synonym)
                     i += 1
             
             gloss_sentence_after_synonym = " ".join(final_gloss)
             
-            return render_template('result.html',
-                                title=app.config['TITLE'],
-                                original_sentence=input_text,
-                                english_translation=english_text,
-                                gloss_sentence_before_synonym=gloss_sentence_before_synonym,
-                                gloss_sentence_after_synonym=gloss_sentence_after_synonym)
+            return render_template(
+                'result.html',
+                title=app.config['TITLE'],
+                original_sentence=input_text,
+                english_translation=english_text,
+                gloss_sentence_before_synonym=gloss_sentence_before_synonym,
+                gloss_sentence_after_synonym=gloss_sentence_after_synonym
+            )
                                 
         except Exception as e:
             logger.error(f"Translation processing error: {str(e)}")
@@ -292,8 +295,10 @@ def result():
 def video_feed():
     sentence = request.args.get('gloss_sentence_to_display', '')
     gloss_list = sentence.split()
-    return Response(dg.generate_video(gloss_list, dataset, list_2000_tokens), 
-                   mimetype='multipart/x-mixed-replace; boundary=frame')
+    return Response(
+        dg.generate_video(gloss_list, dataset, list_2000_tokens), 
+        mimetype='multipart/x-mixed-replace; boundary=frame'
+    )
 
 @app.route('/download_video/<path:gloss_sentence>')
 def download_video(gloss_sentence: str):
@@ -320,4 +325,4 @@ def download_video(gloss_sentence: str):
         return f"Error downloading video: {str(e)}", 500
 
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860, debug=True)
+    app.run(host="0.0.0.0", port=7860, debug=True)