Update app.py

app.py

@@ -42,47 +42,46 @@ else:
 # Pexels API 키 설정
 PEXELS_API_KEY = "5woz23MGx1QrSY0WHFb0BRi29JvbXPu97Hg0xnklYgHUI8G0w23FKH62"
 
-def search_image(keyword):
+def search_images(keyword, per_page=80):
     """이미지 검색 함수"""
-    url = f"https://api.pexels.com/v1/search?query={keyword}+query&per_page=1"
+    url = f"https://api.pexels.com/v1/search?query={keyword}&per_page={per_page}"
     headers = {"Authorization": PEXELS_API_KEY}
     response = requests.get(url, headers=headers).json()
-    if response["photos"]:
-        # 고해상도 이미지 URL 반환
-        return response["photos"][0]["src"]["original"]
-    return ""
+    images = []
+    if 'photos' in response:
+        for photo in response['photos']:
+            images.append(photo['src']['original'])
+    return images
 
-def search_video(keyword):
+def search_videos(keyword, per_page=80):
     """비디오 검색 함수"""
-    url = f"https://api.pexels.com/videos/search?query={keyword}+query&per_page=1"
+    url = f"https://api.pexels.com/videos/search?query={keyword}&per_page={per_page}"
     headers = {"Authorization": PEXELS_API_KEY}
     response = requests.get(url, headers=headers).json()
-    if response["videos"]:
-        # 고해상도 비디오 URL 반환
-        return response["videos"][0]["video_files"][0]["link"]
-    return ""
+    videos = []
+    if 'videos' in response:
+        for video in response['videos']:
+            # 최고해상도의 비디오 파일 링크를 선택합니다
+            videos.append(video['video_files'][0]['link'])
+    return videos
 
-# 'Search Image' 메뉴 선택 시 처리 로직
+# 'Search Image' 또는 'Search Video' 메뉴 선택 로직이 적절한 위치에 있어야 합니다.
+
+# 예를 들어 'Search Image' 선택 시
 if selected_menu == "Free Stock" and st.session_state['current_sub_menu'] == "Search Image":
     keyword = st.text_input("Enter a keyword to search for images")
     if keyword:
-        image_url = search_image(keyword)
-        if image_url:
-            st.image(image_url)
-        else:
-            st.write("No results found.")
+        images = search_images(keyword)
+        for img in images:
+            st.image(img)
 
-# 'Search Video' 메뉴 선택 시 처리 로직
+# 'Search Video' 선택 시
 if selected_menu == "Free Stock" and st.session_state['current_sub_menu'] == "Search Video":
     keyword = st.text_input("Enter a keyword to search for videos")
     if keyword:
-        video_url = search_video(keyword)
-        if video_url:
-            st.video(video_url)
-        else:
-            st.write("No results found.")
-
-
+        videos = search_videos(keyword)
+        for video in videos:
+            st.video(video)
 
 
 # 'Home' 페이지 표시