Update app.py

app.py

@@ -104,10 +104,12 @@ MAJOR_COUNTRIES = list(COUNTRY_LOCATIONS.keys())
 
 def translate_query(query, country):
     try:
-        if country in COUNTRY_LANGUAGES:
-            # 입력 검색어 길이 제한
-            query = query[:100]  # 입력 검색어 길이 제한
+        # 한글 입력이고 South Korea가 선택된 경우 원문 그대로 반환
+        if country == "South Korea":
+            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}"
             
@@ -116,16 +118,17 @@ def translate_query(query, country):
                 max_new_tokens=50,
                 temperature=0.1
             )
-            # 번역된 텍스트 정제 및 길이 제한
-            translated = translated.strip()[:255]  # API 제한에 맞춰 255자로 제한
+            translated = translated.strip()[:255]
             print(f"Original query: {query}")
             print(f"Translated query: {translated}")
             return translated
-        return query[:255]  # 번역하지 않는 경우에도 길이 제한
+        return query[:255]
     except Exception as e:
         print(f"Translation error: {str(e)}")
         return query[:255]
 
+
+
 def search_serphouse(query, country, page=1, num_result=10):
     url = "https://api.serphouse.com/serp/live"
     
@@ -237,15 +240,20 @@ 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="검색어 번역 중...")
-        
+        progress(0, desc="검색 시작...")
+    
+    # 번역 처리 (South Korea 선택 시 번역하지 않음)
         translated_query = translate_query(query, country)
-        if translated_query != query:
+        if country == "South Korea":
+            translated_display_text = f"검색어: {query}"
+        elif translated_query != query:
             translated_display_text = f"원본 검색어: {query}\n번역된 검색어: {translated_query}"
         else:
             translated_display_text = f"검색어: {query}"
         
-        progress(0.2, desc="검색 시작...")
+        progress(0.2, desc="검색 중...")
+
+
         response_data = search_serphouse(query, country)
         error_message, articles = format_results_from_raw(response_data)