Update app.py

app.py

@@ -136,12 +136,17 @@ def translate_query(query, country):
 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}")
+    
     payload = {
         "data": {
-            "q": query,  # 원본 검색어 사용
+            "q": query,
             "domain": "google.com",
             "loc": COUNTRY_LOCATIONS.get(country, "United States"),
-            "lang": "en",  # 기본 영어로 설정
+            "lang": "en",
             "device": "desktop",
             "serp_type": "news",
             "page": "1",
@@ -161,15 +166,18 @@ def search_serphouse(query, country, page=1, num_result=10):
         print("Response status:", response.status_code)
         
         response.raise_for_status()
-        return response.json()
+        return {"results": response.json(), "translated_query": translated_query}
     except requests.RequestException as e:
-        return {"error": f"Error: {str(e)}"}
+        return {"error": f"Error: {str(e)}", "translated_query": query}
 
-def format_results_from_raw(results, translated_query):
-    if isinstance(results, dict) and "error" in results:
-        return "Error: " + results["error"], []
+def format_results_from_raw(response_data):
+    if "error" in response_data:
+        return "Error: " + response_data["error"], []
 
     try:
+        results = response_data["results"]
+        translated_query = response_data["translated_query"]
+        
         news_results = results.get('results', {}).get('results', {}).get('news', [])
         if not news_results:
             return "검색 결과가 없습니다.", []
@@ -191,8 +199,8 @@ def format_results_from_raw(results, translated_query):
         return f"결과 처리 중 오류 발생: {str(e)}", []
 
 def serphouse_search(query, country):
-    results, translated_query = search_serphouse(query, country)
-    return format_results_from_raw(results, translated_query)
+    response_data = search_serphouse(query, country)
+    return format_results_from_raw(response_data)
 
 css = """
 footer {visibility: hidden;}