Update src/ASL_gloss_functions.py

src/ASL_gloss_functions.py

@@ -56,13 +56,11 @@ def question_type(doc):
     except IndexError:
         return None
 
-# add question id as a prefix
 def process_sentence(doc):
     nms = {
         "wh-question": "wh-q",
         "yes-no-question": "y/n-q"
     }
-    
     classifiers = {
         "car": "CL:3",
         "person": "CL:1"
@@ -70,11 +68,12 @@ def process_sentence(doc):
     
     glossed_sentence = []
     for token in doc:
-        ## utilize token.lemma_, not .text
-        #word = token.text.lower()
         word = token.lemma_.lower()
         
-        if word in ["i", "me"]:
+        # 고유명사 처리
+        if token.pos_ == "PROPN":
+            glossed_word = token.text.upper()
+        elif word in ["i", "me"]:
             glossed_word = handle_indexing("I", 1)
         elif word in ["you"]:
             glossed_word = handle_indexing("YOU", 2)
@@ -82,18 +81,6 @@ def process_sentence(doc):
             glossed_word = classifiers[word]
         else:
             glossed_word = gloss_word(word)
+            
         glossed_word = skip_stop_words(glossed_word)
-        
-        glossed_sentence.append(glossed_word)    
-
-    for gloss in glossed_sentence:
-        if gloss.lower() in time_words:
-            # move gloss at beginning
-            glossed_sentence.insert(0, glossed_sentence.pop(glossed_sentence.index(gloss)))
-            break
-        
-    type_doc = question_type(doc)
-    if type_doc != None:
-        glossed_sentence.insert(0, nms[type_doc])
-        
-    return " ".join(glossed_sentence)
+        glossed_sentence.append(glossed_word)