Update src/main.py

src/main.py

@@ -47,55 +47,83 @@ def translate_quoted_word(word):
         print(f"Word translation error: {e}")
         return word
 
+
+
+def normalize_quotes(text):
+    """따옴표 형식을 정규화하는 함수"""
+    # 따옴표가 없는 단어 끝에 붙은 따옴표 처리
+    text = re.sub(r"(\w+)'", r"'\1'", text)
+    # 한쪽 따옴표만 있는 경우 처리
+    text = re.sub(r"'(\w+)(?!')", r"'\1'", text)
+    text = re.sub(r"(?<!')(\w+)'", r"'\1'", text)
+    return text
+
+def normalize_quotes(text):
+    """따옴표 형식을 정규화하는 함수"""
+    # 따옴표가 없는 단어 끝에 붙은 따옴표 처리
+    text = re.sub(r"(\w+)'", r"'\1'", text)
+    # 한쪽 따옴표만 있는 경우 처리
+    text = re.sub(r"'(\w+)(?!')", r"'\1'", text)
+    text = re.sub(r"(?<!')(\w+)'", r"'\1'", text)
+    return text
+
 def is_english(text):
     """텍스트가 영어인지 확인하는 함수"""
     # 영어 알파벳과 기본적인 문장부호만 포함되어 있는지 확인
     english_pattern = re.compile(r'^[A-Za-z\s\'".,!?-]+$')
     return bool(english_pattern.match(text.replace("'", "")))
 
-def process_english_input(text):
-    """영어 입력을 처리하는 함수"""
-    # 따옴표로 묶인 단어들을 그대로 유지하면서 나머지 텍스트는 그대로 반환
-    return text
-
 def translate_korean_to_english(text):
     """전체 텍스트 번역 함수"""
     try:
-        # 입력이 영어인지 확인
+        # 입력 텍스트 정규화
+        text = normalize_quotes(text)
+        
+        # 영어 입력 확인
         if is_english(text):
-            return process_english_input(text)
+            return text
 
-        # 한국어 입력 처리
-        # 1. 따옴표로 묶인 부분을 찾아서 따로 번역
-        pattern = r"'([^']*)'|([^']+)"
-        parts = re.findall(pattern, text)
-        translated_parts = []
-        
-        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)
-        
-        # 번역된 부분들을 합치기
-        result = ''.join(translated_parts).strip()
-        return result
+        # 따옴표로 묶인 단어들 찾기
+        quoted_words = re.findall(r"'([^']*)'", text)
+        translated_quoted = {}
+
+        # 따옴표 안의 단어들 먼저 번역
+        for word in quoted_words:
+            url = "https://translate.googleapis.com/translate_a/single"
+            params = {
+                "client": "gtx",
+                "sl": "ko",
+                "tl": "en",
+                "dt": "t",
+                "q": word
+            }
+            response = requests.get(url, params=params)
+            if response.status_code == 200:
+                translated = response.json()[0][0][0].upper()
+                translated_quoted[word] = translated
+                # 임시 마커로 대체
+                text = text.replace(f"'{word}'", f"QUOTED_{len(translated_quoted)}_")
 
+        # 전체 문장 번역
+        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}'")
+            
+            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
@@ -108,11 +136,11 @@ def result():
             return render_template('error.html', error="Please enter text to translate")
         
         try:
-            # 영어 입력 확인
-            is_eng = is_english(input_text)
+            # 입력 텍스트 정규화
+            input_text = normalize_quotes(input_text)
             
-            # 번��� 또는 직접 처리
-            english_text = input_text if is_eng else translate_korean_to_english(input_text)
+            # 번역 수행
+            english_text = translate_korean_to_english(input_text)
             if not english_text:
                 raise Exception("Translation failed")