Update src/main.py

src/main.py

@@ -43,7 +43,20 @@ def normalize_quotes(text):
     # 불필요한 공백 제거
     text = re.sub(r'\s+', ' ', text).strip()
     
-    # 첫 번째 단어에만 따옴표 처리
+    # 이미 따옴표로 묶인 단어가 있는지 확인
+    existing_quotes = re.findall(r"'([^']*)'", text)
+    if existing_quotes:
+        return text  # 이미 따옴표가 있으면 그대로 반환
+    
+    # 대문자로 된 단어 찾기 (예: JOHN)
+    uppercase_words = re.findall(r'\b[A-Z]+\b', text)
+    if uppercase_words:
+        # 대문자 단어에 따옴표 추가
+        for word in uppercase_words:
+            text = text.replace(word, f"'{word}'")
+        return text
+    
+    # 위의 조건에 해당하지 않는 경우 첫 단어에 따옴표 추가
     words = text.split()
     if words:
         # 모든 따옴표 제거 후 첫 단어에만 따옴표 추가