Update app.py

app.py

@@ -1,51 +1,42 @@
-import gradio as gr
 import requests
+import gradio as gr
 
-# Pexels API 키 설정
-PEXELS_API_KEY = '여기에_당신의_API_키를_입력하세요'
+# Pexels API 키를 여기에 입력하세요
+API_KEY = '5woz23MGx1QrSY0WHFb0BRi29JvbXPu97Hg0xnklYgHUI8G0w23FKH62'
+API_URL = 'https://api.pexels.com/v1/search'
 
 def search_images(keyword):
-    headers = {
-        'Authorization': PEXELS_API_KEY
-    }
-    params = {
-        'query': keyword,
-        'per_page': 80  # 한 번에 반환받을 이미지 수
-    }
-    url = 'https://api.pexels.com/v1/search'
-    response = requests.get(url, headers=headers, params=params)
-    data = response.json()
-    
-    # 검색된 이미지의 URL을 리스트로 추출
-    image_urls = [photo['src']['original'] for photo in data['photos']]
-    return image_urls
-
-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_download_links = [video['video_files'][0]['link'] for video in data['videos']]
-    return video_download_links
+    headers = {'Authorization': API_KEY}
+    params = {'query': keyword, 'per_page': 8}  # per_page 값을 8로 설정하여 8개의 이미지를 검색
+    response = requests.get(API_URL, headers=headers, params=params)
+   
+    if response.status_code == 200 and response.json()['photos']:
+        photos = response.json()['photos']
+        result_html = ""
+        for photo in photos:
+            image_url = photo['src']['original']
+            photographer = photo['photographer']
+            photographer_url = photo['photographer_url']
+            photo_url = photo['url']
+           
+            # 각 이미지와 크레딧을 HTML 형식으로 추가
+            result_html += f"""
+            <div style='margin-bottom: 20px;'>
+                <img src='{image_url}' width='100%' />
+                <p>This <a href='{photo_url}'>Photo</a> was taken by <a href='{photographer_url}'>{photographer}</a> on <a href='https://www.pexels.com'>Pexels</a>.</p>
+            </div>
+            """
+        return result_html
+    else:
+        return "검색 결과가 없습니다."
 
-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.Markdown(label="검색된 비디오")
-            video_search_input.change(search_videos, inputs=video_search_input, outputs=video_search_output)
+interface = gr.Interface(
+    fn=search_images,
+    inputs=gr.Textbox(lines=2, placeholder="검색할 이미지 키워드를 입력하세요"),
+    outputs=gr.HTML(label="검색 결과"),
+    title="Pexels 이미지 검색기",
+    description="Pexels에서 키워드에 맞는 이미지를 검색하고, 해당 이미지의 출처와 작가에 대한 정보를 제공합니다."
+)
 
-demo.launch()
+if __name__ == "__main__":
+    interface.launch()