Update src/main.py

src/main.py

@@ -19,6 +19,8 @@ dataset, list_2000_tokens = dg.load_data()
 
 def find_quoted_words(text):
     """작은따옴표로 묶인 단어들을 찾는 함수"""
+    # 따옴표가 없는 단어에 따옴표 추가 (예: 한국'을 '한국'으로)
+    text = re.sub(r"(\w+)'", r"'\1'", text)
     return re.findall(r"'([^']*)'", text)
 
 def spell_out_word(word):
@@ -48,35 +50,46 @@ def translate_quoted_word(word):
 def translate_korean_to_english(text):
     """전체 텍스트 번역 함수"""
     try:
+        # 입력 텍스트에서 따옴표 형식 검사 및 수정
+        text = text.replace("'", "'").replace("'", "'")  # 스마트 따옴표를 일반 따옴표로 변환
+        
+        # 잘못된 따옴표 위치 수정 (예: 한국'을 '한국'으로)
+        text = re.sub(r"(\w+)'", r"'\1'", text)
+        
         # 1. 따옴표로 묶인 부분을 찾아서 따로 번역
-        pattern = r"'([^']*)'|([^']+)"
-        parts = re.findall(pattern, text)
-        translated_parts = []
+        quoted_words = find_quoted_words(text)
+        translated_quoted = {}
         
-        for quoted, unquoted in parts:
-            if quoted:  # 따옴표로 묶인 부분
-                translated_word = translate_quoted_word(quoted)
-                translated_parts.append(f"'{translated_word}'")
-            elif unquoted:  # 일반 텍스트
-                # 일반 텍스트 번역
-                url = "https://translate.googleapis.com/translate_a/single"
-                params = {
-                    "client": "gtx",
-                    "sl": "ko",
-                    "tl": "en",
-                    "dt": "t",
-                    "q": unquoted.strip()
-                }
-                response = requests.get(url, params=params)
-                if response.status_code == 200:
-                    translated = ' '.join(item[0] for item in response.json()[0] if item[0])
-                    translated_parts.append(translated)
-                else:
-                    translated_parts.append(unquoted)
+        # 따옴표 안의 단어들 먼저 번역
+        for word in quoted_words:
+            translated = translate_quoted_word(word)
+            translated_quoted[word] = translated
+            # 임시 마커로 대체
+            text = text.replace(f"'{word}'", f"QUOTED_{len(translated_quoted)}_")
         
-        # 번역된 부분들을 합치기
-        result = ''.join(translated_parts).strip()
-        return result
+        # 전체 문장 번역
+        url = "https://translate.googleapis.com/translate_a/single"
+        params = {
+            "client": "gtx",
+            "sl": "ko",
+            "tl": "en",
+            "dt": "t",
+            "q": text
+        }
+        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, (original, translated) in enumerate(translated_quoted.items(), 1):
+                translated_text = translated_text.replace(f"QUOTED_{i}_", f"'{translated}'")
+            
+            # 불필요한 공백 정리
+            translated_text = re.sub(r'\s+', ' ', translated_text).strip()
+            return translated_text
+        else:
+            raise Exception(f"Translation API returned status code: {response.status_code}")
     except Exception as e:
         print(f"Translation error: {e}")
         return text