Update app.py

app.py

@@ -122,37 +122,46 @@ MAJOR_COUNTRIES = list(COUNTRY_LOCATIONS.keys())
 
 def translate_query(query, country):
     try:
+        # 번역기 초기화
+        translator = Translator()
+        
+        # 영어 입력은 모든 국가에 대해 그대로 사용
+        if is_english(query):
+            print(f"영어 검색어 감지 - 원본 사용: {query}")
+            return query
+            
+        # 선택된 국가가 번역 지원 국가인 경우    
         if country in COUNTRY_LANGUAGES:
-            translator = Translator()
             target_lang = COUNTRY_LANGUAGES[country]
             
-            # 1. 영어로 입력된 경우 - 모든 국가에 대해 영어 그대로 사용
-            if is_english(query):
-                print(f"영어 검색어 감지: {query}")
+            # South Korea 선택시 한글 입력은 그대로 사용
+            if country == "South Korea" and is_korean(query):
+                print(f"한국어 검색어 - 원본 사용: {query}")
                 return query
                 
-            # 2. 한글로 입력된 경우
-            if is_korean(query):
-                # South Korea 선택시 한글 그대로 사용
-                if country == "South Korea":
-                    print(f"한글 검색어 유지: {query}")
-                    return query
-                # 다른 국가 선택시 해당 국가 언어로 번역    
-                else:
-                    translated = translator.translate(query, dest=target_lang)
-                    print(f"한글 검색어를 {country} 언어로 번역: {translated.text}")
-                    return translated.text
-            
-            # 3. 기타 언어로 입력된 경우 - 선택된 국가의 언어로 번역
+            # 다른 모든 경우 선택된 국가 언어로 번역
             translated = translator.translate(query, dest=target_lang)
-            print(f"입력 검색어를 {country} 언어로 번역: {translated.text}")
+            print(f"번역 결과: {query} -> {translated.text} ({country})")
             return translated.text
             
         return query
+        
     except Exception as e:
-        print(f"Translation error: {str(e)}")
+        print(f"번역 오류: {str(e)}")
         return query
 
+def is_english(text):
+    return all(ord(char) < 128 for char in text.replace(' ', '').replace('-', ''))
+    
+def is_korean(text):
+    return any('\uAC00' <= char <= '\uD7A3' for char in text)
+
+# 사용 예시:
+# query = "삼성전자"
+# country = "Japan"
+# translated = translate_query(query, country)
+# print(f"입력: {query}, 국가: {country}, 번역: {translated}")
+
 def is_english(text):
     return all(ord(char) < 128 for char in text.replace(' ', ''))