Update src/main.py

src/main.py

@@ -1,4 +1,3 @@
-# app.py
 import display_gloss as dg
 import synonyms_preprocess as sp
 from NLP_Spacy_base_translator import NlpSpacyBaseTranslator 
@@ -17,6 +16,14 @@ 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 spell_out_word(word):
+    """단어를 개별 알파벳으로 분리하는 함수"""
+    return ' '.join(list(word.lower()))
+
 def translate_korean_to_english(text):
     try:
         url = "https://translate.googleapis.com/translate_a/single"
@@ -40,11 +47,21 @@ def translate_korean_to_english(text):
 def generate_complete_video(gloss_list, dataset, list_2000_tokens):
     try:
         frames = []
-        for frame in dg.generate_video(gloss_list, dataset, list_2000_tokens):
-            frame_data = frame.split(b'\r\n\r\n')[1]
-            nparr = np.frombuffer(frame_data, np.uint8)
-            img = cv2.imdecode(nparr, cv2.IMREAD_COLOR)
-            frames.append(img)
+        is_spelling = False
+        
+        for gloss in gloss_list:
+            if gloss == 'FINGERSPELL-START':
+                is_spelling = True
+                continue
+            elif gloss == 'FINGERSPELL-END':
+                is_spelling = False
+                continue
+                
+            for frame in dg.generate_video([gloss], dataset, list_2000_tokens):
+                frame_data = frame.split(b'\r\n\r\n')[1]
+                nparr = np.frombuffer(frame_data, np.uint8)
+                img = cv2.imdecode(nparr, cv2.IMREAD_COLOR)
+                frames.append(img)
         
         if not frames:
             raise Exception("No frames generated")
@@ -89,12 +106,43 @@ def result():
             eng_to_asl_translator = NlpSpacyBaseTranslator(sentence=english_text)
             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)
+            # 단어 처리
+            processed_gloss = []
+            words = generated_gloss.split()
+            
+            for word in words:
+                if is_proper_noun(word):
+                    # 고유명사인 경우 철자를 하나씩 분리
+                    spelled_word = spell_out_word(word)
+                    processed_gloss.extend(['FINGERSPELL-START'] + spelled_word.split() + ['FINGERSPELL-END'])
+                else:
+                    # 일반 단어는 기존 방식대로 처리
+                    word_lower = word.lower()
+                    if word_lower.isalnum():
+                        processed_gloss.append(word_lower)
+            
+            gloss_sentence_before_synonym = " ".join(processed_gloss)
+            
+            # 고유명사가 아닌 단어들만 동의어 처리
+            final_gloss = []
+            i = 0
+            while i < len(processed_gloss):
+                if processed_gloss[i] == 'FINGERSPELL-START':
+                    # 철자 처리 부분을 그대로 유지
+                    final_gloss.append(processed_gloss[i])
+                    i += 1
+                    while i < len(processed_gloss) and processed_gloss[i] != 'FINGERSPELL-END':
+                        final_gloss.append(processed_gloss[i])
+                        i += 1
+                    final_gloss.append('FINGERSPELL-END')
+                    i += 1
+                else:
+                    # 일반 단어는 동의어 처리
+                    word = processed_gloss[i]
+                    final_gloss.append(sp.find_synonyms(word, nlp, dict_docs_spacy, list_2000_tokens))
+                    i += 1
             
-            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)
+            gloss_sentence_after_synonym = " ".join(final_gloss)
             
             return render_template('result.html',
                                 title=app.config['TITLE'],