Update app.py

app.py

@@ -77,32 +77,35 @@ def create_deepseek_interface():
             
             # 한글 검색어인지 확인하고 적절한 설정 선택
             is_korean = any('\uAC00' <= c <= '\uD7A3' for c in search_query)
-            lang = "ko" if is_korean else "en"
-            loc = "kr" if is_korean else "us"
             
-            # MoneyRadar 방식 정확히 따르기 - serp_type을 news로 변경하고 날짜 범위 추가
+            # MoneyRadar 방식 정확히 따르기
             from datetime import datetime, timedelta
             now = datetime.utcnow()
             yesterday = now - timedelta(days=1)
             date_range = f"{yesterday.strftime('%Y-%m-%d')},{now.strftime('%Y-%m-%d')}"
             
-            # SerpHouse API 호출 실행 - POST 메서드 사용 (MoneyRadar 코드 참고)
+            # SerpHouse API 호출 실행 - POST 메서드 사용 (MoneyRadar 코드 정확히 따르기)
             url = "https://api.serphouse.com/serp/live"
             
-            # MoneyRadar 방식으로 페이로드 구성
+            # 국가 설정
+            country = "KOREA" if is_korean else "United States"
+            
+            # COUNTRY_LOCATIONS과 COUNTRY_LANGUAGES 정의 (MoneyRadar에서 사용되는 매핑)
+            COUNTRY_LOCATIONS = {"United States": "United States", "KOREA": "kr"}
+            COUNTRY_LANGUAGES = {"United States": "en", "KOREA": "ko"}
+            
+            # MoneyRadar 방식으로 페이로드 구성 (100개 결과 요청)
             payload = {
-                "data": {
-                    "q": search_query.replace(',', ' ').strip(),
-                    "domain": "google.com",
-                    "loc": loc,
-                    "lang": lang,
-                    "device": "desktop",
-                    "serp_type": "news",  # web에서 news로 변경
-                    "page": "1",
-                    "num": "5",
-                    "date_range": date_range,  # 날짜 범위 추가
-                    "sort_by": "date"  # 날짜순 정렬 추가
-                }
+                "q": search_query.replace(',', ' ').strip(),
+                "domain": "google.com",
+                "loc": COUNTRY_LOCATIONS.get(country, "United States"),
+                "lang": COUNTRY_LANGUAGES.get(country, "en"),
+                "device": "desktop",
+                "serp_type": "news",
+                "page": "1",
+                "num": "100",
+                "date_range": date_range,
+                "sort_by": "date"
             }
             
             headers = {
@@ -115,14 +118,18 @@ def create_deepseek_interface():
             print(f"요청 방식: POST, 페이로드: {payload}")
             
             # POST 요청 수행 (세션과 재시도 로직 추가)
+            import requests
+            from requests.adapters import HTTPAdapter
+            from requests.packages.urllib3.util.retry import Retry
+            
             session = requests.Session()
-            retries = requests.packages.urllib3.util.retry.Retry(
+            retries = Retry(
                 total=5,
                 backoff_factor=1,
                 status_forcelist=[500, 502, 503, 504, 429],
                 allowed_methods=["POST"]
             )
-            adapter = requests.adapters.HTTPAdapter(max_retries=retries)
+            adapter = HTTPAdapter(max_retries=retries)
             session.mount('http://', adapter)
             session.mount('https://', adapter)
             
@@ -136,40 +143,47 @@ def create_deepseek_interface():
             response.raise_for_status()
             
             print(f"SerpHouse API 응답 상태 코드: {response.status_code}")
-            search_results = response.json()
+            response_data = response.json()
             
-            # MoneyRadar의 결과 처리 방식 분석 및 적용
+            # MoneyRadar 방식으로 결과 반환
+            search_results = {
+                "results": response_data,
+                "translated_query": search_query
+            }
+            
+            # 응답 구조 로깅
             print(f"응답 구조: {list(search_results.keys()) if isinstance(search_results, dict) else '딕셔너리 아님'}")
             
-            # 검색 결과 파싱 및 포맷팅 
+            # 검색 결과 파싱 및 포맷팅 (MoneyRadar format_results_from_raw 함수 참고)
             formatted_results = []
             formatted_results.append(f"검색어: {search_query}\n\n")
             
             # 결과 구조 파싱 (MoneyRadar와 동일하게)
             if "results" in search_results and "results" in search_results["results"]:
-                # 뉴스 결과 파싱
-                news_results = search_results["results"]["results"].get("news", [])
-                if news_results:
-                    for result in news_results[:5]:
-                        title = result.get("title", "제목 없음")
-                        snippet = result.get("snippet", "내용 없음")
-                        url = result.get("url", result.get("link", "#"))
-                        source = result.get("source", result.get("channel", "알 수 없음"))
-                        time = result.get("time", result.get("date", "시간 정보 없음"))
+                if "results" in search_results["results"]["results"]:
+                    # 뉴스 결과 파싱
+                    news_results = search_results["results"]["results"]["results"].get("news", [])
+                    if news_results:
+                        for result in news_results[:5]:  # 상위 5개만 표시
+                            title = result.get("title", "제목 없음")
+                            snippet = result.get("snippet", "내용 없음")
+                            url = result.get("url", result.get("link", "#"))
+                            source = result.get("source", result.get("channel", "알 수 없음"))
+                            time = result.get("time", result.get("date", "시간 정보 없음"))
+                            
+                            formatted_results.append(
+                                f"제목: {title}\n"
+                                f"출처: {source}\n"
+                                f"시간: {time}\n"
+                                f"내용: {snippet}\n"
+                                f"링크: {url}\n\n"
+                            )
                         
-                        formatted_results.append(
-                            f"제목: {title}\n"
-                            f"출처: {source}\n"
-                            f"시간: {time}\n"
-                            f"내용: {snippet}\n"
-                            f"링크: {url}\n\n"
-                        )
-                    
-                    print(f"검색 결과 {len(news_results)}개 찾음")
-                    return "".join(formatted_results)
+                        print(f"검색 결과 {len(news_results)}개 찾음")
+                        return "".join(formatted_results)
             
             print("검색 결과 없음 또는 응답 형식 불일치")
-            print(f"응답 내용 미리보기: {str(search_results)[:300]}...")
+            print(f"응답 구조: {search_results.keys() if hasattr(search_results, 'keys') else '불명확한 구조'}")
             return f"검색어 '{search_query}'에 대한 검색 결과가 없거나 API 응답 형식이 예상과 다릅니다."
                 
         except Exception as e: