Update app.py

app.py

@@ -1,4 +1,51 @@
 import streamlit as st
+import requests
+
+# Pexels API 키 설정
+PEXELS_API_KEY = "5woz23MGx1QrSY0WHFb0BRi29JvbXPu97Hg0xnklYgHUI8G0w23FKH62"
+
+def search_image(keyword):
+    """이미지 검색 함수"""
+    url = f"https://api.pexels.com/v1/search?query={keyword}+query&per_page=1"
+    headers = {"Authorization": PEXELS_API_KEY}
+    response = requests.get(url, headers=headers).json()
+    if response["photos"]:
+        # 고해상도 이미지 URL 반환
+        return response["photos"][0]["src"]["original"]
+    return ""
+
+def search_video(keyword):
+    """비디오 검색 함수"""
+    url = f"https://api.pexels.com/videos/search?query={keyword}+query&per_page=1"
+    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 ""
+
+# '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.")
+
+# '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.")
+
+
+
 
 # 페이지 설정
 st.set_page_config(page_title="ViDraft", layout="wide")