Update src/main.py

src/main.py

@@ -9,6 +9,7 @@ import os
 import requests
 from urllib.parse import quote, unquote
 import tempfile
+import re
 
 app = Flask(__name__, static_folder='static')
 app.config['TITLE'] = 'Sign Language Translate'
@@ -16,9 +17,9 @@ app.config['TITLE'] = 'Sign Language Translate'
 nlp, dict_docs_spacy = sp.load_spacy_values()
 dataset, list_2000_tokens = dg.load_data()
 
-def is_proper_noun(word):
-    """고유명사 여부를 확인하는 함수"""
-    return word[0].isupper() if word else False
+def find_quoted_words(text):
+    """작은따옴표로 묶인 단어들을 찾는 함수"""
+    return re.findall(r"'([^']*)'", text)
 
 def spell_out_word(word):
     """단어를 개별 알파벳으로 분리하는 함수"""
@@ -26,17 +27,36 @@ def spell_out_word(word):
 
 def translate_korean_to_english(text):
     try:
+        # 작은따옴표로 묶인 단어들 찾기
+        quoted_words = find_quoted_words(text)
+        
+        # 번역을 위해 임시로 마커로 대체
+        temp_text = text
+        for i, word in enumerate(quoted_words):
+            temp_text = temp_text.replace(f"'{word}'", f"QUOTED_WORD_{i}")
+        
         url = "https://translate.googleapis.com/translate_a/single"
         params = {
             "client": "gtx", 
             "sl": "ko",
             "tl": "en",
             "dt": "t",
-            "q": text.strip()
+            "q": temp_text.strip()
         }
         response = requests.get(url, params=params)
         if response.status_code == 200:
             translated_text = ' '.join(item[0] for item in response.json()[0] if item[0])
+            
+            # 번역된 텍스트에서 마커를 다시 원래 단어로 대체
+            for i, word in enumerate(quoted_words):
+                # 해당 단어를 영어로 별도 번역
+                word_params = params.copy()
+                word_params['q'] = word
+                word_response = requests.get(url, params=word_params)
+                if word_response.status_code == 200:
+                    translated_word = word_response.json()[0][0][0].upper()
+                    translated_text = translated_text.replace(f"QUOTED_WORD_{i}", f"'{translated_word}'")
+            
             return translated_text
         else:
             raise Exception(f"Translation API returned status code: {response.status_code}")
@@ -103,7 +123,10 @@ def result():
             if not english_text:
                 raise Exception("Translation failed")
                 
-            eng_to_asl_translator = NlpSpacyBaseTranslator(sentence=english_text)
+            # 작은따옴표로 묶인 단어들 찾기
+            quoted_words = find_quoted_words(english_text)
+            
+            eng_to_asl_translator = NlpSpacyBaseTranslator(sentence=english_text.replace("'", ""))
             generated_gloss = eng_to_asl_translator.translate_to_gloss()
             
             # 단어 처리
@@ -111,7 +134,8 @@ def result():
             words = generated_gloss.split()
             
             for word in words:
-                if is_proper_noun(word):
+                word_upper = word.upper()
+                if any(quoted_word.upper() == word_upper for quoted_word in quoted_words):
                     # 고유명사인 경우 철자를 하나씩 분리
                     spelled_word = spell_out_word(word)
                     processed_gloss.extend(['FINGERSPELL-START'] + spelled_word.split() + ['FINGERSPELL-END'])