Update src/main.py

src/main.py

@@ -77,41 +77,45 @@ def spell_out_word(word):
 def translate_korean_text(text):
     """한글 전용 번역 함수"""
     try:
-        # 따옴표로 묶인 단어 찾기
+        # 1. 따옴표로 묶인 단어 찾기 및 임시 마커로 대체
         quoted_match = re.search(r"'([^']*)'", text)
         if not quoted_match:
             return text
 
-        # 1. 따옴표 안의 단어만 먼저 번역
-        korean_word = quoted_match.group(1)
+        quoted_word = quoted_match.group(1)
+        marker = "PROPER_NOUN_MARKER"
+        text_with_marker = text.replace(f"'{quoted_word}'", marker)
+
+        # 2. 전체 문장 번역
         url = "https://translate.googleapis.com/translate_a/single"
         params = {
             "client": "gtx",
             "sl": "ko",
             "tl": "en",
             "dt": "t",
-            "q": korean_word
+            "q": text_with_marker
         }
         response = requests.get(url, params=params)
         if response.status_code != 200:
             return text
-            
-        proper_noun = response.json()[0][0][0].upper()
-
-        # 2. 나머지 문장만 따로 번역
-        remaining_text = text[text.find("'", text.find(korean_word) + len(korean_word)) + 1:].strip()
-        if not remaining_text:
-            return f"'{proper_noun}'"
-
-        params["q"] = remaining_text
-        response = requests.get(url, params=params)
-        if response.status_code != 200:
-            return text
 
-        remaining_translation = ' '.join(item[0] for item in response.json()[0] if item[0])
+        # 3. 번역된 전체 문장에서 마커를 원래 단어로 대체
+        translated_text = ' '.join(item[0] for item in response.json()[0] if item[0])
         
-        # 3. 번역된 부분들 조합
-        return f"'{proper_noun}' {remaining_translation}"
+        # 영어로 된 단어는 그대로 유지
+        if re.match(r'^[A-Z]+$', quoted_word):
+            final_text = translated_text.replace(marker, f"'{quoted_word}'")
+        else:
+            # 한글 단어는 번역
+            params["q"] = quoted_word
+            response = requests.get(url, params=params)
+            if response.status_code == 200:
+                proper_noun = response.json()[0][0][0].upper()
+                final_text = translated_text.replace(marker, f"'{proper_noun}'")
+            else:
+                final_text = translated_text.replace(marker, f"'{quoted_word}'")
+
+        return final_text
 
     except Exception as e:
         print(f"Korean translation error: {e}")