Update app.py

app.py

@@ -1,4 +1,4 @@
-import gradio as gr
+iimport gradio as gr
 import requests
 import json
 import os
@@ -6,12 +6,29 @@ from datetime import datetime, timedelta
 
 API_KEY = os.getenv("SERPHOUSE_API_KEY")
 
-MAJOR_COUNTRIES = [
-    "United States", "United Kingdom", "Canada", "Australia", "Germany",
-    "France", "Japan", "South Korea", "China", "India",
-    "Brazil", "Mexico", "Russia", "Italy", "Spain",
-    "Netherlands", "Singapore", "Hong Kong"
-]
+# 국가와 country_code 매핑
+COUNTRY_CODES = {
+    "United States": "US",
+    "United Kingdom": "GB",
+    "Canada": "CA",
+    "Australia": "AU",
+    "Germany": "DE",
+    "France": "FR",
+    "Japan": "JP",
+    "South Korea": "KR",
+    "China": "CN",
+    "India": "IN",
+    "Brazil": "BR",
+    "Mexico": "MX",
+    "Russia": "RU",
+    "Italy": "IT",
+    "Spain": "ES",
+    "Netherlands": "NL",
+    "Singapore": "SG",
+    "Hong Kong": "HK"
+}
+
+MAJOR_COUNTRIES = list(COUNTRY_CODES.keys())
 
 def search_serphouse(query, country, page=1, num_result=100):
     url = "https://api.serphouse.com/serp/live"
@@ -23,8 +40,8 @@ def search_serphouse(query, country, page=1, num_result=100):
     payload = {
         "data": {
             "q": query,
-            "domain": "google.com",  # 기본 도메인 사용
-            "loc": country,    # 국가 이름을 location으로 사용
+            "domain": "google.com",
+            "country_code": COUNTRY_CODES.get(country, "US"),  # country_code 사용
             "lang": "en",
             "device": "desktop",
             "serp_type": "news",