Update app.py

app.py

@@ -33,32 +33,56 @@ conversation_histories = {}
 
 def google_search(query):
     logger.info(f"Searching for query: {query}")
-    url = f"https://www.googleapis.com/customsearch/v1?key={API_KEY}&cx={CX}&q={query}&num=10"
-    logger.debug(f"Request URL: {url}")
+    
+    # 기본 매개변수 설정
+    params = {
+        'key': API_KEY,
+        'cx': CX,
+        'q': query,
+        'num': 10,
+        'sort': 'date:r:1m',  # 최근 1개월 내의 결과를 우선적으로 정렬
+        'dateRestrict': 'm1',  # 최근 1개월 내의 결과만 표시
+    }
+    
+    # 뉴스 검색을 위한 매개변수 추가
+    news_params = params.copy()
+    news_params['tbm'] = 'nws'
+    
+    url = "https://www.googleapis.com/customsearch/v1"
     
     try:
-        response = requests.get(url)
-        response.raise_for_status()
-        search_results = response.json()
-        logger.debug(f"API Response: {search_results}")
+        # 뉴스 검색
+        news_response = requests.get(url, params=news_params)
+        news_response.raise_for_status()
+        news_results = news_response.json()
+        
+        # 일반 검색
+        general_response = requests.get(url, params=params)
+        general_response.raise_for_status()
+        general_results = general_response.json()
+        
+        # 결과 합치기
+        combined_results = []
+        
+        # 뉴스 결과 추가 (최대 3개)
+        if 'items' in news_results:
+            combined_results.extend(news_results['items'][:3])
+        
+        # 일반 검색 결과 추가
+        if 'items' in general_results:
+            combined_results.extend(general_results['items'])
+        
+        # 최대 10개 결과로 제한
+        combined_results = combined_results[:10]
         
         results = []
-        if 'items' in search_results:
-            for item in search_results['items'][:10]:
-                title = item['title']
-                link = item['link']
-                snippet = item.get('snippet', '')
-                results.append(f"Title: {title}\nLink: {link}\nSnippet: {snippet}\n\n")
-        else:
-            logger.warning("No items found in search results")
-            if 'error' in search_results:
-                error_message = search_results['error']['message']
-                logger.error(f"API Error: {error_message}")
-                results.append(f"Error: {error_message}")
-            else:
-                results.append("No results found")
+        for item in combined_results:
+            title = item['title']
+            link = item['link']
+            snippet = item.get('snippet', '')
+            results.append(f"Title: {title}\nLink: {link}\nSnippet: {snippet}\n\n")
         
-        return '\n'.join(results[:10])
+        return '\n'.join(results)
     
     except requests.exceptions.RequestException as e:
         logger.error(f"Request failed: {e}")
@@ -163,7 +187,7 @@ async def generate_response(message):
     except Exception as e:
         logging.error(f"Error in generate_response: {e}")
         return f"{user_mention}, 죄송합니다. 응답을 생성하는 중 오류가 발생했습니다. 다시 시도해 주세요."
-        
+
 if __name__ == "__main__":
     # Discord 클라이언트 실행
     bot = MyClient()