Update src/main.py

src/main.py

@@ -61,6 +61,48 @@ def spell_out_word(word):
     """단어를 개별 알파벳으로 분리하는 함수"""
     return ' '.join(list(word.lower()))
 
+def translate_korean_text(text):
+    """한글 전용 번역 함수"""
+    try:
+        # 따옴표로 묶인 단어 찾기
+        quoted_match = re.search(r"'([^']*)'", text)
+        if not quoted_match:
+            return text
+
+        # 1. 따옴표 안의 단어만 먼저 번역
+        korean_word = quoted_match.group(1)
+        url = "https://translate.googleapis.com/translate_a/single"
+        params = {
+            "client": "gtx",
+            "sl": "ko",
+            "tl": "en",
+            "dt": "t",
+            "q": korean_word
+        }
+        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. 번역된 부분들 조합
+        return f"'{proper_noun}' {remaining_translation}"
+
+    except Exception as e:
+        print(f"Korean translation error: {e}")
+        return text
 
 def translate_korean_to_english(text):
     """전체 텍스트 번역 함수"""
@@ -70,56 +112,16 @@ def translate_korean_to_english(text):
         
         # 영어 입력 확인
         if is_english(text):
-            # 영어 입력의 경우 따옴표 처리만 하고 그대로 반환
+            # 기존 영어 처리 방식 유지
             quoted_match = re.search(r"'([^']*)'", text)
             if quoted_match:
                 quoted_word = quoted_match.group(1).upper()
                 text = re.sub(r"'[^']*'", f"'{quoted_word}'", text, 1)
             return 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",
-                "tl": "en",
-                "dt": "t",
-                "q": quoted_word
-            }
-            response = requests.get(url, params=params)
-            if response.status_code == 200:
-                translated_word = response.json()[0][0][0].upper()
-                
-                # 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])
+        # 한글 입력인 경우 새로운 함수로 처리
+        if is_korean(text):
+            return translate_korean_text(text)
 
         return text
     except Exception as e:
@@ -127,6 +129,7 @@ def translate_korean_to_english(text):
         return text
 
 
+
 @app.route('/')
 def index():
     return render_template('index.html', title=app.config['TITLE'])
@@ -278,4 +281,4 @@ def download_video(gloss_sentence):
         return f"Error downloading video: {str(e)}", 500
 
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860, debug=True)
+    app.run(host="0.0.0.0", port=7860, debug=True)