Update src/main.py

src/main.py

@@ -47,49 +47,55 @@ def translate_quoted_word(word):
         print(f"Word translation error: {e}")
         return word
 
+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:
-        # 1. 따옴표로 묶인 단어 찾기
-        quoted_words = re.findall(r"'([^']*)'", text)
+        # 입력이 영어인지 확인
+        if is_english(text):
+            return process_english_input(text)
+
+        # 한국어 입력 처리
+        # 1. 따옴표로 묶인 부분을 찾아서 따로 번역
+        pattern = r"'([^']*)'|([^']+)"
+        parts = re.findall(pattern, text)
+        translated_parts = []
         
-        # 2. 각 따옴표 단어 개별 번역
-        translated_words = {}
-        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_words[word] = translated
-                # 임시로 원본 텍스트의 따옴표 부분을 특수 마커로 대체
-                text = text.replace(f"'{word}'", "PROPER_NOUN_MARKER")
-
-        # 3. 전체 문장 번역
-        params = {
-            "client": "gtx",
-            "sl": "ko",
-            "tl": "en",
-            "dt": "t",
-            "q": text
-        }
-        response = requests.get(url, params=params)
-        if response.status_code == 200:
-            full_translation = response.json()[0][0][0]
-            
-            # 4. 번역된 문장에서 마커를 번역된 고유명사로 대체
-            for original, translated in translated_words.items():
-                full_translation = full_translation.replace("PROPER_NOUN_MARKER", f"'{translated}'", 1)
-            
-            return full_translation
-        else:
-            raise Exception(f"Translation API returned status code: {response.status_code}")
+        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
+
     except Exception as e:
         print(f"Translation error: {e}")
         return text
@@ -102,15 +108,18 @@ def result():
             return render_template('error.html', error="Please enter text to translate")
         
         try:
-            # 번역 수행
-            english_text = translate_korean_to_english(input_text)
+            # 영어 입력 확인
+            is_eng = is_english(input_text)
+            
+            # 번역 또는 직접 처리
+            english_text = input_text if is_eng else translate_korean_to_english(input_text)
             if not english_text:
                 raise Exception("Translation failed")
             
-            # 따옴표로 묶인 단어 추출 (번역된 영어 텍스트에서)
+            # 따옴표로 묶인 단어 추출
             quoted_words = [word.strip("'") for word in re.findall(r"'([^']*)'", english_text)]
             
-            # 번역된 텍스트에서 따옴표 제거하고 ASL 변환
+            # ASL 변환을 위해 따옴표 제거
             clean_english = re.sub(r"'([^']*)'", r"\1", english_text)
             eng_to_asl_translator = NlpSpacyBaseTranslator(sentence=clean_english)
             generated_gloss = eng_to_asl_translator.translate_to_gloss()
@@ -138,13 +147,13 @@ def result():
             i = 0
             while i < len(processed_gloss):
                 if processed_gloss[i] == 'FINGERSPELL-START':
-                    final_gloss.extend(processed_gloss[i:i+2])  # START와 첫 글자 추가
+                    final_gloss.extend(processed_gloss[i:i+2])
                     i += 2
                     while i < len(processed_gloss) and processed_gloss[i] != 'FINGERSPELL-END':
                         final_gloss.append(processed_gloss[i])
                         i += 1
                     if i < len(processed_gloss):
-                        final_gloss.append(processed_gloss[i])  # END 추가
+                        final_gloss.append(processed_gloss[i])
                         i += 1
                 else:
                     word = processed_gloss[i]