Update app.py

app.py

@@ -8,8 +8,7 @@ from huggingface_hub import InferenceClient
 API_KEY = os.getenv("SERPHOUSE_API_KEY")
 hf_client = InferenceClient("CohereForAI/c4ai-command-r-plus-08-2024", token=os.getenv("HF_TOKEN"))
 
-
-
+# 국가별 코드 매핑
 COUNTRY_CODES = {
     "United States": "US",
     "United Kingdom": "GB",
@@ -20,7 +19,7 @@ COUNTRY_CODES = {
     "Japan": "JP",
     "South Korea": "KR",
     "China": "CN",
-    "Taiwan": "TW",  # 대만 추가
+    "Taiwan": "TW",
     "India": "IN",
     "Brazil": "BR",
     "Mexico": "MX",
@@ -34,125 +33,29 @@ COUNTRY_CODES = {
     "Malaysia": "MY",
     "Philippines": "PH",
     "Thailand": "TH",
-    "Vietnam": "VN",
-    "Belgium": "BE",
-    "Denmark": "DK",
-    "Finland": "FI",
-    "Ireland": "IE",
-    "Norway": "NO",
-    "Poland": "PL",
-    "Sweden": "SE",
-    "Switzerland": "CH",
-    "Austria": "AT",
-    "Czech Republic": "CZ",
-    "Greece": "GR",
-    "Hungary": "HU",
-    "Portugal": "PT",
-    "Romania": "RO",
-    "Turkey": "TR",
-    "Israel": "IL",
-    "Saudi Arabia": "SA",
-    "United Arab Emirates": "AE",
-    "South Africa": "ZA",
-    "Argentina": "AR",
-    "Chile": "CL",
-    "Colombia": "CO",
-    "Peru": "PE",
-    "Venezuela": "VE",
-    "New Zealand": "NZ",
-    "Bangladesh": "BD",
-    "Pakistan": "PK",
-    "Egypt": "EG",
-    "Morocco": "MA",
-    "Nigeria": "NG",
-    "Kenya": "KE",
-    "Ukraine": "UA",
-    "Croatia": "HR",
-    "Slovakia": "SK",
-    "Bulgaria": "BG",
-    "Serbia": "RS",
-    "Estonia": "EE",
-    "Latvia": "LV",
-    "Lithuania": "LT",
-    "Slovenia": "SI",
-    "Luxembourg": "LU",
-    "Malta": "MT",
-    "Cyprus": "CY",
-    "Iceland": "IS"
+    "Vietnam": "VN"
 }
 
+MAJOR_COUNTRIES = list(COUNTRY_CODES.keys())
 
 def is_english(text):
-    # 영어로만 구성되어 있는지 확인
     return all(ord(char) < 128 for char in text.replace(' ', ''))
-    
-COUNTRY_LANGUAGES = {
-    "South Korea": "ko",
-    "Japan": "ja",
-    "China": "zh",
-    "Taiwan": "zh-tw",  # 대만어(번체 중국어) 추가
-    "Russia": "ru",
-    "France": "fr",
-    "Germany": "de",
-    "Spain": "es",
-    "Italy": "it",
-    "Netherlands": "nl",
-    "Portugal": "pt",
-    "Thailand": "th",
-    "Vietnam": "vi",
-    "Indonesia": "id",
-    "Malaysia": "ms",
-    "Saudi Arabia": "ar",
-    "United Arab Emirates": "ar",
-    "Egypt": "ar",
-    "Morocco": "ar",
-    "Greece": "el",
-    "Poland": "pl",
-    "Czech Republic": "cs",
-    "Hungary": "hu",
-    "Turkey": "tr",
-    "Romania": "ro",
-    "Bulgaria": "bg",
-    "Croatia": "hr",
-    "Serbia": "sr",
-    "Slovakia": "sk",
-    "Slovenia": "sl",
-    "Estonia": "et",
-    "Latvia": "lv",
-    "Lithuania": "lt",
-    "Ukraine": "uk",
-    "Israel": "he",
-    "Bangladesh": "bn",
-    "Pakistan": "ur",
-    "Finland": "fi",
-    "Denmark": "da",
-    "Norway": "no",
-    "Sweden": "sv",
-    "Iceland": "is",
-    "Philippines": "fil",
-    "Brazil": "pt-br",
-    "Argentina": "es-ar",
-    "Chile": "es-cl",
-    "Colombia": "es-co",
-    "Peru": "es-pe",
-    "Venezuela": "es-ve"
-}
 
-# 번역 프롬프트 수정
 def translate_query(query, country):
     try:
+        # 영어 입력인 경우 번역하지 않음
         if is_english(query):
             print(f"English query detected, using original: {query}")
             return query[:255]
             
+        # 한글 입력이고 South Korea가 선택된 경우 번역하지 않음
         if country == "South Korea":
             return query[:255]
             
-        if country in COUNTRY_LANGUAGES:
+        if country in COUNTRY_CODES:
             query = query[:100]
-            target_lang = COUNTRY_LANGUAGES[country]
+            target_lang = COUNTRY_CODES[country]
             
-            # 번역 프롬프트 개선
             prompt = f"""Translate this text to {target_lang} language. 
             For Japanese, use Kanji and Kana.
             For Chinese (China), use Simplified Chinese.
@@ -175,15 +78,10 @@ def translate_query(query, country):
         print(f"Translation error: {str(e)}")
         return query[:255]
 
-
-
-# MAJOR_COUNTRIES 정의 수정
-MAJOR_COUNTRIES = list(COUNTRY_CODES.keys())  # COUNTRY_LOCATIONS 대신 COUNTRY_CODES 사용
-
-# search_serphouse 함수 수정
 def search_serphouse(query, country, page=1, num_result=10):
     url = "https://api.serphouse.com/serp/live"
     
+    # 검색어 번역
     translated_query = translate_query(query, country)
     print(f"Original query: {query}")
     print(f"Translated query: {translated_query}")
@@ -192,14 +90,12 @@ def search_serphouse(query, country, page=1, num_result=10):
         "data": {
             "q": translated_query,
             "domain": "google.com",
-            "loc": country,
-            "lang": COUNTRY_LANGUAGES.get(country, "en"),
+            "country_code": COUNTRY_CODES.get(country, "US"),
+            "lang": "en",
             "device": "desktop",
-            "serp_type": "web",
+            "serp_type": "news",
             "page": "1",
-            "num": "10",
-            "verbatim": "0",
-            "gfilter": "0"
+            "num": "10"
         }
     }
 
@@ -213,7 +109,6 @@ def search_serphouse(query, country, page=1, num_result=10):
         response = requests.post(url, json=payload, headers=headers)
         print("Request payload:", json.dumps(payload, indent=2, ensure_ascii=False))
         print("Response status:", response.status_code)
-        print("Full response content:", response.text)  # 전체 응답 내용 출력
         
         response.raise_for_status()
         return {"results": response.json(), "translated_query": translated_query}
@@ -231,13 +126,12 @@ def format_results_from_raw(response_data):
         results = response_data["results"]
         translated_query = response_data["translated_query"]
         
-        # 응답 구조 변경에 따른 수정
-        search_results = results.get('results', {}).get('organic_results', [])  # organic_results로 변경
-        if not search_results:
+        news_results = results.get('results', {}).get('organic_results', [])
+        if not news_results:
             return "검색 결과가 없습니다.", []
 
         articles = []
-        for idx, result in enumerate(search_results, 1):
+        for idx, result in enumerate(news_results, 1):
             articles.append({
                 "index": idx,
                 "title": result.get("title", "제목 없음"),
@@ -250,7 +144,6 @@ def format_results_from_raw(response_data):
             })
         return "", articles
     except Exception as e:
-        print(f"Format error: {str(e)}")  # 디버깅을 위한 에러 출력 추가
         return f"결과 처리 중 오류 발생: {str(e)}", []
 
 def serphouse_search(query, country):
@@ -296,9 +189,8 @@ 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="검색어 번역 중...")
+        
         translated_query = translate_query(query, country)
         if is_english(query):
             translated_display_text = f"영어 검색어: {query}"
@@ -310,8 +202,6 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as
             translated_display_text = f"검색어: {query}"
         
         progress(0.2, desc="검색 중...")
-
-
         response_data = search_serphouse(query, country)
         error_message, articles = format_results_from_raw(response_data)
         
@@ -354,19 +244,12 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as
         outputs.append(gr.update(visible=False))
         return outputs
 
-    # 검색 버튼 클릭 시 업데이트될 출력 컴포넌트 목록
-    search_outputs = [translated_display, gr.Markdown(visible=False)]  # 번역 결과 표시 컴포넌트
+    search_outputs = [translated_display, gr.Markdown(visible=False)]
     for comp in article_components:
-        search_outputs.extend([
-            comp['group'],
-            comp['title'],
-            comp['image'],
-            comp['snippet'],
-            comp['info']
-        ])
+        search_outputs.extend([comp['group'], comp['title'], comp['image'],
+                             comp['snippet'], comp['info']])
     search_outputs.extend([articles_state, status_message])
 
-    # 검색 버튼 클릭 이벤트 설정
     search_button.click(
         fn=search_and_display,
         inputs=[query, country, articles_state],