Update app.py

app.py

@@ -10,6 +10,25 @@ import PIL
 from PIL import Image
 from typing import Tuple
 
+import os
+import requests
+from moviepy.editor import VideoFileClip
+from moviepy.audio.AudioClip import AudioClip
+
+def search_pexels_images(query):
+    API_KEY = os.getenv("API_KEY")
+    url = f"https://api.pexels.com/v1/search?query={query}&per_page=80"
+    headers = {"Authorization": API_KEY}
+    response = requests.get(url, headers=headers)
+    data = response.json()
+    images_urls = [photo['src']['medium'] for photo in data['photos']]
+    return images_urls
+
+def show_search_results(query):
+    images_urls = search_pexels_images(query)
+    return images_urls
+    
+
 net=BriaRMBG()
 # model_path = "./model1.pth"
 model_path = hf_hub_download("briaai/RMBG-1.4", 'model.pth')
@@ -134,4 +153,17 @@ with gr.Blocks() as demo:
                 outputs=result_merge,
             )
 
+            
+        with gr.TabItem("Image Search"):
+            with gr.Column():
+                gr.Markdown("### FREE Image Search")
+                search_query = gr.Textbox(label="사진 검색")
+                search_btn = gr.Button("검색")
+                images_output = gr.Gallery(label="검색 결과 이미지")
+                search_btn.click(
+                    fn=show_search_results,
+                    inputs=search_query,
+                    outputs=images_output
+                )
+
 demo.launch()