Update app.py

app.py

@@ -1,29 +1,35 @@
-from googleapiclient.discovery import build
+from serpapi import GoogleSearch
 import gradio as gr
 
-# API 키와 검색 엔진 ID 설정
-API_KEY = 'AIzaSyDUz3wkGal0ewRtPlzeMit88bV4hS4ZIVY'
-CSE_ID = '56b34994f47704ddd'
+# SerpAPI 설정
+SERP_API_KEY = "56b76bc0db7f66e70958810f3486e99a7ad4fc9b4ad0719e34478b20d2f7ec4f"
 
-def multi_search(query):
-    # Google Custom Search JSON API를 사용하여 웹 검색
-    service = build("customsearch", "v1", developerKey=API_KEY)
-    web_search_result = service.cse().list(q=query, cx=CSE_ID).execute()
-    web_results = [item['link'] for item in web_search_result.get('items', [])]
+def multi_search_serpapi(query):
+    # SerpAPI를 사용하여 Google 웹 검색
+    params = {
+        "engine": "google",
+        "q": query,
+        "api_key": SERP_API_KEY
+    }
+    search = GoogleSearch(params)
+    results = search.get_dict()
+    organic_results = results.get("organic_results", [])
+    web_results = [(result['link'],) for result in organic_results]
 
-    # YouTube Data API를 사용하여 비디오 검색
-    youtube = build('youtube', 'v3', developerKey=API_KEY)
-    video_search_result = youtube.search().list(q=query, part='snippet', type='video', maxResults=5).execute()
-    video_results = [f"https://www.youtube.com/watch?v={item['id']['videoId']}" for item in video_search_result.get('items', [])]
+    # SerpAPI를 사용하여 YouTube 검색
+    youtube_params = {
+        "engine": "youtube",
+        "search_query": query,
+        "api_key": SERP_API_KEY
+    }
+    youtube_search = GoogleSearch(youtube_params)
+    youtube_results = youtube_search.get_dict()
+    video_results = [(f"https://www.youtube.com/watch?v={video['video_id']}",) for video in youtube_results.get("videos", [])]
 
-    # 검색 결과를 튜플의 리스트로 변환
-    web_results_tuples = [(url,) for url in web_results]
-    video_results_tuples = [(url,) for url in video_results]
-
-    return web_results_tuples, video_results_tuples
+    return web_results, video_results
 
 iface = gr.Interface(
-    fn=multi_search,
+    fn=multi_search_serpapi,
     inputs=gr.Textbox(lines=2, placeholder="검색할 텍스트를 입력하세요..."),
     outputs=[gr.List(label="웹 검색 결과"), gr.List(label="YouTube 비디오 검색 결과")],
     title="멀티 검색 결과 출력",