Update src/main.py

src/main.py

@@ -81,9 +81,12 @@ def translate_korean_to_english(text):
         quoted_match = re.search(r"'([^']*)'", text)
         if quoted_match:
             quoted_word = quoted_match.group(1)
+            rest_of_text = text[text.find("'", text.find(quoted_word) + len(quoted_word)) + 1:].strip()
             
             # 따옴표로 묶인 단어 먼저 번역
             url = "https://translate.googleapis.com/translate_a/single"
+            
+            # 1. 고유명사 번역
             params = {
                 "client": "gtx",
                 "sl": "ko",
@@ -95,17 +98,28 @@ def translate_korean_to_english(text):
             if response.status_code == 200:
                 translated_word = response.json()[0][0][0].upper()
                 
-                # 원본 텍스트에서 따옴표로 묶인 부분을 임시 마커로 대체
-                text = text.replace(f"'{quoted_word}'", "QUOTED_WORD_MARKER")
-                
-                # 나머지 문장 번역
-                params["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])
-                    # 마커를 번역된 단어로 대체
-                    translated_text = translated_text.replace("QUOTED_WORD_MARKER", f"'{translated_word}'")
-                    return translated_text
+                # 2. 나머지 문장 번역
+                if rest_of_text:
+                    params["q"] = rest_of_text
+                    response = requests.get(url, params=params)
+                    if response.status_code == 200:
+                        rest_translated = ' '.join(item[0] for item in response.json()[0] if item[0])
+                        return f"'{translated_word}' {rest_translated}"
+                else:
+                    return f"'{translated_word}'"
+
+        # 따옴표가 없는 경우 전체 문장 번역
+        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:
+            return ' '.join(item[0] for item in response.json()[0] if item[0])
 
         return text
     except Exception as e:
@@ -113,8 +127,6 @@ def translate_korean_to_english(text):
         return text
 
 
-
-
 @app.route('/')
 def index():
     return render_template('index.html', title=app.config['TITLE'])