Update app.py

app.py

@@ -22,9 +22,27 @@ MAJOR_COUNTRIES = [
     "Indonesia", "Philippines", "Vietnam", "Pakistan", "Bangladesh"
 ]
 
-def search_serphouse(query, country, page=1, num_result=100):  # num_result를 100으로 변경
+def search_serphouse(query, country, page=1, num_result=100):
     url = "https://api.serphouse.com/serp/live"
 
+    # 국가별 도메인 매핑
+    country_domains = {
+        "United States": "google.com",
+        "United Kingdom": "google.co.uk",
+        "Canada": "google.ca",
+        "Australia": "google.com.au",
+        "Germany": "google.de",
+        "France": "google.fr",
+        "Japan": "google.co.jp",
+        "South Korea": "google.co.kr",
+        "China": "google.com.hk",  # 중국은 홍콩 도메인 사용
+        "India": "google.co.in",
+        # 필요한 다른 국가들의 도메인을 추가할 수 있습니다.
+    }
+
+    # 선택된 국가의 도메인 가져오기 (없으면 기본값 google.com 사용)
+    domain = country_domains.get(country, "google.com")
+
     now = datetime.utcnow()
     yesterday = now - timedelta(days=1)
     date_range = f"{yesterday.strftime('%Y-%m-%d')},{now.strftime('%Y-%m-%d')}"
@@ -32,8 +50,8 @@ def search_serphouse(query, country, page=1, num_result=100):  # num_result를 1
     payload = {
         "data": {
             "q": query,
-            "domain": "google.com",
-            "loc": country,  # 국가 이름을 직접 사용합니다.
+            "domain": domain,  # 국가별 도메인 사용
+            "loc": country,    # 국가 이름
             "lang": "en",
             "device": "desktop",
             "serp_type": "news",
@@ -271,4 +289,5 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as
             show_progress=True  # Gradio의 기본 로딩 표시를 끕니다.
         )
 
-iface.launch(auth=("gini", "pick"))
+#iface.launch(auth=("gini", "pick"))
+iface.launch()