Update app.py

app.py

@@ -1,25 +1,33 @@
 import gradio as gr
 import requests
 
-def fetch_high_quality_images(api_key, query, per_page=80):
-    url = "https://api.pexels.com/v1/search"
+# Pexels API 키 설정
+PEXELS_API_KEY = '5woz23MGx1QrSY0WHFb0BRi29JvbXPu97Hg0xnklYgHUI8G0w23FKH62'
+
+def search_images(keyword):
     headers = {
-        "Authorization": api_key
+        'Authorization': PEXELS_API_KEY
     }
     params = {
-        "query": query,
-        "per_page": per_page,
-        "size": "large"
+        'query': keyword,
+        'per_page': 80  # 한 번에 반환받을 이미지 수
     }
+    url = 'https://api.pexels.com/v1/search'
     response = requests.get(url, headers=headers, params=params)
-    if response.status_code == 200:
-        images = response.json()['photos']
-        for image in images:
-            print(image['src']['original'])
-    else:
-        print("Error:", response.status_code)
+    data = response.json()
+    
+    # 검색된 이미지의 URL을 리스트로 추출
+    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에서 키워드에 해당하는 고화질 이미지를 검색합니다."
+)
 
-if __name__ == "__main__":
-    api_key = "5woz23MGx1QrSY0WHFb0BRi29JvbXPu97Hg0xnklYgHUI8G0w23FKH62"  # 여기에 Pexels API 키를 입력하세요.
-    query = "자연"  # 검색할 이미지의 키워드를 여기에 직접 지정하세요.
-    fetch_high_quality_images(api_key, query)
+# 인터페이스 실행
+iface.launch()