Update app.py

app.py

@@ -20,14 +20,32 @@ def search_images(keyword):
     image_urls = [photo['src']['original'] for photo in data['photos']]
     return image_urls
 
-# Gradio 인터페이스 생성
-iface = gr.Interface(
-    fn=search_images,
-    inputs=gr.Textbox(label="검색 키워드"),
-    outputs=gr.Gallery(label="검색된 이미지"),
-    title="Pexels 이미지 검색기",
-    description="Pexels에서 키워드에 해당하는 고화질 이미지를 검색합니다."
-)
+def search_videos(keyword):
+    headers = {
+        'Authorization': PEXELS_API_KEY
+    }
+    params = {
+        'query': keyword,
+        'per_page': 80  # 한 번에 반환받을 비디오 수
+    }
+    url = 'https://api.pexels.com/videos/search'
+    response = requests.get(url, headers=headers, params=params)
+    data = response.json()
+    
+    # 검색된 비디오의 URL을 리스트로 추출
+    video_urls = [video['video_files'][0]['link'] for video in data['videos']]
+    return video_urls
+
+with gr.Blocks() as demo:
+    with gr.Tabs():
+        with gr.Tab("이미지 검색"):
+            image_search_input = gr.Textbox(label="검색 키워드")
+            image_search_output = gr.Gallery(label="검색된 이미지")
+            image_search_input.change(search_images, inputs=image_search_input, outputs=image_search_output)
+        
+        with gr.Tab("비디오 검색"):
+            video_search_input = gr.Textbox(label="검색 키워드")
+            video_search_output = gr.Gallery(label="검색된 비디오")
+            video_search_input.change(search_videos, inputs=video_search_input, outputs=video_search_output)
 
-# 인터페이스 실행
-iface.launch()
+demo.launch()