Update app.py

app.py

@@ -102,16 +102,32 @@ COUNTRY_LOCATIONS = {
 
 MAJOR_COUNTRIES = list(COUNTRY_LOCATIONS.keys())
 
+def is_english(text):
+    # 영어로만 구성되어 있는지 확인
+    return all(ord(char) < 128 for char in text.replace(' ', ''))
+
 def translate_query(query, country):
     try:
-        # 한글 입력이고 South Korea가 선택된 경우 원문 그대로 반환
+        # 영어 입력인 경우 번역하지 않고 그대로 사용
+        if is_english(query):
+            print(f"English query detected, using original: {query}")
+            return query[:255]
+            
+        # 한글 입력이고 South Korea가 선택된 경우
         if country == "South Korea":
-            return query[:255]  # 길이 제한만 적용
+            return query[:255]
             
         if country in COUNTRY_LANGUAGES:
             query = query[:100]
             target_lang = COUNTRY_LANGUAGES[country]
-            prompt = f"Translate only this text to {target_lang} language without any explanation. Output only translated text: {query}"
+            
+            # 번역 프롬프트 개선
+            prompt = f"""Translate this text to {target_lang} language. 
+            For Japanese, use Kanji and Kana.
+            For Chinese, use Simplified Chinese.
+            For Korean, use Hangul.
+            Only output the translated text without any explanation.
+            Text to translate: {query}"""
             
             translated = hf_client.text_generation(
                 prompt,
@@ -242,9 +258,11 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as
     def search_and_display(query, country, articles_state, progress=gr.Progress()):
         progress(0, desc="검색 시작...")
     
-    # 번역 처리 (South Korea 선택 시 번역하지 않음)
+    # 번역 및 표시 텍스트 처리
         translated_query = translate_query(query, country)
-        if country == "South Korea":
+        if is_english(query):
+            translated_display_text = f"영어 검색어: {query}"
+        elif country == "South Korea":
             translated_display_text = f"검색어: {query}"
         elif translated_query != query:
             translated_display_text = f"원본 검색어: {query}\n번역된 검색어: {translated_query}"