Update app.py

app.py

@@ -1,29 +1,29 @@
+from googleapiclient.discovery import build
 import gradio as gr
-import requests
 
-# 구글 이미지 검색 함수
-def google_image_search(query):
-    API_KEY = 'AIzaSyDUz3wkGal0ewRtPlzeMit88bV4hS4ZIVY'
-    CX = '56b34994f47704ddd'
-    URL = f"https://www.googleapis.com/customsearch/v1?q={query}&cx={CX}&searchType=image&key={API_KEY}"
-    
-    response = requests.get(URL)
-    results = response.json()
-    image_urls = []
+# API 키와 검색 엔진 ID 설정
+API_KEY = 'AIzaSyDUz3wkGal0ewRtPlzeMit88bV4hS4ZIVY'
+CSE_ID = '56b34994f47704ddd'
 
-    for item in results.get('items', []):
-        image_url = item['link']
-        image_urls.append(image_url)
+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', [])]
 
-    # 이미지 URL 리스트를 직접 반환
-    return image_urls
+    # 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', [])]
 
-# Gradio 인터페이스 정의
-iface = gr.Interface(fn=google_image_search, 
-                     inputs=gr.Textbox(lines=2, placeholder="검색할 이미지의 텍스트를 입력하세요..."),
-                     outputs=gr.Gallery(label="검색 결과 이미지"),              
-                     title="구글 이미지 검색",
-                     description="텍스트를 입력하면 구글 이미지 검색 결과를 이미지로 보여줍니다.")
+    return web_results, video_results
+
+iface = gr.Interface(
+    fn=multi_search,
+    inputs=gr.Textbox(lines=2, placeholder="검색할 텍스트를 입력하세요..."),
+    outputs=[gr.List(label="웹 검색 결과"), gr.List(label="YouTube 비디오 검색 결과")],
+    title="멀티 검색 결과 출력",
+    description="입력 텍스트를 기준으로 웹 검색 결과와 YouTube 비디오 검색 결과를 구분하여 출력합니다."
+)
 
-# 인터페이스 실행
 iface.launch()