Update app.py

app.py

@@ -1,44 +1,52 @@
 from googleapiclient.discovery import build
 import gradio as gr
 
-# Google Cloud API 키와 Custom Search Engine ID를 여기에 입력하세요.
-# 주의: 실제 키와 ID를 공개적으로 공유하지 마세요.
 API_KEY = 'AIzaSyDUz3wkGal0ewRtPlzeMit88bV4hS4ZIVY'
 CSE_ID = '56b34994f47704ddd'
 
-def search_google(query):
-    # Google Custom Search JSON API를 사용하여 검색을 수행합니다.
+def search_google(query, search_type='image'):
     service = build("customsearch", "v1", developerKey=API_KEY)
-    res = service.cse().list(q=query, cx=CSE_ID, searchType='image', num=10).execute()
+    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']:
-            results.append({
-                'title': item['title'],
-                'link': item['link'],
-                'thumbnail': item.get('image', {}).get('thumbnailLink', 'No thumbnail available')
-            })
+            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
 
-def format_output(results):
-    # 검색 결과를 Gradio 인터페이스에 맞게 포맷팅합니다.
+def format_output(results, search_type='image'):
     output = []
     for result in results:
-        output.append(f"**{result['title']}**\nURL: [{result['link']}]({result['link']})\n![Thumbnail]({result['thumbnail']})")
+        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)
 
-# Gradio 인터페이스를 정의합니다.
+def search_and_format(query, search_type='image'):
+    return format_output(search_google(query, search_type), search_type)
+
+# Gradio 인터페이스 정의
 iface = gr.Interface(
-    fn=lambda query: format_output(search_google(query)),
-    inputs=gr.Textbox(placeholder="검색할 키워드를 입력하세요..."),
+    fn=search_and_format,
+    inputs=[gr.Textbox(placeholder="검색할 키워드를 입력하세요..."), gr.Radio(choices=["image", "video"], label="검색 유형")],
     outputs=gr.Markdown(),
-    examples=["Python", "Gradio", "Hugging Face"],
+    examples=[["Python", "image"], ["Gradio", "video"]],
     title="Google 검색 결과 및 썸네일 보기",
-    description="입력한 키워드에 대한 구글 검색 결과의 웹 링크와 썸네일을 보여줍니다."
+    description="입력한 키워드에 대한 구글 검색 결과의 웹 링크와 썸네일(이미지) 또는 비디오 링크를 보여줍니다."
 )
 
-# Gradio 앱을 실행합니다.
 if __name__ == "__main__":
     iface.launch()