Update app.py

app.py

@@ -1,51 +1,33 @@
 from googleapiclient.discovery import build
 import gradio as gr
 
+# API 키와 검색 엔진 ID 설정
 API_KEY = 'AIzaSyDUz3wkGal0ewRtPlzeMit88bV4hS4ZIVY'
 CSE_ID = '56b34994f47704ddd'
 
-def search_google(query, search_type='image'):
+def multi_search(query):
+    # Google Custom Search JSON API를 사용하여 웹 검색
     service = build("customsearch", "v1", developerKey=API_KEY)
-    res = service.cse().list(q=query, cx=CSE_ID, searchType=search_type, num=10).execute()
-    
-    results = []
-    if 'items' in res:
-        for item in res['items']:
-            if search_type == 'image':
-                results.append({
-                    'title': item['title'],
-                    'link': item['link'],
-                    'thumbnail': item.get('image', {}).get('thumbnailLink', 'No thumbnail available')
-                })
-            elif search_type == 'video':
-                # 비디오 검색 결과 처리 (비디오 썸네일이나 링크가 다를 수 있음)
-                results.append({
-                    'title': item['title'],
-                    'link': item['link']
-                })
-    return results
+    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 format_output(results, search_type='image'):
-    output = []
-    for result in results:
-        if search_type == 'image':
-            output.append(f"**{result['title']}**\nURL: [{result['link']}]({result['link']})\n![Thumbnail]({result['thumbnail']})")
-        elif search_type == 'video':
-            # 비디오 결과 포맷팅
-            output.append(f"**{result['title']}**\nURL: [{result['link']}]({result['link']})")
-    return "\n\n".join(output)
+    # 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', [])]
 
-def search_and_format(query, search_type='image'):
-    return format_output(search_google(query, search_type), search_type)
+    # 검색 결과를 튜플의 리스트로 변환
+    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
 
-# Gradio 인터페이스 정의
 iface = gr.Interface(
-    fn=search_and_format,
-    inputs=[gr.Textbox(placeholder="검색할 키워드를 입력하세요..."), gr.Radio(choices=["image", "video"], label="검색 유형")],
-    outputs=gr.Markdown(),
-    examples=[["Python", "image"], ["Gradio", "video"]],
-    title="Google 검색 결과 및 썸네일 보기",
-    description="입력한 키워드에 대한 구글 검색 결과의 웹 링크와 썸네일(이미지) 또는 비디오 링크를 보여줍니다."
+    fn=multi_search,
+    inputs=gr.Textbox(lines=2, placeholder="검색할 텍스트를 입력하세요..."),
+    outputs=[gr.List(label="웹 검색 결과"), gr.List(label="YouTube 비디오 검색 결과")],
+    title="멀티 검색 결과 출력",
+    description="입력 텍스트를 기준으로 웹 검색 결과와 YouTube 비디오 검색 결과를 구분하여 출력합니다."
 )
 
 if __name__ == "__main__":