|
import gradio as gr |
|
import requests |
|
from PIL import Image |
|
from io import BytesIO |
|
|
|
|
|
def google_image_search(query): |
|
API_KEY = 'AIzaSyDUz3wkGal0ewRtPlzeMit88bV4hS4ZIVY' |
|
CX = '여기에_당신의_커스텀_검색엔진_ID를_입력하세요' |
|
URL = f"https://www.googleapis.com/customsearch/v1?q={query}&cx={CX}&searchType=image&key={API_KEY}" |
|
|
|
response = requests.get(URL) |
|
results = response.json() |
|
images = [] |
|
|
|
for item in results.get('items', []): |
|
image_url = item['link'] |
|
response = requests.get(image_url) |
|
img = Image.open(BytesIO(response.content)) |
|
images.append(img) |
|
|
|
return images |
|
|
|
|
|
iface = gr.Interface(fn=google_image_search, |
|
inputs=gr.inputs.Textbox(lines=2, placeholder="검색할 이미지의 텍스트를 입력하세요..."), |
|
outputs=gr.outputs.Image(type="pil", label="검색 결과"), |
|
title="구글 이미지 검색", |
|
description="텍스트를 입력하면 구글 이미지 검색 결과를 보여줍니다.") |
|
|
|
|
|
iface.launch() |
