Update app.py

app.py

@@ -5,6 +5,9 @@ import os
 from datetime import datetime, timedelta
 from huggingface_hub import InferenceClient
 
+from googletrans import Translator
+
+    
 API_KEY = os.getenv("SERPHOUSE_API_KEY")
 hf_client = InferenceClient("CohereForAI/c4ai-command-r-plus-08-2024", token=os.getenv("HF_TOKEN"))
 
@@ -116,23 +119,38 @@ COUNTRY_LOCATIONS = {
 
 MAJOR_COUNTRIES = list(COUNTRY_LOCATIONS.keys())
 
+
 def translate_query(query, country):
     try:
         if country in COUNTRY_LANGUAGES:
+            translator = Translator()
             target_lang = COUNTRY_LANGUAGES[country]
-            prompt = f"Translate the following English text to {target_lang} language. Only output the translated text without any explanations or quotes: {query}"
             
-            translated = hf_client.text_generation(
-                prompt,
-                max_new_tokens=250,
-                temperature=0.3
-            )
-            return translated.strip()
+            # 영어로 된 입력은 번역하지 않음
+            if is_english(query):
+                return query
+                
+            # 한국어 입력에 대해 South Korea가 선택된 경우 번역하지 않음    
+            if country == "South Korea" and is_korean(query):
+                return query
+                
+            # 구글 번역 실행    
+            translated = translator.translate(query, dest=target_lang)
+            print(f"Original query: {query}")
+            print(f"Translated query: {translated.text}")
+            return translated.text
+            
         return query
     except Exception as e:
         print(f"Translation error: {str(e)}")
         return query
 
+def is_english(text):
+    return all(ord(char) < 128 for char in text.replace(' ', ''))
+    
+def is_korean(text):
+    return any('\uAC00' <= char <= '\uD7A3' for char in text)
+    
 def search_serphouse(query, country, page=1, num_result=100):
     url = "https://api.serphouse.com/serp/live"