Update src/main.py

src/main.py

@@ -31,23 +31,30 @@ def is_korean(text):
 
 def normalize_quotes(text):
     """따옴표 형식을 정규화하는 함수"""
-    # 먼저 모든 따옴표를 정리
-    text = clean_quotes(text)
+    # 연속된 따옴표 제거
+    text = re.sub(r"'+", "'", text)
+    # 불필요한 공백 제거
+    text = re.sub(r'\s+', ' ', text).strip()
     
     if is_korean(text):
-        # 한글 문장의 경우, 첫 번째 명사구만 따옴표로 처리
+        # 한글 문장의 경우
         words = text.split()
-        first_word = words[0]
-        if not (first_word.startswith("'") and first_word.endswith("'")):
+        if words:
+            # 첫 번째 단어에서 따옴표 제거 후 다시 추가
+            first_word = words[0].replace("'", "")
             words[0] = f"'{first_word}'"
-        return ' '.join(words)
+            # 나머지 단어들에서 따옴표 제거
+            words[1:] = [w.replace("'", "") for w in words[1:]]
+            return ' '.join(words)
     else:
         # 영어 문장의 경우, 첫 단어만 따옴표로 처리
         words = text.split()
         if words:
-            if not (words[0].startswith("'") and words[0].endswith("'")):
-                words[0] = f"'{words[0]}'"
-        return ' '.join(words)
+            # 첫 번째 단어에서 따옴표 제거 후 다시 추가
+            first_word = words[0].replace("'", "")
+            words[0] = f"'{first_word}'"
+            return ' '.join(words)
+    return text
 
 def find_quoted_words(text):
     """작은따옴표로 묶인 단어들을 찾는 함수"""
@@ -75,41 +82,29 @@ def translate_korean_to_english(text):
         
         # 영어 입력 확인
         if is_english(text):
-            # 영어 입력의 경우 첫 단어만 따옴표로 처리
-            words = text.split()
-            if words:
-                # 이미 따옴표가 있는 경우는 그대로, 없는 경우 추가
-                if not (words[0].startswith("'") and words[0].endswith("'")):
-                    words[0] = f"'{words[0]}'"
-                return ' '.join(words)
             return text
 
         # 한글 입력 처리
         # 따옴표로 묶인 단어 찾기
-        quoted_words = re.findall(r"'([^']*)'", text)
-        if not quoted_words:
-            # 따옴표로 묶인 단어가 없는 경우, 첫 단어를 따옴표로 묶기
-            words = text.split()
-            if words:
-                text = f"'{words[0]}'" + text[len(words[0]):]
-                quoted_words = [words[0]]
-
-        # 첫 번째 따옴표 단어 번역
+        quoted_word = re.findall(r"'([^']*)'", text)[0]
+        
+        # 따옴표로 묶인 단어 먼저 번역
         url = "https://translate.googleapis.com/translate_a/single"
         params = {
             "client": "gtx",
             "sl": "ko",
             "tl": "en",
             "dt": "t",
-            "q": quoted_words[0]
+            "q": quoted_word
         }
         response = requests.get(url, params=params)
         if response.status_code == 200:
             translated_word = response.json()[0][0][0].upper()
-            # 임시 마커로 대체
-            text = text.replace(f"'{quoted_words[0]}'", "QUOTED_WORD_MARKER")
-
-            # 전체 문장 번역
+            
+            # 원본 텍스트에서 따옴표로 묶인 부분을 임시 마커로 대체
+            text = text.replace(f"'{quoted_word}'", "QUOTED_WORD_MARKER")
+            
+            # 나머지 문장 번역
             params["q"] = text
             response = requests.get(url, params=params)
             if response.status_code == 200:
@@ -123,9 +118,6 @@ def translate_korean_to_english(text):
         print(f"Translation error: {e}")
         return text
 
-
-
-
 @app.route('/')
 def index():
     return render_template('index.html', title=app.config['TITLE'])